为何Scala中foreach性能优于while循环？基于微基准测试的技术问询

基于M1芯片运行Azul JDK 11.0.12的微基准测试结果：

Scala 2.13.6

Benchmark               Mode  Cnt      Score      Error  Units
HashMapBenchmark.vectorMutable  thrpt   10  53220.300 ±  220.481  ops/s
HashMapBenchmark.whileMutable   thrpt   10  49586.007 ±   48.883  ops/s

Scala 2.12.15

Benchmark               Mode  Cnt      Score      Error  Units
HashMapBenchmark.vectorMutable  thrpt   10  41298.593 ±  245.915  ops/s
HashMapBenchmark.whileMutable   thrpt   10  36607.998 ±  476.617  ops/s

测试代码：

case class Cookie(name: String, value: String)
state.cookies = (1 to 1000).map(i => Cookie(s"name_$i", s"value_$i")).toVector

@Benchmark
def vectorMutable(state: BenchmarkState, blackhole: Blackhole): Unit = {
  val map = new mutable.HashMap[String, String]()
  state.cookies.foreach(c => map.put(c.name, c.value))
  blackhole.consume(map)
}

@Benchmark
def whileMutable(state: BenchmarkState, blackhole: Blackhole): Unit = {
  val map = new mutable.HashMap[String, String]()
  var i = 0
  while (i < state.cookies.length) {
    val cookie = state.cookies(i)
    map.put(cookie.name, cookie.value)
    i+=1
  }
  blackhole.consume(map)
}

这确实是个反直觉的发现——我们通常会默认手动编写的while循环比高阶函数foreach更高效，但结合Scala集合的实现细节和JVM即时编译（JIT）的优化逻辑，这个现象完全可以解释清楚。

1. Scala集合的foreach是高度优化的底层实现 #

Scala标准库中的集合方法（比如Vector.foreach）并非简单的高阶函数调用，而是针对具体集合类型做了定制化的底层优化：

• Vector的foreach方法内部直接遍历其底层的数组结构，并且通过@inline注解让编译器将传入的闭包逻辑内联到循环中，彻底消除函数调用的开销。
• 相比之下，你手动编写的while循环中，每次调用state.cookies(i)都会触发Vector.apply方法，这个方法默认包含边界检查（判断i是否在合法索引范围内）。而foreach的内部实现可以避免重复的边界检查，因为它是在已知集合长度的前提下遍历整个集合，不需要每次都验证索引合法性。

2. JIT编译器对标准库代码的优化更充分 #

JVM的即时编译器会对频繁执行的热点代码进行深度优化，而Scala标准库的代码是被广泛使用的成熟代码，已经被JVM优化到极致：

• JIT可以对foreach的内部循环进行循环展开、逃逸分析等优化，让执行效率接近甚至超过手动编写的循环。比如循环展开可以减少循环本身的控制开销，逃逸分析可以帮助消除不必要的对象分配。
• 你的手动while循环虽然逻辑简单，但JIT可能无法像优化标准库代码那样充分识别并优化它的模式，比如变量i的递增逻辑、数组访问的边界检查消除等。

3. Scala版本迭代带来的优化缩小了性能差距 #

你观察到Scala 2.13中两者的性能差距缩小，主要得益于两个方面：

• Scala 2.13对集合库进行了全面重构，优化了Vector等集合的底层实现，同时改进了编译器对循环代码的优化能力，让手动while循环的编译结果更高效。
• 你使用的Azul JDK 11本身的JIT编译器也比旧版本更强大，能够更好地识别并优化手动循环的模式，消除不必要的开销。

额外验证思路 #

如果想进一步验证边界检查的影响，可以尝试在手动循环中使用scala.collection.mutable.ArrayBuffer代替Vector（因为ArrayBuffer的apply方法边界检查更容易被JIT消除），或者添加@inline注解帮助编译器优化。但本质上，标准库的foreach已经帮你做了这些底层优化，不需要手动处理。

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