基于AST的组件化自动插桩方案

本文将带你实现一个一百多行代码实现的自动化插桩方案,解决组件化子模块的初始化和路由器的自动注册,支持多种类型的插桩、支持前插后插、支持插入代码的优先级设置。我们将使用编辑器的API来操作AST实现代码插桩,而非重量级的编译器(Aspectj)或者Gradle插件(ASM/Javassisit)。

第一步,定义AST注解:

@Retention(RetentionPolicy.SOURCE)
@Target(ElementType.METHOD)
public @interface AST {

    /**
     * 类型  0插入点  1 需要插入的原始代码块
     */
    int type();

    /**
     * 插桩的类型的id,支持多种需求的代码插桩
     */
    int value();

    /**
     * 类型  为0 时   0表示前插   1表示后插
     * 类型  为1 时   level表示优先级  0在最前  值越大  插入时排序的优先级越低
     */
    int level();

    /**
     * 类型  0插入点  1 需要插入的代码块
     */
    interface TYPE {
        int TARGET = 0;
        int SOURCE = 1;
    }

    /**
     * 插入类型的id
     */
    interface ID {
        int MODULE_INIT = 0x0001;//模块初始化的插桩
        int ROUTER_INIT = 0x0002;//路由注册的插桩
    }

    /**
     * 类型  为0 时   0表示前插   1表示后插
     */
    interface LEVEL {
        int BEFORE = 0;
        int AFTER = 1;
    }
}

第二步,实现注解处理器

    private Trees trees;
    private HashMap mTargets = new HashMap<>();
    private HashMap> mSources = new HashMap<>();

    @Override
    public synchronized void init(ProcessingEnvironment env) {
        super.init(env);
        trees = Trees.instance(env);
    }

    @Override
    public boolean process(Set set, RoundEnvironment roundEnv) {
        if (!roundEnv.processingOver()) {
            roundEnv.getRootElements().stream()
                    .filter(it -> it.getKind() == ElementKind.CLASS)
                    .forEach(it -> ((JCTree) trees.getTree(it)).accept(new AOPTreeTranslator()));
        } else {
            mTargets.entrySet().stream()
                    .filter(it -> it.getValue().level == AST.LEVEL.AFTER)
                    .forEach(it -> mSources.get(it.getKey()).forEach(node
                            -> it.getValue().med.body.stats
                            = it.getValue().med.body.stats.appendList(node.med.body.stats)));
        }
        return false;
    }

解读:
1、使用mSources存储插桩类型对应的插入代码块的优先级队列
2、使用mTargets存储插桩类型对应的插入点代码块
3、根据注解收集器处理被注解标注的源代码
4、把所有需要插入的代码按照优先级插入到相应的插入点

注意点:
javac的List不同于常见的list,操作方式完全颠覆你的习惯。编译器用了它自己的数据类型来实现List,而不是使用java集合框架(Java Collection Framework)。
并且有许多静态的方法,可以很方便的创建List:
l List.nil()
l List.of(A)
同样,非传统的命名方式也带来了更漂亮的代码
不像传统java中用的代码:
List list = new List();
list.add(a);
list.add(b);
list.add(c);
而现在只需要写:
List.of(a, b, c);
List.from(list);
同时改变list使用:
list=list.appendList(newList);
而list.appendList或者list.append 是不会影响原来list的值,是不是很有趣?

第三步、实现TreeTranslator:

 private class AOPTreeTranslator extends TreeTranslator {
        @Override
        public void visitMethodDef(JCTree.JCMethodDecl jcMethod) {
            super.visitMethodDef(jcMethod);
            if (jcMethod.getModifiers().annotations.toString().contains(AST.class.getSimpleName())) {
                String type = getValue(jcMethod, 0);
                String id = getValue(jcMethod, 1);
                String levelType = getValue(jcMethod, 2);
                if (type.equals("AST.TYPE.SOURCE")) {
                    int level = Integer.parseInt(levelType);
                    if (mSources.get(id) != null) {
                        mSources.get(id).add(new CodeItem(level, jcMethod));
                    } else {
                        PriorityQueue list = new PriorityQueue<>(Comparator.comparingInt(p -> p.level));
                        list.add(new CodeItem(level, jcMethod));
                        mSources.put(id, list);
                    }
                } else {
                    int level = levelType.equals("AST.LEVEL.BEFORE") ? AST.LEVEL.BEFORE : AST.LEVEL.AFTER;
                    mTargets.put(id, new CodeItem(level, jcMethod));
                }
            }
        }
    }

