Scala 集合：有安全的地图操作吗？

Question

假设我有一个列表（或地图中的值），我想对每个项目执行操作。但不幸的是，无论出于何种原因，这个值列表都可能包含空值。

scala> val players = List("Messi", null, "Xavi", "Iniesta", null)
players: List[java.lang.String] = List(Messi, null, Xavi, Iniesta, null)

为了避免被 NPE 炸毁，我需要执行以下操作：

scala> players.filterNot(_ == null ).map(_.toUpperCase)
res84: List[java.lang.String] = List(MESSI, XAVI, INIESTA)

有没有更好的方法？

最好是这样的：

players.safeMap(_.toUpperCase)

在scala-language mailing list，Simon 提出了这个：

players.filter ( null !=).map(_.toUpperCase )

这是我最初拍摄的较短版本，并且在没有专用方法的情况下尽可能短。

更好的是，Stefan 和 Kevin 提出了 withFilter 方法，该方法将返回一个惰性代理，因此可以合并两个操作。

players.withFilter ( null !=).map(_.toUpperCase )

Answer 1

如果您无法避免 nulls（例如，如果您从 Java 代码中获取列表），另一种选择是使用 collect 而不是 map：

scala> players.collect { case player if player != null => player.toUpperCase }
res0: List[java.lang.String] = List(MESSI, XAVI, INIESTA)

Answer 2

我会这样做：

players flatMap Option map (_.toUpperCase)

但这比collect 更糟糕。 filter + map 总是用collect 做得更好。

Answer 3

你可以转换成Option[String]的列表：

scala> val optionPlayers = players.map(Option(_))
optionPlayers: List[Option[java.lang.String]] = List(Some(Messi), None, Some(Xavi), Some(Iniesta), None)

Option 普遍优于null，它在如何安全处理数据方面为您提供了很大的灵活性。以下是获得所需结果的简单方法：

scala> optionPlayers.collect { case Some(s) => s.toUpperCase }
res0: List[java.lang.String] = List(MESSI, XAVI, INIESTA)

scala> optionPlayers.flatMap(_.map(_.toUpperCase))
res1: List[java.lang.String] = List(MESSI, XAVI, INIESTA)

scala> optionPlayers.flatten.map(_.toUpperCase)
res2: List[java.lang.String] = List(MESSI, XAVI, INIESTA)

您可以在其他 StackOverflow 问题中或通过网络搜索找到有关 Option 的更多信息。

或者，您始终可以将您想要的 safeMap 方法定义为 List 的隐含方法：

implicit def enhanceList[T](list: List[T]) = new {
  def safeMap[R](f: T => R) = list.filterNot(_ == null).map(f)
}

所以你可以这样做：

scala> players.safeMap(_.toUpperCase)
res4: List[java.lang.String] = List(MESSI, XAVI, INIESTA)

虽然如果您定义一个隐式，您可能希望使用 CanBuildFrom 样式，就像基本集合所做的那样，使其不仅仅适用于 List。您可以在其他地方找到更多相关信息。