在模式匹配中记录警告,不返回任何内容

问题描述 投票:1回答:2

我有一个可迭代的数组,我试图将它们转换为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!
  }
}

我想编写一个满足以下条件的函数:

  1. 映射到Iterable,将每个元素转换为MyCaseClass
  2. 当我收到无法转换为MyCaseClass的数组时记录警告
  3. 记录警告后,将忽略/丢弃传递给匹配条件的数组
  4. 返回类型应为Iterable[MyCaseClass]

我如何满足这些条件?

scala
2个回答
1
投票

您可以执行以下操作:

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"))

1
投票

考虑使用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
© www.soinside.com 2019 - 2024. All rights reserved.