注解处理器的编写(Kotlin)

注解处理器的的作用

注解处理器可以在编译时对@Retention值为AnnotationRetention.BINARY(kotlin)\RetentionPolicy.CLASS(java)的注解进行处理 ,实现对注解所在代码的处理,例如打印日志、生成代码等。是实现面向切面编程(AOP)的一种方式。

编写方式

目标

目标

通过注解对person进行实例化,并赋值。

实现

Demo是一个安卓项目。


目录结构

标记3个部分是javaLibray,作用分别是:

  • annotation:注解所在类
  • complier:注解处理器所在类
  • runtime:用于直接调用的类

PS:app在依赖着三个的模块的时候,对complier不要使用api或implementation,而是使用kapt。

依赖

annotation部分

首先是在创建一个注解。

@Retention(value = AnnotationRetention.BINARY)
@Target(AnnotationTarget.FIELD)
annotation class Binding(val text: String) 

这段代码的主要有两个注解一个是@Retention,作用就是在什么情况下保留,选择在编译时保留。@Target为了方便选择目标是Field。传入参数的话是用于给person赋值使用。

complier部分

这部分主要就是注解处理的具体部分。

class BindProcessor : AbstractProcessor() {
    //用于打印消息
    private lateinit var messager: Messager
    //处理Element的工具类
    private lateinit var elementUtils: Elements
    //用于输出文件
    private lateinit var filer: Filer
    //Type工具类
    private lateinit var typeUtils: Types
    override fun process(
        annotations: MutableSet<out TypeElement>,
        roundEnv: RoundEnvironment
    ): Boolean {  
        //获取到被binding标记的元素
        roundEnv.getElementsAnnotatedWith(Binding::class.java).filter { it.kind.isField }.forEach {
            val variableElement = it as VariableElement
            val packageOf = elementUtils.getPackageOf(variableElement)
            val annotation = variableElement.getAnnotation(Binding::class.java)
            var element: Element = variableElement
            while (element.kind != ElementKind.CLASS) {
                element = element.enclosingElement
            }
            kotlin.runCatching {
                CodeProcess(
                    variableElement,
                    packageOf,
                    element as TypeElement,
                    annotation.text
                ).createCode(filer)
            }.onFailure { e ->
                messager.printMessage(Diagnostic.Kind.WARNING, e.localizedMessage)
            }
        }

        return true

    }

    override fun init(processingEnv: ProcessingEnvironment) {
        super.init(processingEnv)
        typeUtils = processingEnv.typeUtils
        elementUtils = processingEnv.elementUtils
        filer = processingEnv.filer
        messager = processingEnv.messager

    }

    //标记支持的版本
    override fun getSupportedSourceVersion(): SourceVersion {
        return SourceVersion.latestSupported()
    }
    
     //添加支持的注解有哪些
    override fun getSupportedAnnotationTypes(): MutableSet<String> {
        val annotations: MutableSet<String> = LinkedHashSet()
        annotations.add(Binding::class.java.canonicalName)
        return annotations
    }

}

代码中可以看到主要重写的4个方法

  • process
    为对注解进行处理的核心方法
  • init
    在调用注解处理器时会执行的方法
  • getSupportedSourceVersion
    用于确定支持的java版本
  • getSupportedAnnotationTypes

process的主要过程就是
1.获取注解
2.对注解进行分类,获取注解标记的对应属性,类,方法等。
3.对获取到的信息进行处理,就本Demo而言,就是通过filer输出用于创建的类。代码通过CodeProcess类来处理信息

class CodeProcess(
    val element: VariableElement,
    pack: PackageElement,
    type: TypeElement,
    var text: String
) {
    var name: String = element.simpleName.toString()
    var packNameString: String = pack.toString()
    var codeClazzName = type.simpleName.toString() + "Binding"
    fun createCode(filer: Filer) {
        //创建方法
        val methodBuild = MethodSpec.methodBuilder("create")
            .addModifiers(Modifier.PUBLIC, Modifier.STATIC)   //设置为公开的静态方法
            .returns(ClassName.get(element.asType()))    //输入返回值类型
            .addStatement(" return new \$T(\$S)", ClassName.get(element.asType()), text)  //加入代码语句
            .build()
        //创建class
        val typeBuild = TypeSpec.classBuilder(codeClazzName)
            .addModifiers(Modifier.PUBLIC)
            .addMethod(methodBuild)
            .build()
        //输出java文件
        JavaFile.builder(packNameString, typeBuild)
            .build()
            .writeTo(filer)

    }

    override fun toString(): String {
        return "CodeProcess(element=$element, text='$text', name='$name', packNameString='$packNameString', codeClazzName='$codeClazzName')"
    }
}

代码创建java文件,使用的时JavaPoet,具体的使用方法可以查看GitHub。其中$T、$S分别是类型和支付字符串的替换占位符。
在上面代码完成以后,需要把注解处理器注册到编译器中。方式是在创建 resources文件。并在内创建META-INFservicejavax.annotation.processing.Processor文件。然后在javax.annotation.processing.Processor内添加,注解处理器的完整名称(包名+类名)。

注册注解处理器

注册内容

runtime部分

这一部分的主要目的就是提供一个对外的统一调用的方法。对于本Demo来说,就是他通过反射对注解标注的属性进行实例化和赋值。

class BindingData {
    companion object {

        @JvmStatic
        fun bind(any: Any) {
            val bindingName = any.javaClass.name + "Binding"
            print(bindingName)
            val clazz = Class.forName(bindingName)
            val invoke = clazz.getMethod("create").invoke(null)//调用静态方法
            val fields = any.javaClass.fields
            fields.filter { it.type == invoke.javaClass }.forEach {
                it.isAccessible = true
                it.set(any, invoke)//对属性进行赋值
            }
        }
    }
}

结果

class MainActivity : AppCompatActivity() {
    @Binding("张三")
    lateinit var person: Person
    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        setContentView(R.layout.activity_main)
        BindingData.bind(this)
        val tv = findViewById<TextView>(R.id.tv)
        tv.setOnClickListener {
            tv.text = person.name
        }
    }
}

通过这一段文字就可以点击获取personname

注解处理器Demo地址

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,558评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,002评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,036评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,024评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,144评论 6 385
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,255评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,295评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,068评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,478评论 1 305
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,789评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,965评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,649评论 4 336
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,267评论 3 318
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,982评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,223评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,800评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,847评论 2 351

推荐阅读更多精彩内容