Scala使用反射检查至少一个案例类字段值是否为非空

时间:2018-08-30 23:07:09

标签: scala validation reflection optional case-class

我有一个要求,对于具有大约30个以上的给定案例类,Option [T]字段必须至少具有1个字段nonEmpty才有效。 我没有单独检查每个字段,而是选择了以通用方式通过反射检查所有字段。我想出的代码(也基于SO中的其他答案)是:

import scala.reflect.runtime.universe._
import scala.reflect.runtime.{universe => ru}

  // gets all methods of a Case Class
  def getMethods[T: ru.TypeTag] = typeOf[T].members.collect {
    case m: MethodSymbol if m.isCaseAccessor => m
  }.toList


  /**
    *  Returns the value of all Case Class fields
    * @param obj case class object
    * @return a Sequence of all field values
    */
  def getAllCaseClassFieldValues[T: ru.TypeTag](obj: Object): Seq[Any] = {
    val mirror = ru.runtimeMirror(getClass.getClassLoader)
    getMethods[T].map(m => mirror.reflect(obj).reflectField(m).get)
  }

案例类:

case class SampleRequest(field1: Option[String], field2: Option[String], //.... up to 30 or more fields

检查至少1是否为非空的代码:

 val fieldValues = getAllCaseClassFieldValues[SampleRequest](r)
 val noneCount = fieldValues.count(_ == None)
 val atLeastOneNonEmpty = noneCount < fieldValues.size

我想知道是否有更好的方法通过反射或其他机制进行验证?

2 个答案:

答案 0 :(得分:3)

您可以尝试以下方法:

abstract class AtLeastOneNonEmptyOption[T] { self: T with Product =>

  def nonEmpty: Boolean = this.productIterator.exists {
    case Some(_) => true
    case _ => false
  }

}

case class MyClass(foo: Option[String], bar: Option[Int]) extends AtLeastOneNonEmptyOption[MyClass]

val nonEmptyClass = MyClass(Some("foo"), None)
val emptyClass = MyClass(None, None)

nonEmptyClass.nonEmpty //true
emptyClass.nonEmpty //false

也许有更好的方法。

答案 1 :(得分:3)

这是一个案例类,因此它实现了Product特质。产品特征具有迭代器def productIterator : Iterator[Any]

case class Foo(one: Option[Int], two: Option[String], three : Option[Double])

var i = Foo(None, Some("Woot!"), None)
var j = Foo(Some(1), Some("There Can Be Only"), None)
var k = Foo(None, None, None)

i.productIterator.exists(_.isInstanceOf[Some[_]]) // True
j.productIterator.exists(_.isInstanceOf[Some[_]]) // True
k.productIterator.exists(_.isInstanceOf[Some[_]]) // False

在评估稀疏填充的流数据时,我一直都使用此技巧。