使用kotlinpoet生成kotlin代碼的時候通常會遇到這樣一個問題,比如我希望生成這樣一段kotlin代碼:
var string: String? = null
string = bundle.get("test") as %T
當我直接把element.asType().asTypeName()
作為參數(shù)傳給%T
(假如這個元素element
是String
類型(或者是其他需要從Java類型映射到kotlin類型的數(shù)據(jù)類型)挤庇,最終會生成這樣一段代碼:
var string: String? = null
string = bundle.get("test") as java.lang.String
這時候編譯器就會報錯钞速,因為注解處理器是javac
提供的一個工具,它只認識Java代碼嫡秕,所以注解處理器中的所有元素element
都是Java中的數(shù)據(jù)類型渴语,因此element.asType().asTypeName()
生成的類型就是java.lang.String
,然后將其強制轉(zhuǎn)化成kotlin.String
當然就報錯了昆咽,平時不會出現(xiàn)這種錯誤是因為kotlinc
在編譯的時候幫我們完成了java.lang.String->kotlin.String的映射驾凶,所以沒有問題牙甫。
在kotlin的反射庫kotlin-reflect
源碼中找到了這么一個類JavaToKotlinClassMap
,大致閱讀了一下源碼發(fā)現(xiàn)這個類里面有我們想要的東西调违,最終我們?yōu)?code>Element定義一個擴展方法:
/**
* 獲取需要把java類型映射成kotlin類型的ClassName 如:java.lang.String 在kotlin中的類型為kotlin.String 如果是空則表示該類型無需進行映射
*/
private fun Element.javaToKotlinType(): ClassName? {
val className = JavaToKotlinClassMap.INSTANCE.mapJavaToKotlin(FqName(this.asType().asTypeName().toString()))?.asSingleFqName()?.asString()
return if (className == null) {
null
} else {
ClassName.bestGuess(className)
}
}
在需要獲取類型的地方這樣調(diào)用即可:
val className = element.javaToKotlinType() ?: element.asType().asTypeName()