Scala for推导式中的模式匹配赋值如何转换为Monadic操作？

Scala for推导式模式匹配过滤的底层原理 #

你观察到的现象是Scala for推导式的标准脱糖规则，它并不会直接生成collect调用，而是通过你已知的withFilter+map的组合实现了和collect等价的效果，具体逻辑如下：

• 当for推导式的生成器（<-左侧的部分）使用了非单个变量的模式匹配时，脱糖阶段会自动插入withFilter调用，先过滤掉所有无法匹配该模式的元素，再对剩余元素执行后续的map/flatMap操作。

你给出的示例：

case class Account(id: String, role: String)
val accounts = Set(Account("a", "ADMIN"), Account("b", "USER"), Account("c", "ADMIN"), Account("d", "USER"), Account("e", "USER"))

val adminIds = for (Account(id, "ADMIN") <- accounts) yield id

实际脱糖后的等价代码为：

val adminIds = accounts
  .withFilter {
    case Account(_, "ADMIN") => true
    case _ => false
  }
  .map {
    case Account(id, "ADMIN") => id
  }

由于withFilter已经提前筛掉了所有不匹配Account(id, "ADMIN")模式的元素，走到map阶段的元素100%符合模式要求，因此不会抛出MatchError。
而collect方法本身的实现逻辑就是「先做模式匹配过滤，再对匹配到的元素执行转换」，和上述withFilter+map的执行逻辑完全一致，所以你会觉得两者效果相同。

注意：该自动过滤规则仅适用于生成器（<-左侧）的模式匹配场景，如果是for推导式内的赋值语句，或是普通变量赋值时使用模式匹配，不匹配时依然会抛出MatchError。

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