android apt自動(dòng)生成代碼

最近新了一個(gè)架構(gòu)，之前用dagger2時(shí)候，每當(dāng)添加新activity還要修改或者新建component來完成dagger的注入。用了apt以后，在activity上標(biāo)注一個(gè)注解就可以了。

本文章用最簡(jiǎn)單的方法最直白的話 來搭建一個(gè)簡(jiǎn)單的apt編譯時(shí)期生成代碼

首先是新建一個(gè)android項(xiàng)目。就不說了

然后然后是新建立一個(gè)java的Module。注意是javalib。這個(gè)lib用來專門寫注解就好。為啥要單獨(dú)后面會(huì)說。

這個(gè)lib里面就先放一個(gè)注解，叫TestAnno

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;


@Retention(RetentionPolicy.CLASS)
@Target(ElementType.TYPE)
public @interface TestAnno {
}

RetentionPolicy.CLASS表示編譯時(shí)候注解。。
你需要關(guān)系的就是@Target(ElementType.TYPE)這個(gè)type是類的注解，可以有方法的，屬性的等等。

然后這個(gè)javalib的gradle文件要這么寫

apply plugin: 'java'

dependencies {
    compile fileTree(dir: 'libs', include: ['*.jar'])
}
sourceCompatibility = "1.7"
targetCompatibility = "1.7"

注解庫弄好了，在弄新建一個(gè)java lib 叫inject_comiler。這個(gè)是就是核心代碼了，在編譯時(shí)候，執(zhí)行這個(gè)個(gè)庫里面的代碼，然后 生成代碼。這個(gè)工程 三個(gè)類。一個(gè)是注解注解處理器的核心。一個(gè)是定義生成java文件的，方法拼接。還有一個(gè)就是常量包名類名了。
先看這個(gè)的gradle文件

apply plugin: 'java'
sourceCompatibility = "1.7"
targetCompatibility = "1.7"
dependencies {
    compile fileTree(include: ['*.jar'], dir: 'libs')
    compile 'com.google.auto.service:auto-service:1.0-rc2'//谷歌的幫助我們快速實(shí)現(xiàn)注解處理器
    compile project(':inject_annotation')//自己定義的注解的java lib
    compile 'com.squareup:javapoet:1.7.0'//用來生成java文件的，避免字符串拼接的尷尬
}

//這個(gè)注解是谷歌提供了，快速實(shí)現(xiàn)注解處理器，會(huì)幫你生成配置文件啥的 。直接用就好
@AutoService(Processor.class)
public class ActivityInjectProcesser extends AbstractProcessor {
    private Filer mFiler; //文件相關(guān)的輔助類
    private Elements mElementUtils; //元素相關(guān)的輔助類  許多元素
    private Messager mMessager; //日志相關(guān)的輔助類

    private Map<String, AnnotatedClass> mAnnotatedClassMap;

    @Override
    public synchronized void init(ProcessingEnvironment processingEnv) {
        super.init(processingEnv);
        mFiler = processingEnv.getFiler();
        mElementUtils = processingEnv.getElementUtils();
        mMessager = processingEnv.getMessager();
        mAnnotatedClassMap = new TreeMap<>();
    }

//這個(gè)方法是核心方法，在這里處理的你的業(yè)務(wù)。檢測(cè)類別參數(shù)，勝場(chǎng)java文件等
    @Override
    public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
        mAnnotatedClassMap.clear();

        try {
            processActivityCheck(roundEnv);
        } catch (Exception e) {
            e.printStackTrace();
            error(e.getMessage());
        }

        for (AnnotatedClass annotatedClass : mAnnotatedClassMap.values()) {
            try {
                annotatedClass.generateActivityFile().writeTo(mFiler);
            } catch (Exception e) {
                error("Generate file failed, reason: %s", e.getMessage());
            }
        }
        return true;
    }


    private void processActivityCheck(RoundEnvironment roundEnv) throws IllegalArgumentException, ClassNotFoundException {
        //check ruleslass forName(String className
        for (Element element : roundEnv.getElementsAnnotatedWith((Class<? extends Annotation>) Class.forName(TypeUtil.ANNOTATION_PATH))) {
            if (element.getKind() == ElementKind.CLASS) {
                getAnnotatedClass(element);
            } else
                error("ActivityInject only can use  in ElementKind.CLASS");
        }
    }

    private AnnotatedClass getAnnotatedClass(Element element) {
        // tipe . can not use chines  so  ....
        // get TypeElement  element is class's --->class  TypeElement typeElement = (TypeElement) element
        //  get TypeElement  element is method's ---> TypeElement typeElement = (TypeElement) element.getEnclosingElement();
        TypeElement typeElement = (TypeElement) element;
        String fullName = typeElement.getQualifiedName().toString();
        AnnotatedClass annotatedClass = mAnnotatedClassMap.get(fullName);
        if (annotatedClass == null) {
            annotatedClass = new AnnotatedClass(typeElement, mElementUtils, mMessager);
            mAnnotatedClassMap.put(fullName, annotatedClass);
        }
        return annotatedClass;
    }


    @Override
    public SourceVersion getSupportedSourceVersion() {
        return SourceVersion.latestSupported();
    }


    //這個(gè)個(gè)方法返回你要處理注解的類型
    @Override
    public Set<String> getSupportedAnnotationTypes() {
        Set<String> types = new LinkedHashSet<>();
        types.add(TypeUtil.ANNOTATION_PATH);
        return types;
    }

    private void error(String msg, Object... args) {
        mMessager.printMessage(Diagnostic.Kind.ERROR, String.format(msg, args));
    }

    private void log(String msg, Object... args) {
        mMessager.printMessage(Diagnostic.Kind.NOTE, String.format(msg, args));
    }
}

