使用kotlinpoet生成kotlin代碼的時(shí)候通常會(huì)遇到這樣一個(gè)問題,比如我希望生成這樣一段kotlin代碼:
var string: String? = null
string = bundle.get("test") as %T
當(dāng)我直接把element.asType().asTypeName()作為參數(shù)傳給%T(假如這個(gè)元素element是String類型(或者是其他需要從Java類型映射到kotlin類型的數(shù)據(jù)類型),最終會(huì)生成這樣一段代碼:
var string: String? = null
string = bundle.get("test") as java.lang.String
這時(shí)候編譯器就會(huì)報(bào)錯(cuò),因?yàn)樽⒔馓幚砥魇?code>javac提供的一個(gè)工具,它只認(rèn)識(shí)Java代碼,所以注解處理器中的所有元素element都是Java中的數(shù)據(jù)類型,因此element.asType().asTypeName()生成的類型就是java.lang.String,然后將其強(qiáng)制轉(zhuǎn)化成kotlin.String當(dāng)然就報(bào)錯(cuò)了,平時(shí)不會(huì)出現(xiàn)這種錯(cuò)誤是因?yàn)?code>kotlinc在編譯的時(shí)候幫我們完成了java.lang.String->kotlin.String的映射,所以沒有問題。
在kotlin的反射庫(kù)kotlin-reflect源碼中找到了這么一個(gè)類JavaToKotlinClassMap,大致閱讀了一下源碼發(fā)現(xiàn)這個(gè)類里面有我們想要的東西,最終我們?yōu)?code>Element定義一個(gè)擴(kuò)展方法:
/**
* 獲取需要把java類型映射成kotlin類型的ClassName 如:java.lang.String 在kotlin中的類型為kotlin.String 如果是空則表示該類型無(wú)需進(jìn)行映射
*/
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()