編寫最基本的APT Demo

-l囧l-發表於2017-11-14

簡介

APT,就是Annotation Processing Tool 的簡稱,簡單來說就是通過編碼來動態得到解析Annotation的工具。一般分為兩類:
1.執行時註解:比如大名鼎鼎的retrofit就是用執行時註解,通過動態代理來生成網路請求
2.編譯時註解:比如Dagger2, ButterKnife, EventBus3

程式碼實現

這裡我們要實現一個怎樣的功能呢?第一個就是給我們的activity新增一個@Flag,然後當我們編譯的時候就會生成一個java main函式。第二個就是我簡易版的butterknife。2個註解都是寫在用一個外掛中。好了下面直接開始。

Annotation module

首先我們建立一個my_annotation的java module,這個專案只放我們的註解檔案,不涉及到註解處理等其他邏輯,關於註解處理我們會新建一個module來處理。專案結構如下:

image.png
image.png

其中build.gradle中基本不用修改,保持預設的配置就可以,如下:
image.png
image.png

Flag註解

image.png
image.png

就是這麼簡單,關於註解中元註解的解釋請參考我的另一篇文章:元註解簡介
關於註解module就到這了。

註解處理 module

首先我們建立一個my_compiler的java module。專案結構如下:

image.png
image.png

build.gradle的配置如下:
image.png
image.png

簡單解釋下幾個dependencies

auto-service:這是google推出的方便我們編寫annotation外掛,在沒有這個這個庫之前,我們需要對我們的外掛做很多的配置才能使用,有了這個庫以後就方便多了,下文會看到怎麼用,這裡就不介紹了。
javapoet:這是方便我們在編譯時動態生成class檔案的,下文也會有具體怎麼使用,這裡不做過多解釋。

關於上面2個庫大家感興趣可以去查詢相關資料進行進一步瞭解。下面我們看下真正的註解處理類:

/**
 * @author Jin
 */
//來自auto-service 只要新增這個註解以後就不需要做其他配置,現在已經可以在專案中直接使用了
@AutoService(Processor.class)
public class FlagAnnotationProcessor extends AbstractProcessor {


    /**
     * getSupportedSourceVersion()方法返回 Java 版本 預設為Java6
     *
     * @return Java 版本
     */
    @Override
    public SourceVersion getSupportedSourceVersion() {
        return SourceVersion.latestSupported();
    }

    /**
     * 返回要處理的註解的結合 這裡只處理RouterAnnotion型別的註解
     *
     * @return
     */
    @Override
    public Set<String> getSupportedAnnotationTypes() {
        LinkedHashSet<String> types = new LinkedHashSet<>();
        types.add(Flag.class.getCanonicalName());
        return types;
    }

    /**
     * 註解的具體處理類
     *
     * @param annotations
     * @param roundEnv
     * @return
     */
    @Override
    public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
        //來自javapoet  動態生成方法
        MethodSpec main = MethodSpec.methodBuilder("main")
                .addModifiers(Modifier.PUBLIC, Modifier.STATIC)
                .returns(void.class)
                .addParameter(String[].class, "args")
                .addStatement("$T.out.println($S)", System.class, "Hello, JavaPoet!")
                .build();
        //來自javapoet  動態生成類
        TypeSpec helloWorld = TypeSpec.classBuilder("HelloWorld")
                .addModifiers(Modifier.PUBLIC, Modifier.FINAL)
                .addMethod(main)
                .build();
        //來自javapoet  動態生成檔案
        JavaFile javaFile = JavaFile.builder("com.jin.helloworld", helloWorld)
                .build();
        try {
            javaFile.writeTo(processingEnv.getFiler());
        } catch (IOException e) {
            e.printStackTrace();
        }
        return true;
    }
}複製程式碼

註解處理類已經寫好了,下面我們看下怎麼關聯到我們的專案中。

註解使用

image.png
image.png

AndroidStudio3.0使用annotationProcessor來處理註解。
image.png
image.png

新增我們的@Flag註解。ok。大功告成,下面重新編譯(rebuild project)一下我們的專案,就會在build目錄下看到自動生成的程式碼了。
image.png
image.png

你可能會說,生成一個HelloWorld main函式並木有什麼卵用啊,是的,但是你起碼掌握了最基本的關於Annotation專案的建立、編譯、使用了是不是,麻雀雖小但是五臟俱全啊。

