1

問題:

注釈付きのクラスを、scala マクロを使用して別のクラスのサブクラスにしたい:

フィールドのラッパー:

class Field(fieldType: DbModelFieldType, fieldName: String) 

抽象クラス (すべての注釈付きクラスの基本クラス):

abstract class DatabaseModel {
  def fields: Seq[Fields]
}

ケースクラスがあります:

Model(num: Int, sym: Char, descr: String)

そして、そのクラスに注釈を付ける場合@GetFromDB

@GetFromDB
Model(num: Int, sym: Char, descr: String)
case class Model(num: Int, sym: Char, descr: String) extends DatabaseModel {
   override def fields: Seq[Fields] = 
       Seq(Field(IntFieldType(), "num"),
           Field(CharFieldType(), "sym"),
           Field(StringFieldType(), "descr")
          ) 
}

私の望ましい結果は次のようになります。

val m: DatabaseModel = Model(1, 'A', "First Name")

私は同様の質問を見てきました

メソッドを使用してケース クラスのコンパニオン オブジェクトを生成する (フィールド = メソッド)

では、そのソリューションを拡張して目的の結果を得るにはどうすればよいでしょうか?

import scala.annotation.{StaticAnnotation, compileTimeOnly}
import scala.language.experimental.macros
import scala.reflect.macros.whitebox

object Macros {
  @compileTimeOnly("enable macro paradise")
  class GenerateCompanionWithFields extends StaticAnnotation {
    def macroTransform(annottees: Any*): Any = macro Macro.impl
  }

  object Macro {
    def impl(c: whitebox.Context)(annottees: c.Tree*): c.Tree = {
      import c.universe._
      annottees match {
        case (cls @ q"$_ class $tpname[..$_] $_(...$paramss) extends { ..$_ } with ..$_ { $_ => ..$_ }") :: Nil =>

          val newMethods = paramss.flatten.map {
            case q"$_ val $tname: $tpt = $_" =>
              q"def $tname(): String = ${tpt.toString}"
          }

          q"""
             $cls

             object ${tpname.toTermName} {
               ..$newMethods
             }
           """
      }
    }
  }
}
4

1 に答える 1