com.zhucan.jpa.conversion.processor.EnumConvertProcessor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-boot-starter-jpa-conversion Show documentation
Demo project for Spring Boot
The newest version!
package com.zhucan.jpa.conversion.processor;

import com.google.auto.service.AutoService;
import com.squareup.javapoet.*;
import com.zhucan.jpa.conversion.annotation.EnumAutoConverter;
import com.zhucan.jpa.conversion.converter.AbstractEnumConverter;

import javax.annotation.processing.*;
import javax.lang.model.SourceVersion;
import javax.lang.model.element.Element;
import javax.lang.model.element.Modifier;
import javax.lang.model.element.PackageElement;
import javax.lang.model.element.TypeElement;
import javax.lang.model.util.Elements;
import javax.persistence.Converter;
import javax.tools.Diagnostic;
import java.io.IOException;
import java.util.Set;

/**
 * @author: zhuCan
 * @date: 2020/1/18 10:56
 * @description:
 */
@AutoService(Processor.class)
@SupportedSourceVersion(SourceVersion.RELEASE_8)
@SupportedAnnotationTypes("com.zhucan.jpa.conversion.annotation.EnumAutoConverter")
public class EnumConvertProcessor extends AbstractProcessor {

    private Filer filer;
    private Messager messager;
    private Elements elementUtils;
    public static final String doc = "\n This codes are generated automatically. Do not modify! \n -.- \n created by zhuCan \n";

    @Override
    public synchronized void init(ProcessingEnvironment processingEnvironment) {
        super.init(processingEnvironment);
        filer = processingEnv.getFiler();
        messager = processingEnvironment.getMessager();
        elementUtils = processingEnvironment.getElementUtils();
    }

    @Override
    public boolean process(Set annotations, RoundEnvironment roundEnv) {
        messager.printMessage(Diagnostic.Kind.NOTE, "Processor : " + getClass().getSimpleName());
        Set elements = roundEnv.getElementsAnnotatedWith(EnumAutoConverter.class);
        messager.printMessage(Diagnostic.Kind.NOTE, "annotations: " + annotations);
        messager.printMessage(Diagnostic.Kind.NOTE, "roundEnv: " + roundEnv);

        elements.forEach(x -> {
            // 被扫描的类的包路径
            PackageElement packageElement = elementUtils.getPackageOf(x);
            String packageName = packageElement.getQualifiedName().toString();

            // 类上注解
            EnumAutoConverter annotation = x.getAnnotation(EnumAutoConverter.class);

            // 构建类
            TypeSpec clazz = TypeSpec.classBuilder(x.getSimpleName() + "Converter")
                    .addModifiers(Modifier.PUBLIC, Modifier.FINAL)
                    .addAnnotation(AnnotationSpec.builder(Converter.class)
                            .addMember("autoApply", CodeBlock.builder()
                                    .add("$L", annotation.autoApply())
                                    .build())
                            .build())
                    .addJavadoc(" generator for enum converter " + doc)
                    .superclass(ParameterizedTypeName.get(ClassName.get(AbstractEnumConverter.class),
                            ClassName.get((TypeElement) x),
                            ClassName.get(Integer.class)))
                    .build();
            try {
                // 创建java文件
                JavaFile javaFile = JavaFile.builder(packageName, clazz)
                        .build();
                // 写入
                javaFile.writeTo(filer);
            } catch (IOException e) {
                e.printStackTrace();
            }

        });
        return false;
    }


}