类型推断在通用 Scala 函数中失败

Question

考虑一个对集合 distinctBy 进行操作的简单函数，它像 distinct 一样删除 "duplicates"（不需要实际重复项）：

import scala.collection.TraversableLike
import scala.collection.generic.CanBuildFrom
import scala.collection.mutable.{Set=>MSet}

def distinctBy[T,R,Coll]
  (xs: Coll)
  (f: T => R)
  (implicit ev: Coll <:< TraversableLike[T,Coll], 
   cbf: CanBuildFrom[Coll,T,Coll]): Coll = {

  val builder = cbf(xs)
  builder.sizeHint(xs.size)
  val seen = MSet.empty[R]

  xs.foreach { elem =>
    if(!seen(f(elem))){
      builder += elem
    }
  }
  builder.result()
}

现在考虑 class 将其用于：

case class X(i: Int, j: Int)

天真地使用这个函数失败了：

scala> distinctBy(Vector(X(1,2),X(3,2),X(1,1),X(2,2)))(_.i)
<console>:14: error: missing parameter type for expanded function ((x) => x.i)
              distinctBy(Vector(X(1,2),X(3,2),X(1,1),X(2,2)))(_.i)
                                                          ^
<console>:14: error: Cannot construct a collection of type scala.collection.immutable.Vector[X] with elements of type Any based on a collection of type scala.collection.immutable.Vector[X].
              distinctBy(Vector(X(1,2),X(3,2),X(1,1),X(2,2)))(_.i)
                                                             ^

但如果我帮助类型推断器，这会起作用：

scala> distinctBy(Vector(X(1,2),X(3,2),X(1,1),X(2,2)))((x:X) => x.i)
res1: scala.collection.immutable.Vector[X] = Vector(X(1,2), X(3,2), X(1,1), X(2,2))

scala> distinctBy[X,Int,Vector[X]](Vector(X(1,2),X(3,2),X(1,1),X(2,2)))(_.i)
res2: scala.collection.immutable.Vector[X] = Vector(X(1,2), X(3,2), X(1,1), X(2,2))

据我所知，由于该函数是在第二个参数列表中给出的，因此类型推断器应该已经识别出它是从 X 到某物的函数。由于 X 有一个类型为 Int 的成员 i，第一次尝试应该一切正常。那么，我在这里缺少什么？

Answer 1

这个简化版很适合我：

object A {
  def f1[T, R](l: List[T])(f: T=>R) = None

  case class X(i: Int, j: Int)

  f1(List(X(1,1),X(2,1)))(_.i)
}

如您所见，第一个参数列表中的集合具有允许第二个参数列表中的 scala 推理类型的 T 类型。

所以您需要在您的示例中以某种方式在 Col 和 T 之间建立依赖关系。不确定第三个隐式参数列表是否对这里有帮助。

更新。看起来很奇怪但似乎有效：

object A {
  def f1[T, R, Col[Z]](l: Col[T])(f: T => R) = None

  case class X(i: Int, j: Int)

  f1(List(X(1,1),X(2,1)))(_.i)
}

UPD2。重写问题样本。

  import scala.collection.TraversableLike
  import scala.collection.generic.CanBuildFrom
  import scala.collection.mutable.{Set=>MSet}

  def distinctBy[T,R,Coll[Z]]
  (xs: Coll[T])
  (f: T => R)
  (implicit ev: Coll[T] <:< TraversableLike[T,Coll[T]],
   cbf: CanBuildFrom[Coll[T],T,Coll[T]]): Coll[T] = {

    val builder = cbf(xs)
    builder.sizeHint(xs.size)
    val seen = MSet.empty[R]

    xs.foreach { elem =>
      if(!seen(f(elem))){
        builder += elem
        seen.add(f(elem))
      }
    }
    builder.result()
  }

  case class X(i: Int, j: Int)

  distinctBy(Vector(X(1,2),X(1,2),X(3,2),X(1,1),X(2,2)))(_.i)
  distinctBy(Map("1" -> X(1,2), "2" -> X(1,2), "3" -> X(3,2)))(_._2.i)

类型推断在通用 Scala 函数中失败

type inference fails in a generic scala function

collections

scala

type-inference