这里就没什么好说的了,直接扫描AST填充mTargets和mSources而已。

OK,你没看错,以上基本就是所有代码了。

下面进行具体的案例讲解:

1、gradle配置:

重头戏来了:由于annotationProcessor无法扫描子组件里被注解的代码,我们可以采用gradle的配置去曲线救国:

子组件单独建立一个ast文件夹,仅仅在单独编译的时候属于他自己

sourceSets {
        main {
            //默认的作为application运行时Manifest文件路径
            if (isDebug.toBoolean()) {
                manifest.srcFile 'src/main/debug/AndroidManifest.xml'
                java {
                    srcDirs = ['src/main/java',
                               'src/main/ast']
                }
            } else {
                manifest.srcFile 'src/main/AndroidManifest.xml'
                java {
                    exclude 'ast/**'
                }
            }
        }
    }

主模块也是仅仅在最终打包时才包涵子模块的ast文件夹,实现代码完全隔离下的特殊情况特殊编译:

 sourceSets {
            main {
                if (isDebug.toBoolean()) {
                    java {
                        srcDirs = ['src/main/java']
                    }
                }else{
                    java {
                        srcDirs = ['src/main/java',
                                   '../module_a/src/main/ast',
                                   '../module_b/src/main/ast',
                                   '../module_service/src/main/ast']
                    }
                }
            }
        }

这样的话,子组件需要集中注册的代码,放在ast文件夹下即可,子组件单独运行时,赋予它意义,最终打包时,赋予它功能。

2、子模块配置

例如:a模块的ast:

public class ModuleA {

    @AST(type = AST.TYPE.SOURCE, value = AST.ID.MODULE_INIT, level = 1)//插入源,插桩类型是模块初始化,优先级是1
    public static void autoInitModule() {
        (new com.easy.moduler.module_a.AModule()).afterConnected();
    }

    @AST(type = AST.TYPE.SOURCE, value = AST.ID.ROUTER_INIT, level = 1)//插入源,插桩类型是路由器自动注册,优先级是1
    public static void autoInitRouter() {
        com.easy.moduler.lib.router.Router.addRouterRule(
                new com.easy.moduler.apt.processor.module_a_AutoRouterRuleCreator());
    }
}

1、类名用全路径,IDE是不会自动帮你倒入的
2、注解注明,你的插桩类型,插桩id,插桩优先级

3、主模块配置:

public class App extends Application {

    public void onCreate() {
        super.onCreate();
        initModule();
        initRouter();
    }

    @AST(type = AST.TYPE.TARGET, value = AST.ID.MODULE_INIT, level = AST.LEVEL.BEFORE)//插桩目标点,模块初始化的插桩,插桩方式为前插
    private void initModule() {
        Log.e("TAG", "Module初始化本地逻辑");
    }

    @AST(type = AST.TYPE.TARGET, value = AST.ID.ROUTER_INIT, level = AST.LEVEL.AFTER)//插桩目标点,路由初始化的插桩,插桩方式为后插
    private void initRouter() {
        Log.e("TAG", "Router初始化本地逻辑");
    }
}

为了表明插入的类型(前插或者后插),在方法体内放了一个log做参考。

最终结果:

基于AST的组件化自动插桩方案_第1张图片
插桩结果

如图所示,子模块的初始化和路由注册都被自动按照优先级插桩到相应的位置上。

即实现了代码的完全隔离,又实现了子组件的自动注册,而且没用反射,没用编译器,没用插件,使用编辑器的API实现了代码的自动插桩。

总结:

因为AST是IDE编辑器级别,Aspectj是java编译器级别,ASM/javassisit是字节码处理器插件级别。AST使用编辑器的API,还有注解过滤带来的效率提升,而不是把所有代码都扫描一遍。如果插入的代码逻辑有错误,AST在编译之前就会直接报错,而其他由于处理得是编译后的字节码,所以只能在运行期间才会崩溃。有了AST的错误前置,再也不用担心有人在代码里下毒了。

