38

有一个特质

trait Persisted {
  def id: Long
}

如何实现一个方法,该方法接受任何案例类的实例并返回其副本并混合特征?

该方法的签名如下所示:

def toPersisted[T](instance: T, id: Long): T with Persisted
4

5 回答 5

32

这可以通过宏来完成(自 2.10.0-M3 以来,它正式成为 Scala 的一部分)。这是您要查找的内容的要点示例

1)我的宏生成一个本地类,它继承自提供的案例类和持久化,就像new T with Persisted会做的那样。然后它缓存其参数(以防止多次评估)并创建已创建类的实例。

2)我怎么知道要生成什么树?我有一个简单的应用程序 parse.exe,它打印解析输入代码产生的 AST。所以我只是调用parse class Person$Persisted1(first: String, last: String) extends Person(first, last) with Persisted,记下输出并在我的宏中复制它。parse.exe 是scalac -Xprint:parser -Yshow-trees -Ystop-after:parser. 有多种探索 AST 的方法,请阅读“Scala 2.10 中的元编程”了解更多信息。

-Ymacro-debug-lite3) 如果您将宏扩展作为参数提供给 scalac,则可以对宏扩展进行完整性检查。在这种情况下,所有扩展都将被打印出来,您将能够更快地检测代码生成错误。

编辑。更新了 2.10.0-M7 的示例

于 2012-04-30T16:34:45.170 回答
9

使用 vanilla scala 无法实现您想要的。问题是混合如下:

scala> class Foo
defined class Foo

scala> trait Bar
defined trait Bar

scala> val fooWithBar = new Foo with Bar
fooWithBar: Foo with Bar = $anon$1@10ef717

创建一个Foo with Bar混合,但它不是在运行时完成的。编译器只是生成一个新的匿名类:

scala> fooWithBar.getClass
res3: java.lang.Class[_ <: Foo] = class $anon$1

请参阅Scala 中的动态混合 - 有可能吗?了解更多信息。

于 2012-04-29T16:23:55.200 回答
4

您尝试执行的操作称为记录连接,这是 Scala 的类型系统不支持的。(Fwiw,存在提供此功能的类型系统,例如thisthis 。)

我认为类型类可能适合您的用例,但我无法确定,因为该问题没有提供有关您要解决的问题的足够信息。

于 2012-04-29T17:05:29.550 回答
4

更新

您可以找到一个最新的工作解决方案,它利用 Scala 2.10.0-RC1 的 Toolboxes API 作为SORM项目的一部分。


以下解决方案基于 Scala 2.10.0-M3 反射 API 和 Scala Interpreter。它动态创建和缓存从原始案例类继承的类,并混合了特征。由于最大程度地缓存,该解决方案应该为每个原始案例类动态创建一个类,并在以后重用它。

由于新的反射 API 没有太多公开,也不是很稳定,也没有关于它的教程,但是这个解决方案可能涉及一些愚蠢的重复动作和怪癖。

以下代码使用 Scala 2.10.0-M3 进行了测试。

1. 持久化.scala

要混入的特征。请注意,由于程序的更新,我对其进行了一些更改

trait Persisted {
  def key: String
}

2. PersistedEnabler.scala

实际的工作对象

import tools.nsc.interpreter.IMain
import tools.nsc._
import reflect.mirror._

object PersistedEnabler {

  def toPersisted[T <: AnyRef](instance: T, key: String)
                              (implicit instanceTag: TypeTag[T]): T with Persisted = {
    val args = {
      val valuesMap = propertyValuesMap(instance)
      key ::
        methodParams(constructors(instanceTag.tpe).head.typeSignature)
          .map(_.name.decoded.trim)
          .map(valuesMap(_))
    }

    persistedClass(instanceTag)
      .getConstructors.head
      .newInstance(args.asInstanceOf[List[Object]]: _*)
      .asInstanceOf[T with Persisted]
  }


  private val persistedClassCache =
    collection.mutable.Map[TypeTag[_], Class[_]]()

  private def persistedClass[T](tag: TypeTag[T]): Class[T with Persisted] = {
    if (persistedClassCache.contains(tag))
      persistedClassCache(tag).asInstanceOf[Class[T with Persisted]]
    else {
      val name = generateName()

      val code = {
        val sourceParams =
          methodParams(constructors(tag.tpe).head.typeSignature)

        val newParamsList = {
          def paramDeclaration(s: Symbol): String =
            s.name.decoded + ": " + s.typeSignature.toString
          "val key: String" :: sourceParams.map(paramDeclaration) mkString ", "
        }
        val sourceParamsList =
          sourceParams.map(_.name.decoded).mkString(", ")

        val copyMethodParamsList =
          sourceParams.map(s => s.name.decoded + ": " + s.typeSignature.toString + " = " + s.name.decoded).mkString(", ")

        val copyInstantiationParamsList =
          "key" :: sourceParams.map(_.name.decoded) mkString ", "

        """
        class """ + name + """(""" + newParamsList + """)
          extends """ + tag.sym.fullName + """(""" + sourceParamsList + """)
          with """ + typeTag[Persisted].sym.fullName + """ {
            override def copy(""" + copyMethodParamsList + """) =
              new """ + name + """(""" + copyInstantiationParamsList + """)
          }
        """
      }

      interpreter.compileString(code)
      val c =
        interpreter.classLoader.findClass(name)
          .asInstanceOf[Class[T with Persisted]]

      interpreter.reset()

      persistedClassCache(tag) = c

      c
    }
  }

  private lazy val interpreter = {
    val settings = new Settings()
    settings.usejavacp.value = true
    new IMain(settings, new NewLinePrintWriter(new ConsoleWriter, true))
  }


  private var generateNameCounter = 0l

  private def generateName() = synchronized {
    generateNameCounter += 1
    "PersistedAnonymous" + generateNameCounter.toString
  }


  // REFLECTION HELPERS

  private def propertyNames(t: Type) =
    t.members.filter(m => !m.isMethod && m.isTerm).map(_.name.decoded.trim)

  private def propertyValuesMap[T <: AnyRef](instance: T) = {
    val t = typeOfInstance(instance)

    propertyNames(t)
      .map(n => n -> invoke(instance, t.member(newTermName(n)))())
      .toMap
  }

  private type MethodType = {def params: List[Symbol]; def resultType: Type}

  private def methodParams(t: Type): List[Symbol] =
    t.asInstanceOf[MethodType].params

  private def methodResultType(t: Type): Type =
    t.asInstanceOf[MethodType].resultType

  private def constructors(t: Type): Iterable[Symbol] =
    t.members.filter(_.kind == "constructor")

  private def fullyQualifiedName(s: Symbol): String = {
    def symbolsTree(s: Symbol): List[Symbol] =
      if (s.enclosingTopLevelClass != s)
        s :: symbolsTree(s.enclosingTopLevelClass)
      else if (s.enclosingPackageClass != s)
        s :: symbolsTree(s.enclosingPackageClass)
      else
        Nil

    symbolsTree(s)
      .reverseMap(_.name.decoded)
      .drop(1)
      .mkString(".")
  }

}

3.沙盒.scala

测试应用

import PersistedEnabler._

object Sandbox extends App {
  case class Artist(name: String, genres: Set[Genre])
  case class Genre(name: String)

  val artist = Artist("Nirvana", Set(Genre("rock"), Genre("grunge")))

  val persisted = toPersisted(artist, "some-key")

  assert(persisted.isInstanceOf[Persisted])
  assert(persisted.isInstanceOf[Artist])
  assert(persisted.key == "some-key")
  assert(persisted.name == "Nirvana")
  assert(persisted == artist)  //  an interesting and useful effect

  val copy = persisted.copy(name = "Puddle of Mudd")

  assert(copy.isInstanceOf[Persisted])
  assert(copy.isInstanceOf[Artist])
  //  the only problem: compiler thinks that `copy` does not implement `Persisted`, so to access `key` we have to specify it manually:
  assert(copy.asInstanceOf[Artist with Persisted].key == "some-key")
  assert(copy.name == "Puddle of Mudd")
  assert(copy != persisted)

}
于 2012-05-01T12:24:52.893 回答
-1

虽然不可能在创建对象后组合它,但您可以使用类型别名和定义结构进行非常广泛的测试来确定对象是否具有特定组合:

  type Persisted = { def id: Long }

  class Person {
    def id: Long = 5
    def name = "dude"
  }

  def persist(obj: Persisted) = {
    obj.id
  }

  persist(new Person)

任何带有 a 的对象def id:Long都符合 Persisted 的条件。

通过隐式转换可以实现我认为您正在尝试做的事情:

  object Persistable {
    type Compatible = { def id: Long }
    implicit def obj2persistable(obj: Compatible) = new Persistable(obj)
  }
  class Persistable(val obj: Persistable.Compatible) {
    def persist() = println("Persisting: " + obj.id)
  }

  import Persistable.obj2persistable
  new Person().persist()
于 2012-04-29T16:40:12.040 回答