我有一个可迭代的数组,我试图将它们转换为case类,并且我正在映射它们以这样做。如果数组不可转换为案例类,我想记录一个警告并继续进行映射。但是,当我实施警告时,返回类型从Iterable[MyCaseClass]
更改为Iterable[Any]
,这不是我想要的。例如:
case class MyCaseClass(s1: String, s2: String)
object MyCaseClass {
def apply(sa: Array[String]) = new MyCaseClass(sa(0), sa(1))
}
val arrayIterable: Iterable[Array[String]] = Iterable(Array("a", "b"), Array("a", "b", "c"))
def badReturnType(): Iterable[Any] = { // Iterable[Any] is undesireable
arrayIterable map {
case sa: Array[String] if sa.length == 2 => MyCaseClass(sa)
case _ => println("something bad happened!") // but warnings are good
}
}
def desiredReturnType(): Iterable[MyCaseClass] = { // Iterable[MyCaseClass] is desireable
arrayIterable map {
case sa: Array[String] if sa.length == 2 => MyCaseClass(sa)
// but no warnings if things go wrong!
}
}
我想编写一个满足以下条件的函数:
MyCaseClass
MyCaseClass
的数组时记录警告Iterable[MyCaseClass]
。我如何满足这些条件?
您可以执行以下操作:
final case class MyCaseClass(s1: String, s2: String)
def parse(input: Array[String]): Either[String, MyCaseClass] = input match {
case Array(s1, s2) => Right(MyCaseClass(s1, s2))
case _ => Left(s"Bad input: ${input.mkString("[", ", ", "]")}")
}
def logErrors(validated: Either[String, _]): Unit = validated match {
case Left(error) => println(error)
case Right(_) => ()
}
def validateData(data: IterableOnce[Array[String]]): List[MyCaseClass] =
data
.iterator
.map(parse)
.tapEach(logErrors)
.collect {
case Right(value) => value
}.toList
您可以这样使用:
val arrayIterable = Iterable(Array("a", "b"), Array("a", "b", "c"))
validateData(arrayIterable)
// Bad input: [a, b, c]
// res14: List[MyCaseClass] = List(MyCaseClass("a", "b"))
考虑使用List
而不是Array
,并尝试与Option
一起包装在flatMap
中>
l flatMap { case e if e.length == 2 => Some(MyCaseClass(e)) case e => println(s"$e is wrong length"); None }
另一种方法是
partitionMap
partitionMap