Question

这可能对于在这里发布有点过于具体，但我正在尝试使用kotlinpoet生成这样的类：

class Query<out E: Model>(val onSuccess: (E) -> Unit, val onError: (Int, String) -> Unit = { i, m -> })

如何使用kotlinpoet创建该类型/构造函数参数？ docs确实列出了“Unit”类型和原始类型，因此它似乎是一种特殊情况。

Answer 1

这是一个产生所需输出的程序：

class Model

fun main(args: Array<String>) {
  val onSuccessType = LambdaTypeName.get(
      parameters = TypeVariableName(name = "E"),
      returnType = Unit::class.asTypeName())
  val onErrorType = LambdaTypeName.get(
      parameters = listOf(Int::class.asTypeName(), String::class.asTypeName()),
      returnType = Unit::class.asTypeName())
  val primaryConstructor = FunSpec.constructorBuilder()
      .addParameter(ParameterSpec.builder(name = "onSuccess", type = onSuccessType)
          .build())
      .addParameter(ParameterSpec.builder(name = "onError", type = onErrorType)
          .defaultValue("{ i, m -> }")
          .build())
      .build()

  val querySpec = TypeSpec.classBuilder("Query")
      .addTypeVariable(TypeVariableName(name = "out E", bounds = Model::class))
      .addProperty(PropertySpec.builder(name = "onSuccess", type = onSuccessType)
          .initializer("onSuccess")
          .build())
      .addProperty(PropertySpec.builder(name = "onError", type = onErrorType)
          .initializer("onError")
          .build())
      .primaryConstructor(primaryConstructor)
      .build()

  val file = KotlinFile.builder(packageName = "", fileName = "test")
      .addType(querySpec)
      .build()
  file.writeTo(System.out)
}

打印（不包括生成的导入）以下内容：

class Query<out E : Model>(val onSuccess: (E) -> Unit,
    val onError: (Int, String) -> Unit = { i, m -> })

我在这里以TypeVariableName攻击out E，因为当时似乎没有更好的解决方案。我还使用0.4.0-SNAPSHOT版本。

Answer 2

很容易就是使用LambdaTypeName类完成的。习惯了kotlin的功能类型风格，而不是java严格的功能接口，这有点令人困惑。这是我使用的：

val typeVariable = TypeVariableName.invoke("E")
    .withBounds(QueryData::class)
ParameterSpec.builder("onSuccess", 
    LambdaTypeName.get(null, listOf(typeVariable), UNIT)).build()

生成（当然是使用类构建器）：

class Query<E : QueryData> {
  constructor(onSuccess: (E) -> Unit) {
  }
}

如何使用kotlinpoet的单个类型参数生成Kotlin“Unit”类型的构造函数参数？

2 个答案: