Scala 2.10反射,如何从案例类中提取字段值,即从案例类中提取字段列表

时间:2013-04-18 09:22:40

标签: scala reflection scala-2.10 case-class

如何使用scala 2.10中的新反射模型从scala中的case类中提取字段值? 例如,使用下面的内容并没有提取字段方法

  def getMethods[T:TypeTag](t:T) =  typeOf[T].members.collect {
    case m:MethodSymbol => m
  }

我计划将它们泵入

  for {field <- fields} {
    currentMirror.reflect(caseClass).reflectField(field).get
  }

2 个答案:

答案 0 :(得分:42)

MethodSymbol有一个isCaseAccessor方法,可以让你做到这一点:

def getMethods[T: TypeTag] = typeOf[T].members.collect {
  case m: MethodSymbol if m.isCaseAccessor => m
}.toList

现在您可以写下以下内容:

scala> case class Person(name: String, age: Int)
defined class Person

scala> getMethods[Person]
res1: List[reflect.runtime.universe.MethodSymbol] = List(value age, value name)

你只得到你想要的方法符号。

如果你只想要实际的字段名称(不是value前缀),并且你想要它们的顺序是相同的,那么:

def getMethods[T: TypeTag]: List[String] =
  typeOf[T].members.sorted.collect {
    case m: MethodSymbol if m.isCaseAccessor => m.name.toString
  }

答案 1 :(得分:13)

如果你想获得更多的爱好者,你可以通过检查构造符号来使它们按顺序排列。即使所讨论的案例类类型定义了多个构造函数,此代码仍然有效。

  import scala.collection.immutable.ListMap
  import scala.reflect.runtime.universe._

  /**
    * Returns a map from formal parameter names to types, containing one
    * mapping for each constructor argument.  The resulting map (a ListMap)
    * preserves the order of the primary constructor's parameter list.
    */
  def caseClassParamsOf[T: TypeTag]: ListMap[String, Type] = {
    val tpe = typeOf[T]
    val constructorSymbol = tpe.decl(termNames.CONSTRUCTOR)
    val defaultConstructor =
      if (constructorSymbol.isMethod) constructorSymbol.asMethod
      else {
        val ctors = constructorSymbol.asTerm.alternatives
        ctors.map(_.asMethod).find(_.isPrimaryConstructor).get
      }

    ListMap[String, Type]() ++ defaultConstructor.paramLists.reduceLeft(_ ++ _).map {
      sym => sym.name.toString -> tpe.member(sym.name).asMethod.returnType
    }
  }