Как динамически определить блок кода с параметрами - PullRequest
Новая
       18

Как динамически определить блок кода с параметрами

0 голосов
/ 01 июня 2018

Я застрял в этой функции, ниже приведен ожидаемый код, который будет сгенерирован, и общее количество параметров не является фиксированным числом, может быть 2 или 3 или более.

экземпляр экземпляра:InstanceType = Instance (параметр1, параметр2)

это внутри одной функции, поэтому я знаю, что мне следует использовать .addCode (CodeBlock.of ("% L", PropertySpec))

НоЯ не нахожу способ определить блок кода с динамическими параметрами, которые необходимо передать. Любое предложение?

1 Ответ

0 голосов
/ 01 июня 2018

Есть два способа решить эту проблему.Во-первых, CodeBlock имеет Builder, что позволяет вам динамически его создавать.Вот пример: 

@Test fun manyParams() {
  val instanceType = ClassName("", "InstanceType")
  val instance = ClassName("", "Instance")
  val params = listOf("param1", "param2")
  val prop = PropertySpec.builder("instance", instanceType)
      .initializer(CodeBlock.builder()
          .add("%T(", instance)
          .apply {
            params.forEachIndexed { index, param ->
              if (index > 0) add(",%W")
              add(param)
            }
          }
          .add(")")
          .build())
      .build()

  assertThat(prop.toString()).isEqualTo("""
    |val instance: InstanceType = Instance(param1, param2)
    |""".trimMargin())
}

Во-вторых, вы можете создать отдельный CodeBlock для каждого параметра и присоединиться к ним: 

@Test fun manyParams() {
  val instanceType = ClassName("", "InstanceType")
  val instance = ClassName("", "Instance")
  val params = listOf("param1", "param2")
  val paramCodeBlocks = params.map { CodeBlock.of(it) }
  val prop = PropertySpec.builder("instance", instanceType)
      .initializer("%T(%L)", instance, paramCodeBlocks.joinToCode(separator = ",%W"))
      .build()

  assertThat(prop.toString()).isEqualTo("""
    |val instance: InstanceType = Instance(param1, param2)
    |""".trimMargin())
}
...