我想使用scala宏进行某种代码生成。具体来说,我定义了一些Event
类型,这些事件将被推送到同一方法(此处为workImpl
),并返回特定类型的值。如果我仅调用workImpl
,则无法确定返回类型,因为它处理不同的输入并返回不同的结果。但是确定输入类型和输出类型之间的映射。 目标是使返回类型可用。
Scala宏很棒,也许会有所帮助。我想通过为每种事件类型添加事件返回类型注释来实现自动代码生成。应该生成的代码是code1
或 code2
。
object A {
private def workImpl(x: Any): Any = ???
@anno(X_RET)
case class X()
@anno(Y_RET)
case class Y()
@anno(Z_RET)
case class Z()
// code1: auto generate work1 and the implicits
def work1[T](x:T)(implicit ev: T=>M) = workImpl(x).asInstanceOf[M]
implicit val X_TO_X_RET: X=>X_RET = null
implicit val Y_TO_Y_RET: Y=>Y_RET = null
implicit val Z_TO_Z_RET: Z=>Z_RET = null
// code2: auto generate these methods
def work2(x:X) = workImpl(x).asInstanceOf[X_RET]
def work2(x:Y) = workImpl(x).asInstanceOf[Y_RET]
def work2(x:Z) = workImpl(x).asInstanceOf[Z_RET]
}
所以,这可能吗,如何实现呢?如果不可能,还有其他解决办法吗?
我终于找到了可行的解决方案。
object A {
private def workImpl(x: Any): Any = x match {
case e: X => XRet(0)
case e: Y => YRet(0)
case e: Z => ZRet(0)
}
trait Event
trait EventReturn[T] extends Event
case class X() extends EventReturn[XRet]
case class Y() extends EventReturn[YRet]
case class Z() extends EventReturn[ZRet]
case class XRet(x:Int)
case class YRet(y:Int)
case class ZRet(z:Int)
def work3[T](x: EventReturn[T]): T = workImpl(x).asInstanceOf[T]
def main(args: Array[String]): Unit = {
work3(X()).x
work3(Y()).y
work3(Z()).z
}
}