无形 - 使用HList应用工厂函数

时间:2017-06-12 01:41:48

标签: scala shapeless hlist

我有一堆工厂函数,它们都采用相同的args并返回不同的类型。目前我明确地调用它们,但它非常详细,我想将工厂通用化为HList并通过映射HList来调用它们。

case class Endpoint[A](a: A)

case class Factory[Ret](f: Int => Endpoint[Ret])

val factories = Factory[Int](a => Endpoint(a)) :: Factory[String](a => Endpoint(a.toString)) :: HNil

我已定义Poly1,以便我可以映射HList并为每个元素应用f

case class ApplyFactory(param: Int) extends Poly1 {
  implicit def generic[A]: Case.Aux[Factory[A], Endpoint[A]] =
    at((factory: Factory[A]) => factory.f(param))
}

val endpoints = factories.map(ApplyFactory(5))

问题在于could not find implicit value for parameter mapper。将ApplyFactory更改为对象会使代码编译。如果将HList定义为类而不是对象,如何映射Poly?或者是否有更好的模式来应用HList函数和一组给定的参数并返回一个新的HList结果?

1 个答案:

答案 0 :(得分:1)

不要为Poly而烦恼,只需手动实现您需要的任何自定义地图:

trait FactoryMap[L <: HList] {
  type Out <: HList
  def map(i: Int, h: L): Out
}

object FactoryMap {
  type Aux[L <: HList, O <: HList] = FactoryMap[L] { type Out = O }

  implicit def caseNil: Aux[HNil, HNil] = new FactoryMap[HNil] {
    type Out = HNil
    def map(i: Int, l: HNil): HNil = l
  }

  implicit def caseCons[T <: HList, O <: HList]
    (implicit ev: Aux[T, O]) = new FactoryMap[Factory[Int] :: T] {
      type Out = Endpoint[Int] :: O
      def map(i: Int, l: Factory[Int] :: T): Endpoint[Int] :: O = {
        val (h :: t) = l
        h.f(i) :: ev.map(i, t)
      }
    }
}

implicit case class MyMap[L <: HList](l: L) {
  def customMap[O <: HList](i: Int)(implicit ev: FactoryMap.Aux[L, O]): O =
    ev.map(i, l)
}

Poly不同,将上述内容概括为任何Ret(而不是像我所做的那样Ret = Int)相对简单。