下面進入我們的另一個demo,簡易版butterknife。

簡易版butterknife

下面我直接添上註解和處理程式碼(相關解釋會在註釋中):

image.png
image.png

image.png
image.png

DIAnnotationProcessor

@AutoService(Processor.class)
public class DIAnnotationProcessor extends AbstractProcessor {
    private Filer mFiler;
    private Elements elementUtils;

    /**
     * init()方法可以初始化拿到一些使用的工具,
     * 比如檔案相關的輔助類 Filer;元素相關的輔助類Elements;日誌相關的輔助類Messager;
     *
     * @param processingEnv
     */
    @Override
    public synchronized void init(ProcessingEnvironment processingEnv) {
        super.init(processingEnv);
        mFiler = processingEnv.getFiler();
        elementUtils = processingEnv.getElementUtils();
    }

    /**
     * getSupportedSourceVersion()方法返回 Java 版本 預設為Java6
     *
     * @return Java 版本
     */
    @Override
    public SourceVersion getSupportedSourceVersion() {
        return SourceVersion.latestSupported();
    }

    /**
     * 返回要處理的註解的結合 這裡只處理RouterAnnotion型別的註解
     *
     * @return
     */
    @Override
    public Set<String> getSupportedAnnotationTypes() {
        LinkedHashSet<String> types = new LinkedHashSet<>();
        types.add(BindActivity.class.getCanonicalName());
        return types;
    }

    /**
     * 註解的具體處理類
     *
     * @param annotations
     * @param roundEnv
     * @return
     */
    @Override
    public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
        System.out.println("DIAnnotationProcessor");
        //得到所有被Bind新增註解的類
        Set<? extends Element> elements = roundEnv.getElementsAnnotatedWith(BindActivity.class);
        for (Element element : elements) {
            //強制轉換成TypeElement 判斷是否是Class
            TypeElement typeElement = (TypeElement) element;
            //得到typeElement類中所有成員變數和成員方法
            List<? extends Element> members = elementUtils.getAllMembers(typeElement);
            MethodSpec.Builder bindViewMethodSpecBuilder = MethodSpec.methodBuilder("bindView")
                    .addModifiers(Modifier.PUBLIC, Modifier.STATIC)
                    .returns(TypeName.VOID)
                    .addParameter(ClassName.get(typeElement.asType()), "activity");
            for (Element item : members) {
                BindMyView bindView = item.getAnnotation(BindMyView.class);
                if (bindView == null) {
                    continue;
                }
                bindViewMethodSpecBuilder.addStatement(String.format("activity.%s = (%s) activity.findViewById(%s)", item.getSimpleName(), ClassName.get(item.asType()).toString(), bindView.value()));
            }

            TypeSpec typeSpec = TypeSpec.classBuilder("DI" + element.getSimpleName())
                    .superclass(TypeName.get(typeElement.asType()))
                    .addModifiers(Modifier.PUBLIC, Modifier.FINAL)
                    .addMethod(bindViewMethodSpecBuilder.build())
                    .build();
            JavaFile javaFile = JavaFile.builder(getPackageName(typeElement), typeSpec).build();
            try {
                javaFile.writeTo(processingEnv.getFiler());
            } catch (IOException e) {
                e.printStackTrace();
            }

        }
        return true;
    }

    private String getPackageName(TypeElement type) {
        return elementUtils.getPackageOf(type).getQualifiedName().toString();
    }
}複製程式碼

DI Annotation使用

image.png
image.png

注意此時一定要先rebuild project生成如下檔案
image.png
image.png

然後在我們的程式碼中寫上
image.png
image.png

這裡ButterKnife.bind(this)是用來對比,請大家注意, DIMainActivity.bindView(this)才是我們直接生成的檔案。

#除錯(AndroidStudio3.0)已解決
如果過你也像我一樣按照網上的教程操作,但是始終報錯:Unable to open debugger port (localhost:5006): java.net.ConnectException "Connection refused: connect"
配置如下:

image.png
image.png

上面的配置程式碼記得在全域性的gradle.properties中新增 該檔案一般位於C:\Users\Jin.gradle下 如果沒有親手動建立

image.png
image.png

相關文章