注释处理器生成未创建文件(JavaPoet-AutoService)

时间:2021-01-06 19:17:15

标签: java spring annotation-processing maven-compiler-plugin javapoet

我正在尝试学习处理器生成,为此我使用 JavaPoet 和 Google AutoService。我用 maven-compiler-plugin 编译我的代码没有任何错误,出现了生成的源文件夹,但我看不到里面有任何生成的文件。

我也没有找到读取 processingEnv.getMessager 日志的方法,所以我现在完全卡住了。请帮忙。

这是我的处理器:

package com.annotationprocessing.demo;

import com.annotationprocessing.demo.annotation.NewLoginApplicable;
import com.google.auto.service.AutoService;
import com.squareup.javapoet.*;

import javax.annotation.processing.AbstractProcessor;
import javax.annotation.processing.Processor;
import javax.annotation.processing.RoundEnvironment;
import javax.annotation.processing.SupportedAnnotationTypes;
import javax.lang.model.element.Element;
import javax.lang.model.element.ElementKind;
import javax.lang.model.element.Modifier;
import javax.lang.model.element.TypeElement;
import javax.tools.Diagnostic;
import java.io.File;
import java.io.IOException;
import java.util.*;
import java.util.stream.Collectors;


@SupportedAnnotationTypes("com.annotationprocessing.demo.annotation.NewLoginApplicable.java")
@AutoService(Processor.class)
public class NewLoginInterceptorProcessor extends AbstractProcessor {

    @Override
    public boolean process(Set<? extends TypeElement> annotations,
                           RoundEnvironment roundEnv) {

        // The elements which have NewLoginApplicable annotation
        Set<? extends Element> annotatedElements = roundEnv
                .getElementsAnnotatedWith(NewLoginApplicable.class);

        Map<Boolean, List<Element>> annotatedMethods = annotatedElements.stream().collect(
                Collectors.partitioningBy(element -> element.getKind() == ElementKind.METHOD));

        Set<Element> annotatedCorrectly = new HashSet<>(annotatedMethods.get(true));
        List<Element> faulty = annotatedMethods.get(false);

        faulty.forEach(element -> processingEnv.getMessager().printMessage(
                Diagnostic.Kind.ERROR,
                "NewLoginApplicable annotation should only be used on methods ",
                element));

        String className = "NewLoginApplicableMap";

        TypeSpec typeSpec = buildClassTypeSpec(annotatedCorrectly, className);
        String packageName = "com.annotationprocessing.demo";

        if (typeSpec != null) {
            try {
                writeToFile(typeSpec, packageName);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        return true;
    }

    private static TypeSpec buildClassTypeSpec(Set<? extends Element> elements, String className) {
        if (elements.isEmpty())
            return null;

        ClassName hashmap = ClassName.get("java.util", "HashMap");
        String hashmapObjectName = "getEndpointMap";

        TypeSpec.Builder typeSpecBuilder = TypeSpec.classBuilder(className)
                .addModifiers(Modifier.PUBLIC);

        FieldSpec fieldSpec = FieldSpec.builder(HashMap.class, hashmapObjectName,
                Modifier.PRIVATE, Modifier.STATIC).initializer("new HashMap()").build();
        typeSpecBuilder.addField(fieldSpec);

        MethodSpec.Builder getterMethodBuilder = MethodSpec.methodBuilder(hashmapObjectName)
                .returns(HashMap.class)
                .addModifiers(Modifier.PUBLIC)
                .addModifiers(Modifier.STATIC)
                .addStatement("return this.endpointMap()");

        typeSpecBuilder.addMethod(getterMethodBuilder.build());

        return typeSpecBuilder.build();
    }

    private void writeToFile(TypeSpec typeSpec, String packageName) throws IOException {
        JavaFile.builder(packageName, typeSpec).build().writeTo(processingEnv.getFiler());
        processingEnv.getMessager().printMessage(Diagnostic.Kind.NOTE, "Code generation complete!");
    }
}

mvn compile 后我的文件夹:

enter image description here

0 个答案:

没有答案
相关问题