使用kotlinpoet生成kotlin代碼數(shù)據(jù)類型不正確的問題

使用kotlinpoet生成kotlin代碼的時(shí)候通常會(huì)遇到這樣一個(gè)問題,比如我希望生成這樣一段kotlin代碼:

var string: String? = null
string = bundle.get("test") as %T

當(dāng)我直接把element.asType().asTypeName()作為參數(shù)傳給%T(假如這個(gè)元素elementString類型(或者是其他需要從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()
?著作權(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)容