scala - 斯卡拉(Scala)。如何创建接受具有不同Arities元组的通用方法？

在我的应用程序中，我有很多地方需要获取元组列表，将其按元组的第一个元素分组，然后将其从其余元素中删除。例如，我有元组

(1, "Joe", "Account"), (1, "Tom", "Employer"), (2, "John", "Account"), and result should be Map(1 -> List(("Joe", "Account"), ("Joe", "Account")), 2 -> List(("John", "Account")))

它很容易实现为

data.groupBy(_._1).map { case (k, v) => k -> v.map(f => (f._2, f._3)) }

但是我正在寻找通用的解决方案，因为我可以使用具有不同Ar项(2、3、4甚至7)的元组。
我认为Shapeless或Scalaz可以帮助我，但是在这些库中我的经验很低，请举一些例子

最佳答案

这可以使用shapeless轻松实现(为简单起见，我不会针对所有集合类型将其通用化)。元组有一个特定的类型类，可以将它们分解为头和尾，称为IsComposite

import shapeless.ops.tuple.IsComposite

def groupTail[P, H, T](tuples: List[P])(
    implicit ic: IsComposite.Aux[P, H, T]): Map[H, List[T]] = {
  tuples
        .groupBy(ic.head)
        .map { case (k, vs) => (k, vs.map(ic.tail)) }
}

这适用于您的情况:

val data =
  List((1, "Joe", "Account"), (1, "Tom", "Employer"), (2, "John", "Account"))

assert {
  groupTail(data) == Map(
    1 -> List(("Joe", "Account"), ("Tom", "Employer")),
    2 -> List(("John", "Account"))
  )
}

以及不同类型的Tuple4:

val data2 = List((1, 1, "a", 'a), (1, 2, "b", 'b), (2, 1, "a", 'b))

assert {
  groupTail(data2) == Map(
    1 -> List((1, "a", 'a), (2, "b", 'b)),
    2 -> List((1, "a", 'b))
  )
}

可运行的代码可用at Scastie

关于scala - 斯卡拉(Scala)。如何创建接受具有不同Arities元组的通用方法？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/45350139/