动态代码生成技术在 Presto 中使用简介

最新推荐文章于 2023-09-05 14:00:58 发布

过往记忆

最新推荐文章于 2023-09-05 14:00:58 发布

阅读量1.4k

点赞数

文章标签： java python 反射 spring leetcode

本文链接：https://blog.csdn.net/wypblog/article/details/120755617

版权

本文介绍了Presto如何利用动态代码生成技术，以EmbedVersion和concat函数为例，展示了如何在运行时创建Runnable实例和实现内置函数。在Presto中，动态代码生成用于提高性能，例如在TaskRunner的启动和concat函数实现中，通过ASM操作字节码，自动生成高效执行逻辑。此外，对于concat函数，生成的代码会根据函数签名进行缓存，以避免重复生成。

摘要由CSDN通过智能技术生成

在《ASM 与 Presto 动态代码生成简介

EmbedVersion

我们往 Presto 提交 SQL 查询以及 TaskExecutor 启动 TaskRunner 执行 Task 的时候都会使用到 EmbedVersion 类里面的 embedVersion 方法。embedVersion 方法其实就是初始化一个 Runnable 实例，比如启动 TaskRunner 的代码片段如下：

executor.execute(embedVersion.embedVersion(new TaskRunner()));

其中 TaskRunner 就是实现 Runnable 接口的。EmbedVersion 的 embedVersion 方法实现如下：

public Runnable embedVersion(Runnable runnable)
{
    requireNonNull(runnable, "runnable is null");
    try {
        return (Runnable) runnableConstructor.invoke(runnable);
    }
    catch (Throwable throwable) {
        throwIfUnchecked(throwable);
        throw new RuntimeException(throwable);
    }
}

其中 runnableConstructor 就是使用 ASM 进行代码生成的类，实现如下：

// 这里定义了一个类，类名大概为 Presto_null__testversion____20211011_105831_1，
// 它的父类是 Object，并实现了 Runnable 接口。
ClassDefinition classDefinition = new ClassDefinition(
    a(PUBLIC, FINAL),
    makeClassName(baseClassName(serverConfig)),
    type(Object.class),
    type(Runnable.class));
// 定义了一个名为 runnable 的局部变量，类型为 Runnable
FieldDefinition field = classDefinition.declareField(a(PRIVATE), "runnable", Runnable.class);
Parameter parameter = arg("runnable", type(Runnable.class));
// 定义了这个类的构造函数，参数为 runnable，参数类型为 Runnable
MethodDefinition constructor = classDefinition.declareConstructor(a(PUBLIC), parameter);
// 构造方法里面其实就是把参数 runnable 的值赋值给局部变量 runnable
constructor.getBody()
    .comment("super(runnable);")
    .append(cons