Scala中从可变Map移除指定元素并收集至另一可变Map的惯用方法

嘿，我明白你遇到的问题了——在Scala里遍历大型可变Map时，想一边移除符合条件的元素，一边把它们收集到另一个Map里，但Scala的Iterator偏偏没有Java里的remove()方法，对吧？别担心，我给你两种可行的解决方案，分别适配不同的场景：

这个方法适合大型Map，因为只需要一次遍历就能完成移除和收集操作，而且对于你用的TrieMap这类并发集合，也能保证遍历过程中修改的安全性。

我们可以把Scala的可变Map转换成Java的Map视图，这样就能拿到支持remove()的Java迭代器。具体代码如下：

import scala.collection.mutable
import scala.collection.concurrent.TrieMap
import java.util.{Iterator => JIterator}

def main(args: Array[String]): Unit = {
  val map = TrieMap[String, Integer]()
  map += "one" -> 1
  map += "two" -> 2
  map += "three" -> 3
  
  // 移除所有值大于1的元素并放入removed中
  val removed = removeIf(map, _._2 > 1)
  
  println(s"处理后原Map: $map")       // 输出: 处理后原Map: Map(one -> 1)
  println(s"被移除的元素: $removed")   // 输出: 被移除的元素: Map(two -> 2, three -> 3)
}

def removeIf[K, V](map: mutable.Map[K, V], predicate: ((K, V)) => Boolean): mutable.Map[K, V] = {
  val removedMap = mutable.Map.empty[K, V]
  // 将Scala Map转为Java Map，获取支持remove()的迭代器
  val javaIterator: JIterator[(K, V)] = map.asJava.entrySet().iterator()
  
  while (javaIterator.hasNext) {
    val entry = javaIterator.next()
    val (key, value) = (entry.getKey, entry.getValue)
    
    if (predicate((key, value))) {
      // 收集符合条件的元素
      removedMap += key -> value
      // 安全移除当前元素，不会触发并发修改异常
      javaIterator.remove()
    }
  }
  
  removedMap
}

为什么这个方法可行？ #

• Scala的mutable.Map通过asJava方法可以转为Java的java.util.Map视图，两者共享底层数据，修改Java视图会同步到Scala Map。
• Java迭代器的remove()方法是遍历过程中修改集合的安全方式，不会触发ConcurrentModificationException，对于TrieMap这类并发集合，操作也是线程安全的。

如果你觉得引入Java API有点别扭，也可以用Scala原生的集合方法实现，虽然会涉及两次遍历，但代码更简洁：

import scala.collection.mutable
import scala.collection.concurrent.TrieMap

def main(args: Array[String]): Unit = {
  val map = TrieMap[String, Integer]()
  map += "one" -> 1
  map += "two" -> 2
  map += "three" -> 3
  
  val removed = removeIf(map, _._2 > 1)
  
  println(s"处理后原Map: $map")       // 输出: 处理后原Map: Map(one -> 1)
  println(s"被移除的元素: $removed")   // 输出: 被移除的元素: Map(two -> 2, three -> 3)
}

def removeIf[K, V](map: mutable.Map[K, V], predicate: ((K, V)) => Boolean): mutable.Map[K, V] = {
  // 第一步：过滤出所有符合移除条件的元素
  val removedMap = map.filter(predicate).to(mutable.Map)
  // 第二步：从原Map中移除这些元素的键
  map --= removedMap.keys
  
  removedMap
}

注意事项 #

• 这个方法会对Map进行两次遍历（一次过滤，一次移除），如果你的Map非常大（比如百万级以上元素），性能会比方案一差一些。
• 如果是并发场景（多个线程同时操作TrieMap），过滤和移除之间可能存在间隙，导致其他线程的修改影响结果，这种情况下方案一更可靠。

内容的提问来源于stack exchange，提问作者Igor Urisman