scala - 如何注入准引号数组

Question

我有一个名为 quasi quotes 的数组definitions，我想将其注入到 quasi quotetree中。我该怎么做呢？

private def generateDaoComponent(files: Array[File]) = {
    val file = createNewFile(compDirectory)

    val definitions = files.map(f => {
      val daoName = f.getName.replace(".java", "")
      val daoType = TypeName(daoName)
      val daoTerm = TermName(daoName)

      q"""def $daoTerm = getValueOrInstantiate($daoName, () => new $daoType(configuration))


       """
    })

    val tree = q"""
        package database.dao {
          import org.jooq.SQLDialect
          import org.jooq.impl.DefaultConfiguration
          import utility.StrongHashMap

          trait $componentType extends StrongHashMap[String, Dao] {
            this: DaoManager =>

            private lazy val configuration = new DefaultConfiguration().set(connection).set(SQLDialect.POSTGRES_9_4)

            ${definitions.foreach(f => q"${f}")}
          }
        }"""

    writeToFile(file, tree)
  }

Answer 1

这是一些疯狂的深夜编码，但由于这个网站，我找到了它

Scala 代码生成的 3 种方法

我注意到当它将$params数组传递给准引号时，它..在类构造函数前面使用了两个，如下所示：

    val params = schema.fields.map { field =>
      val fieldName = newTermName(field.name)
      val fieldType = newTypeName(field.valueType.fullName)
      q"val $fieldName: $fieldType"
    }

    val json = TypeSchema.toJson(schema)

    // rewrite the class definition
    c.Expr(
      q"""
        case class $className(..$params) {

          def schema = ${json}

        }
      """
    )

我在问题中发布的代码有两个步骤可以使其正常工作。

1) 转换$definitions.toList为列表

..2）在前面加上两个

所以最终的代码是这样的：

    val definitions = files.map(f => {
      val daoName = f.getName.replace(".java", "")
      val daoType = TypeName(daoName)
      val daoTerm = TermName(new StringBuilder("get").append(daoName).toString())

      q"""def $daoTerm = getValueOrInstantiate($daoName, () => new $daoType(configuration))"""
    }).toList <--- HERE!

    val tree = q"""
        package database.dao {
          import org.jooq.SQLDialect
          import org.jooq.impl.DefaultConfiguration
          import utility.StrongHashMap

          trait $componentType extends StrongHashMap[String, Dao] {
            this: DaoManager =>

            private lazy val configuration = new DefaultConfiguration().set(connection).set(SQLDialect.POSTGRES_9_4)

            ..$definitions <-- AND HERE!

          }
        }"""