如何映射所有元素都是类型类实例的 HList?

问题描述

假设我有一个这样的类型类:

trait Select[A] {
  def select(selector: String): Set[A]
}

类型类提供了“给定一个选择器字符串,给我一组 A 对象”的功能。但是,当您需要多次执行此操作时,它很快就会变得很累:

val setofGrapes = Select[Grape].select("red-seedless")
val setofApples = Select[Apple].select("fuji-apple")
// and so on...

我的程序允许用户输入包含多个选择器的字符串,所以这一切都发生在运行时。

为了更简洁,我如何使用 shapeless 编写一个可以像这样使用的函数

val setofGrapes :: setofApples :: HNil =
  selectMultiple("red-seedless,fuji")(Select[Grape] :: Select[Apple] :: HNil)

动机是能够将多个 Select 快速链接在一起。本质上,我通过运行 Select[Grape] :: Select[Apple] :: HNil 中的方法Set[Grape] :: Set[Apple] :: HNil 转到 Select

忽略字符串处理(处理逗号、globs 和其他事情),我将如何实现这一点?

我已经试过了:

object fun extends poly1 {
  implicit def selectCase[A] = at[Select[A]](_.select(""))
}

def mapFun[H <: HList](inputs: H) = inputs map fun

编译器告诉我,Could not find implicit value for parameter mapper: shapeless.ops.hlist.Mapper[fun.type,H]。我认为这是因为它不能保证 HList 的每个元素都是 Select[A]。我如何告诉编译器这个?

我试过像这样指定一个 shapeless.UnaryTCConstraint[H,Select]

def mapFun[H <: HList](inputs: H)(implicit ev: shapeless.UnaryTCConstraint[H,Select]) = inputs map fun

但我收到了同样的错误

一位朋友推荐我使用 ComappedMapper~>,所以我尝试了这个:

object mapper extends (Select ~> Set) {
  override def apply[A](s: Select[A]): Set[A] = s.select(???)
}

def mapFun[II <: HList,OO <: HList](inputs: II)(
  implicit ev: Comapped.Aux[II,Select,OO],ev2: Mapper.Aux[mapper.type,II,OO]
): OO = input map fun

但是编译器Could not find implicit value for parameter mapper: shapeless.ops.hlist.Mapper[ammonite.$sess.cmd8.fun.type,II]

解决方法

您基本上应该只将所需的隐式证据添加到您的 mapFun 方法中:

object fun extends Poly1 {
  implicit def selectCase[A] = at[Select[A]](_.select(""))
}

def mapFun[H <: HList](inputs: H)(implicit m: Mapper[fun.type,H]) = inputs map fun
scala> mapFun(Select[Apple] :: Select[Grape] :: Select[Apple] :: HNil)
val res1: scala.collection.immutable.Set[Apple] :: scala.collection.immutable.Set[Grape] :: scala.collection.immutable.Set[Apple] :: shapeless.HNil = Set(Apple@44ebbbe8) :: Set(Grape@76eae32c) :: Set(Apple@666f67b6) :: HNil

要获得您喜欢的 API,您可以引入一个“手动柯里化”函数,如下所示:

class SelectMultipleApply(selector: String) {
  def apply[H <: HList](inputs: H)(implicit m: Mapper[fun.type,H]) = inputs map fun

  object fun extends Poly1 {
    implicit def selectCase[A] = at[Select[A]](_.select(selector))
  }
}

def selectMultiple(selector: String) = new SelectMultipleApply(selector)
scala> selectMultiple("red-seedless,fuji")(Select[Grape] :: Select[Apple] :: HNil)
val res5: scala.collection.immutable.Set[Grape] :: scala.collection.immutable.Set[Apple] :: shapeless.HNil = Set(Grape@63c81efa) :: Set(Apple@29dbae38) :: HNil