关于Scala案例类的思考

时间:2010-02-08 19:27:14

标签: reflection scala scala-2.8 case-class

我正在尝试编写一个特征(在Scala 2.8中),它可以混合到一个case类中,允许在运行时检查它的字段,以用于特定的调试目的。我想按照它们在源文件中声明的顺序返回它们,我想省略case类中的任何其他字段。例如:

trait CaseClassReflector extends Product {

  def getFields: List[(String, Any)] = {
    var fieldValueToName: Map[Any, String] = Map()
    for (field <- getClass.getDeclaredFields) {
      field.setAccessible(true)
      fieldValueToName += (field.get(this) -> field.getName) 
    }
    productIterator.toList map { value => fieldValueToName(value) -> value }
  }

}

case class Colour(red: Int, green: Int, blue: Int) extends CaseClassReflector {
  val other: Int = 42
}

scala> val c = Colour(234, 123, 23)
c: Colour = Colour(234,123,23)

scala> val fields = c.getFields    
fields: List[(String, Any)] = List((red,234), (green,123), (blue,23))

上述实现显然存在缺陷,因为它通过这些字段上的值的相等性来猜测字段在Product中的位置与其名称之间的关系,因此以下内容将不起作用:

Colour(0, 0, 0).getFields

有什么方法可以实现吗?

4 个答案:

答案 0 :(得分:10)

看看后备箱你会发现这个。听取评论,这是不支持的:但因为我也需要这些名字......

/** private[scala] so nobody gets the idea this is a supported interface.
 */
private[scala] def caseParamNames(path: String): Option[List[String]] = {
  val (outer, inner) = (path indexOf '$') match {
    case -1   => (path, "")
    case x    => (path take x, path drop (x + 1))
  }

  for {
    clazz <- getSystemLoader.tryToLoadClass[AnyRef](outer)
    ssig <- ScalaSigParser.parse(clazz)
  }
  yield {
    val f: PartialFunction[Symbol, List[String]] =
      if (inner.isEmpty) {
        case x: MethodSymbol if x.isCaseAccessor && (x.name endsWith " ") => List(x.name dropRight 1)
      }
      else {
        case x: ClassSymbol if x.name == inner  =>
          val xs = x.children filter (child => child.isCaseAccessor && (child.name endsWith " "))
          xs.toList map (_.name dropRight 1)
      }

    (ssig.symbols partialMap f).flatten toList
  }
}

答案 1 :(得分:9)

这是一个简短且有效的版本,基于上面的例子

  trait CaseClassReflector extends Product {
    def getFields = getClass.getDeclaredFields.map(field => {
      field setAccessible true
      field.getName -> field.get(this)
    })
  }

答案 2 :(得分:7)

在每个例子中,我都看到字段的顺序是相反的:getFields数组中的最后一项是case类中列出的第一项。如果您“很好地”使用案例类,那么您应该能够将productElement(n)映射到getDeclaredFields()( getDeclaredFields.length-n-1)

但这是相当危险的,因为我不知道规范中的任何内容都坚持它必须是这样的,并且如果你在case类中覆盖val,它甚至不会出现在getDeclaredFields中(它将出现在那个超类的字段中。

您可能会更改代码以假设事情是这样的,但检查具有该名称的getter方法和productIterator是否返回相同的值并抛出异常(如果它们没有)(这意味着您实际上并不知道什么对应什么)。

答案 3 :(得分:4)

您还可以使用解释器包中的ProductCompletion来获取案例类的属性名称和值:

import tools.nsc.interpreter.ProductCompletion

// get attribute names
new ProductCompletion(Colour(1, 2, 3)).caseNames
// returns: List(red, green, blue)

// get attribute values
new ProductCompletion(Colour(1, 2, 3)).caseFields

编辑:roland和virtualeyes的提示

必须包含属于scala-lang collectionscalap库。

感谢你的提示,roland和virtualeyes。