如何在KotlinPoet中使用FunSpec.overriding方法？

你提到的这个需求完全合理！KotlinPoet提供的FunSpec.overriding()方法就是为了简化这类重写方法的代码生成场景，不用手动重复设置方法名、OVERRIDE修饰符这些基础内容，直接基于已有方法的签名生成模板，让你只需要专注于方法体的实现。

完整简化示例代码 #

把你期望的写法整合到完整的代码生成流程里，是这样的：

FileSpec.builder("com.fleshgrinder", "KotlinPoet").apply {
    val className = ClassName("com.fleshgrinder", "KotlinPoet")
    addType(TypeSpec.classBuilder(className).apply {
        // 用overriding方法快速生成重写的toString()方法
        addFunction(FunSpec.overriding(Any::toString).apply {
            addStatement("""return "KotlinPoet"""")
        }.build())
    }.build())
}.build().writeTo(System.out)

生成结果 #

这段代码运行后，生成的Kotlin类和你之前繁琐写法的结果完全一致：

class KotlinPoet {
  override fun toString() = "KotlinPoet"
}

为什么这个方法好用？ #

FunSpec.overriding()会自动帮你处理以下内容：

• 自动设置方法名（比如这里的toString）
• 自动添加override修饰符
• 自动匹配原方法的返回类型和参数列表（如果重写的方法有参数，比如equals，也能完美适配）

举个重写equals的例子，同样简洁：

FunSpec.overriding(Any::equals).apply {
    addStatement("""return this === other""")
}.build()

这样一来，你就不用再写重复的模板代码，把精力放在方法的核心实现上就好啦！

内容的提问来源于stack exchange，提问作者Egor