然后是生成java文件的輔助類

public class AnnotatedClass {

    private TypeElement mTypeElement;//activity  //fragmemt
    private Elements mElements;
    private Messager mMessager;//日志打印

    public AnnotatedClass(TypeElement typeElement, Elements elements, Messager messager) {
        mTypeElement = typeElement;
        mElements = elements;
        this.mMessager = messager;
    }


    public JavaFile generateActivityFile() {
        // build inject method
        MethodSpec.Builder injectMethod = MethodSpec.methodBuilder(TypeUtil.METHOD_NAME)
                .addModifiers(Modifier.PUBLIC)
                .addParameter(TypeName.get(mTypeElement.asType()), "activity", Modifier.FINAL);
        injectMethod.addStatement("android.widget.Toast.makeText" +"(activity, $S,android.widget.Toast.LENGTH_SHORT).show();", "from build");
        //generaClass
        TypeSpec injectClass = TypeSpec.classBuilder(mTypeElement.getSimpleName() + "$$InjectActivity")
                .addModifiers(Modifier.PUBLIC)
                .addMethod(injectMethod.build())
                .build();
        String packgeName = mElements.getPackageOf(mTypeElement).getQualifiedName().toString();
        return JavaFile.builder(packgeName, injectClass).build();
    }
    
}

這里就生成了一個(gè) 類名+$$的類，有一個(gè)方法叫inject參數(shù)是這個(gè)類本身，彈出一個(gè)toast。
最后一個(gè)就是一個(gè)字符常量類

public class TypeUtil {
    public static final String METHOD_NAME = "inject";
    public static final String ANNOTATION_PATH = "com.example.TestAnno";
}

好了lib工程完畢。然后是主工程引用
首先是在工程的gradle里面配置下apt

dependencies {
        classpath 'com.android.tools.build:gradle:2.3.2'
        classpath 'com.neenbedankt.gradle.plugins:android-apt:1.8'
    }

然后在app的gradle里面配置如下

apply plugin: 'com.neenbedankt.android-apt'
……
 compile project(':inject_annotation')
  apt project(':inject_comiler')

這樣就行了。
注意是apt 。為什么要新建立javalib。因?yàn)閖avalib不能在引用adnroidlib，。而注解處理器是javalib來完成，app里面，可以這引用這2個(gè)，apt如果換成complie 會(huì)提示錯(cuò)誤，但不會(huì)影響啥。
配置都好了，就是最后的使用了

總結(jié)構(gòu)

在我們MainActivity 上面加上注解，使用下

@TestAnno
public class MainActivity extends AppCompatActivity {
    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);
        InjectActivity.inject(this);//調(diào)用build生成的類
    }
}

最后一個(gè)類就是我們的使用這個(gè)在運(yùn)行時(shí)，調(diào)用生成build的類

public class InjectActivity {
    private static final ArrayMap<String, Object> injectMap = new ArrayMap<>();

    public static void inject(AppCompatActivity activity) {
        String className = activity.getClass().getName();
        try {
     
            Object inject = injectMap.get(className);

            if (inject == null) {
               //加載build生成的類
                Class<?> aClass = Class.forName(className + "$$InjectActivity");
                inject = aClass.newInstance();
                injectMap.put(className, inject);
            }
            //反射執(zhí)行方法
            Method m1 = inject.getClass().getDeclaredMethod("inject", activity.getClass());
            m1.invoke(inject, activity);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

這個(gè)類用了反射的方式查找了指定類執(zhí)行了指定方法。有一個(gè)map來緩存，不用每次都重新反射。
當(dāng)然你可可以不用反射，在創(chuàng)建build生成類的時(shí)候，實(shí)現(xiàn)一個(gè)接口，這里直接強(qiáng)轉(zhuǎn)為接口，直接調(diào)用接口的方法也可以。這里簡(jiǎn)單一些用的反射
通過一個(gè)注解，就自動(dòng)生成彈出toast的代碼看下自動(dòng) 生成的代碼

public class MainActivity$$InjectActivity {
  public void inject(final MainActivity activity) {
    android.widget.Toast.makeText(activity, "from build",android.widget.Toast.LENGTH_SHORT).show();;
  }
}

最后看下運(yùn)行效果。

這里寫圖片描述

demo下載：https://github.com/836154942/aptdemo

最后編輯于 ：
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點(diǎn)，簡(jiǎn)書系信息發(fā)布平臺(tái)，僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容