使用注释处理器替换代码

Question

我正在尝试编写一个注释处理器来插入类上的方法和字段......留档如此稀疏。我走不远，我不知道我是否正确地接近它。

处理环境提供了一个Filer对象，它具有创建新源文件和类文件的方便方法。这些工作正常，但后来我试图弄清楚如何读取现有源文件，它提供的只是“getResources”。所以在我的处理器实现中，我这样做了：

@Override
public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
    try {
        for (TypeElement te : annotations) {
            for (Element element : roundEnv.getElementsAnnotatedWith(te)) {
                FileObject in_file = processingEnv.getFiler().getResource(
                    StandardLocation.SOURCE_PATH, "",
                    element.asType().toString().replace(".", "/") + ".java");

                FileObject out_file = processingEnv.getFiler().getResource(
                    StandardLocation.SOURCE_OUTPUT, "",
                    element.asType().toString().replace(".", "/") + ".java");

                //if (out_file.getLastModified() >= in_file.getLastModified()) continue;

                CharSequence data = in_file.getCharContent(false);

                data = transform(data); // run the macro processor

                JavaFileObject out_file2 = processingEnv.getFiler().createSourceFile(
                    element.asType().toString(), element);
                Writer w = out_file2.openWriter();
                w.append(data);
                w.close();
            }
        }
    } catch (Exception e) {
        e.printStackTrace();
        processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, e.getMessage());
    }
    return true;
}

我的第一个困惑是我不禁觉得element.asType（）. toString（）。替换(".", "/") ".java"（获取合格的类型名称并将其转换为包和源文件路径）不是解决问题的好方法。API的其余部分设计过度，但似乎没有一种方便的方法来检索原始源代码。

真正的问题是，然后编译器会被输出目录中的第二个源文件（“错误：duplicate class”）打乱，现在我被卡住了。

我已经编写了其余的内容——一个宏词法分析器和解析器等等，用于计算一些数据并插入字段值和方法——但它是编译器之外的第一步。除了原始文件不能有. java扩展名（以防止编译器看到它们）之外，这很好。然后我听说注释可以进行代码生成，我认为这会更合适和方便，但我找不到太多指导。

Answer 1

我不知道这个提示是否符合您的需求，但您的想法将与依赖注入（例如Spring）一起工作。

1. 为Spring将忽略的类型和保留“源”创建自定义注释
2. 在注释处理器中对代码进行所需的更改，将Spring的@Component注释添加到生成的Java类型中，并将其保存为原始类型的子类型，其类名与原始类型不同
3. 当Spring创建上下文时，它将根据您的处理结果而不是原始结果加载类

相反，Lombok直接操作抽象源代码树，这比生成源代码复杂得多。这里的问题是，Java编译器模块是公开的（Jigsaw）。

另一种替代方法是使用字节码api，比如ByteBuddy，在运行时代理原始类。如果不想为实例化类编写代码，这也需要依赖项注入。这里的魔力是配置类中类型集合的自动连接字段。如果你对此感兴趣，请告诉我。然后我会在这里补充更多细节。

Answer 2

有点晚了：），但一个解决方案可能是在构建过程中使用Byte Buddy的transformer。参见https://github.com/raphw/byte-buddy/tree/master/byte-buddy-maven-plugin.

Answer 3

注释处理器背后的意图是允许开发人员添加新类，而不是替换现有类。也就是说，有一个bug允许您向现有类添加代码。ProjectLombok利用这一点将getter和setter（以及其他内容）添加到编译的java类中。

“替换”方法/字段的方法是从输入类扩展或委托给输入类。这允许您覆盖/转移对目标类的调用。

因此，如果这是您的输入类：

inputinpl。爪哇：

public class InputImpl implements Input{
    public void foo(){
        System.out.println("foo");
    }
    public void bar(){
        System.out.println("bar");
    }
}

您可以生成以下内容来“替换”它：

我mpl.java：

public class InputReplacementImpl implements Input{

    private Input delegate;

    //setup delegate....

    public void foo(){
        System.out.println("foo replacement");
    }
    public void bar(){
        delegate.bar();
    }
}

这就引出了一个问题，如何引用InputReplace ementImpl而不是InputImpl。您可以生成更多代码来执行包装，也可以简单地调用预期生成的代码的构造函数。

我不太确定你的问题是什么，但我希望这能对你的问题有所启发。