映射集合并返回与输入相同类型的容器

Question

出于学习目的，我正在尝试定义一个名为DoubleMap 的包装类，它提供了mapBoth 方法。它本质上需要两个函数f 和g，其中g 的域是f 的共同域。这个函数的组成（g o f）应该映射到DoubleMap

包裹的容器

这是我当前的代码：

implicit class DoubleMap
[A, B, C, F[X] <: List[X]] // just List for now
(xs: F[A])(implicit cbf: CanBuildFrom[F[A], C, F[C]]) {
  def mapBoth(f: A => B)(g: B => C): F[C] =
    xs.map(f andThen g).to[F]
}

但是，当我想使用这样的方法时：

List(true, false, false).mapBoth(!_)(!_)

我收到一条关于（找到的）类型 CanBuildFrom[List[_], Nothing, List[Nothing]] 和（必需的）类型 CanBuildFrom[List[Boolean], C, List[C]] 之间类型不匹配的神秘错误消息

为什么编译器会推断出第一种类型？

Answer 1

您的呼叫扩展为

DoubleMap(xs).mapBoth(!_)(!_)

所以DoubleMap 的所有类型参数和cbf 都需要在处理mapBoth 调用之前进行推断。要修复它，只需将这些参数移至mapBoth：

implicit class DoubleMap
[A, F[X] <: List[X]] // just List for now
(xs: F[A]) {
  def mapBoth[B, C](f: A => B)(g: B => C)(implicit cbf: CanBuildFrom[F[A], C, F[C]]): F[C] =
    xs.map(f andThen g).to[F]
}

那么B和C是从f和g确定的，而cbf是从F、A和C确定的。

Answer 2

如果您不希望与 List 绑定，请考虑 Cats 的 Functor

implicit class DoubleMap[F[_]: Functor, A](xs: F[A]) {
  def mapBoth[B, C](f: A => B)(g: B => C): F[C] =
    xs.map(f andThen g)
}

或 vanilla Scala 2.13 方法（受 Luis 的 2.12 implementation 启发）

implicit class DoubleMap[F[x] <: IterableOnce[x], A](xs: F[A]) {
  def mapBoth[B, C](f: A => B)(g: B => C)(implicit bf: BuildFrom[F[A], C, F[C]]): F[C] =
    bf.fromSpecific(xs)(xs.iterator.map(f andThen g))
}

两者都输出

List(true, false, false).mapBoth(!_)(!_)
Vector(true, false, false).mapBoth(!_)(!_)

res1: List[Boolean] = List(true, false, false)
res2: Vector[Boolean] = Vector(true, false, false)