Scala Cats 2.9.x中能否对Set执行flatMap操作？

为什么代码无法运行？ #

Scala Cats默认不为scala.collection.immutable.Set提供FlatMap类型类实例，核心原因是Set的flatMap操作违反了Cats FlatMap类型类的法则约束。

Cats的FlatMap要求实例必须满足结合律（即fa.flatMap(f).flatMap(g)等价于fa.flatMap(a => f(a).flatMap(g))）和左单位元等monad法则，但Set的flatMap会自动对元素去重，这会导致某些场景下法则等价性被打破。例如当中间操作产生重复元素时，去重行为会改变最终结果的推导逻辑，不符合Cats对类型类一致性的要求，因此官方未提供默认实例。

修复方案 #

方案1：自定义Set的FlatMap实例（仅适用于不依赖法则的场景） #

如果你不需要使用Cats中依赖法则正确性的功能（如traverse、sequence等），可以手动为Set实现FlatMap实例：

import cats.FlatMap
import scala.collection.immutable.Set

// 手动实现Set的FlatMap实例，复用标准库的集合操作
implicit val setFlatMap: FlatMap[Set] = new FlatMap[Set] {
  override def flatMap[A, B](fa: Set[A])(f: A => Set[B]): Set[B] = fa.flatMap(f)
  override def map[A, B](fa: Set[A])(f: A => B): Set[B] = fa.map(f)
  // 基于Iterable的tailRecM实现转换为Set
  override def tailRecM[A, B](a: A)(f: A => Set[Either[A, B]]): Set[B] =
    FlatMap[Iterable].tailRecM(a)(f).toSet
}

定义完成后，可以简化原函数并正常调用：

import cats.syntax.flatMap._

def flatMap2[F[_]: FlatMap](vs: F[Int]): F[Int] =
  vs.flatMap(v => Set(v, v * 2))

// 测试调用
flatMap2(Set(1, 2, 3)) // 返回Set(1,2,3,4,6)

方案2：换用Cats官方支持的集合类型（推荐） #

如果需要严格遵守Cats的法则，建议替换为官方提供FlatMap实例的集合类型：

• List/Vector：保留元素顺序且不去重，完全符合FlatMap法则，是Cats中最常用的集合类型。
• NonEmptySet：Cats提供的非空集合类型，有对应的FlatMap实例，但需要处理非空约束。

示例（用List替换Set）：

import cats.FlatMap
import cats.syntax.flatMap._

def flatMap2[F[_]: FlatMap](vs: F[Int]): F[Int] =
  vs.flatMap(v => List(v, v * 2))

// 测试调用
flatMap2(List(1, 2, 3)) // 返回List(1,2,2,4,3,6)

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