基于AST的组件化自动插桩方案_第2张图片
你TMD在代码里下毒

你还在为Aspectj难懂的切片语法烦恼吗,你还在为ASM晦涩的字节码操作而纠结吗,你还正在为javassisit动不动就无法编译的占用而无奈吗,试试AST吧,一切都是一个清爽的开始。

老司机赶紧进群开车: 555343041

例子十分简单,直接上源码:

@AutoService(Processor.class)//自动生成 javax.annotation.processing.IProcessor 文件
@SupportedSourceVersion(SourceVersion.RELEASE_8)//java版本支持
@SupportedAnnotationTypes({"com.easy.moduler.annotation.AST"})//注意替换成你自己的注解名
public class ASTProcessor extends AbstractProcessor {
   private Trees trees;
   private HashMap> mCodes = new HashMap<>();
   private HashMap mTargets = new HashMap<>();
   private HashMap> mSources = new HashMap<>();

   @Override
   public synchronized void init(ProcessingEnvironment env) {
       super.init(env);
       trees = Trees.instance(env);
   }

   @Override
   public boolean process(Set set, RoundEnvironment roundEnv) {
       if (!roundEnv.processingOver()) {
           roundEnv.getRootElements().stream()
                   .filter(it -> it.getKind() == ElementKind.CLASS)
                   .forEach(it -> ((JCTree) trees.getTree(it)).accept(new AOPTreeTranslator()));
       } else {
           mTargets.entrySet().stream()
                   .filter(it -> it.getValue().level == AST.LEVEL.AFTER)
                   .forEach(it -> mSources.get(it.getKey()).forEach(node
                           -> it.getValue().med.body.stats
                           = it.getValue().med.body.stats.appendList(node.med.body.stats)));

           mTargets.entrySet().stream()
                   .filter(it -> it.getValue().level == AST.LEVEL.BEFORE)
                   .forEach(it -> mSources.get(it.getKey())
                           .forEach(node -> {
                               List list;
                               if (mCodes.get(it.getKey()) != null)
                                   list = mCodes.get(it.getKey()).appendList(node.med.body.stats);
                               else list = List.from(node.med.body.stats);
                               mCodes.put(it.getKey(), list);
                           }));
           mCodes.forEach((key, value) ->
                   mTargets.get(key).med.body.stats = mTargets.get(key).med.body.stats.prependList(value));
       }
       return false;
   }

   class CodeItem {
       int level;//优先级
       JCTree.JCMethodDecl med;//方法体

       CodeItem(int levelType, JCTree.JCMethodDecl jcMethodDecl) {
           this.level = levelType;
           this.med = jcMethodDecl;
       }
   }

   private class AOPTreeTranslator extends TreeTranslator {
       @Override
       public void visitMethodDef(JCTree.JCMethodDecl jcMethod) {
           super.visitMethodDef(jcMethod);
           if (jcMethod.getModifiers().annotations.toString().contains(AST.class.getSimpleName())) {
               String type = getValue(jcMethod, 0);
               String id = getValue(jcMethod, 1);
               String levelType = getValue(jcMethod, 2);
               if (type.equals("AST.TYPE.SOURCE")) {
                   int level = Integer.parseInt(levelType);
                   if (mSources.get(id) != null) {
                       mSources.get(id).add(new CodeItem(level, jcMethod));
                   } else {
                       PriorityQueue list = new PriorityQueue<>(Comparator.comparingInt(p -> p.level));
                       list.add(new CodeItem(level, jcMethod));
                       mSources.put(id, list);
                   }
               } else {
                   int level = levelType.equals("AST.LEVEL.BEFORE") ? AST.LEVEL.BEFORE : AST.LEVEL.AFTER;
                   mTargets.put(id, new CodeItem(level, jcMethod));
               }
           }
       }
   }

   private String getValue(JCTree.JCMethodDecl jcMethod, int index) {
       return jcMethod.getModifiers().annotations.get(0).getArguments().get(index).toString().split("=")[1].trim();
   }
}

你可能感兴趣的:(基于AST的组件化自动插桩方案)