处理Scala映射中缺少键和空值的优雅方式

Question

我理解

• 在Scala中应该避免使用null
• Map.get将返回一个Option[B]，我可以使用. getOrElse获取值并回退到默认值

例如。

map.getOrElse("key1","default")

同时，我正在与一个Java库交互，其中一些值为null。

e、 g.地图（“键1”-

在这种情况下，getOrElse将抛出空指针。

我想处理这两个案件，并导致写这样的东西

  def getOrElseNoNull[A,B](map:Map[A,B],key:A,default:B) = {
    map.get(key) match{
      case Some(x) if x != null => x
      case _ => default
    }
  }

这很难看。（这是Map[任何]，我需要该键的字符串）

getOrElseNoNull(map,"key1","").asInstanceOf[String])

是否可以使用隐式或任何其他优雅的方式来扩展贴图？

Answer 1

一种可能性是将值映射到Option（）：

val withoutNulls: Map[Int, Option[String]] = withNulls.mapValues(Option.apply)

这使您能够以相同的方式处理缺少的值和空值：

val nullsEqualMissing: Map[Int, Option[String]] = withoutNulls.withDefaultValue(None)
nullsEqualMissing(1).fold{ "nullOrMissing" }{ identity }

或单独处理缺失值：

withoutNulls.get(1).fold{ "missing" }{ _.fold{ "null" }{ identity }}

Answer 2

隐式扩展类来拯救：

implicit class NullOccludingMap[K, V](private val underlying: Map[K, V]) extends AnyVal {
  def getNonNullOrElse(key: K, default: V): V = {
    underlying.get(key) match {
      case Some(value) if value != null => value
      case _ => default
    }
  }
}

然后，您可以在范围内的任何地方使用它：

val test = Map("x" -> "Hi", "y" -> null)
test.getNonNullOrElse("z", "") // ""
test.getNonNullOrElse("y", "") // ""

Answer 3

如果您处理的是一个不可变映射，最安全的做法是预先过滤掉所有的空值（这会导致创建另一个映射实例，但除非您有特定的理由关心这里的性能，否则这不应该是一个问题）。

val withoutNulls = map.filter{case (k,v)=> v != null}

任何保留null的键都将消失，因此此键上的getOrElse将返回None。琐碎，一次完成任务。