结构化在 Scala 运行时相等



scala 中是否有一些函数可以在运行时在结构上比较两个对象?由于字段数量众多,我宁愿不必手动创建 equals 函数。这是针对单元测试的,因此性能不是问题。我只需要一些能够反射性地查看字段并比较值的东西。

背景:我有一个案例类,它从一个覆盖等于的特征扩展而来,因此不会生成结构等号(squeryl KeyedEntity,相关:防止 Mixin 覆盖等于打破大小写类相等),我现在无法改变它。这是针对单元测试的,因此不会保留这两个实例。

我想出了这个解决方案,它遍历所有字段,对于私有字段,在scala中找到相应的方法。实际上,案例类具有可通过同名方法访问的私有成员。

implicit class ReallyEqual[T](a: T) {
  import java.lang.reflect.Modifier._
  def ==:==[U](b: U): Boolean = {
    val fieldsA = a.getClass.getDeclaredFields
    val methodsA = a.getClass.getDeclaredMethods
    val mapA = (methodsA.toList map (z => (z.getName(), z))).toMap
    val fieldsB = b.getClass.getDeclaredFields
    val methodsB = b.getClass.getDeclaredMethods
    val mapB = (methodsB.toList map (z => (z.getName(), z))).toMap
    fieldsA.length == fieldsB.length &&
    (true /: (fieldsA zip fieldsB)){ case (res, (aa, bb)) =>
      if(((aa.getModifiers & STATIC) != 0) || ((bb.getModifiers & STATIC) != 0)) { res  // ignore
      } else if(((aa.getModifiers & (PRIVATE | PROTECTED)) == 0) && ((bb.getModifiers & (PRIVATE | PROTECTED)) == 0)) {
        res && aa == bb && aa.get(a) ==:== bb.get(b)
      } else if((mapA contains aa.getName) && (mapB contains bb.getName)) {
        res && mapA(aa.getName).invoke(a) == mapB(aa.getName).invoke(b)
      } else res
    }
  }
}
trait EqualBad {
  override def equals(other: Any) = false
}
case class MyCaseClass(x: Int, y: List[Int]) extends EqualBad
MyCaseClass(1, List(1, 2)) ==:== MyCaseClass(1, List(1, 2)) // true
MyCaseClass(1, List(1, 2)) ==:== MyCaseClass(1, List(2, 2)) // false
MyCaseClass(1, List(1, 2)) ==:== MyCaseClass(2, List(1, 2)) // false
MyCaseClass(1, List(1, 2)) == MyCaseClass(1, List(1, 2))  // false

您甚至可以通过将方法ReallyEqual中的最后一个 == 更改为 ==:== 来使其更具递归性

注意:这不适用于整数,因为 Int 没有任何字段或方法。例如:1 ==:== 2 将返回 true。

AFAIK Scala 标准库不包含这样的函数。但是你有两个选择。

  1. 您编写一个宏,该宏生成一个方法来比较两个给定值的相等性
  2. 如果性能无关紧要,请使用 Java 反射。请参阅此答案:是否有 Java 反射实用程序可以对两个对象进行深入比较?

最新更新