拉出具有依赖关系的无形多态函数

时间:2015-12-03 18:07:34

标签: scala shapeless

对于无形的新手,我对使用需要某些依赖关系的多态函数有疑问。我基本上有这个代码,并希望将somePoly对象从run方法中拉出来:

import shapeless._
object SomeObject {
    type SomeType = Int :+: String :+: (String, Int) :+: CNil

    def run( someList: List[SomeType], someInt:Int, someWord:String ) = {

       object somePoly extends Poly1 {
           implicit def doIt = at[Int]( i => i + someInt + someWord.length)
           implicit def doIt2 = at[String]( i => i.length + someWord.length)
           implicit def doIt3 = at[(String, Int)]( i => i._1.length + someWord.length)
       }

       someList.map( _.map(somePoly) )
    }
}

我想做的一种方式就是这样,但它似乎很乱:

object TypeContainer {
  type SomeType = Int :+: String :+: (String, Int) :+: CNil
}
case class SomePolyWrapper( someList: List[TypeContainer.SomeType], someInt:Int, someWord:String ){
    object somePoly extends Poly1 {
       implicit def doIt = at[Int]( i => i + someInt + someWord.length)
       implicit def doIt2 = at[String]( i => i.length + someWord.length)
      implicit def doIt3 = at[(String, Int)]( i => i._1.length + someWord.length)
   }
}  
object SomeObject {

    def run( someList: List[TypeContainer.SomeType], someInt:Int, someWord:String ) = {

       val somePolyWrapper = SomePolyWrapper(someList, someInt, someWord)

       someList.map( _.map(somePolyWrapper.somePoly) )
    }
}

有人有什么建议吗?

1 个答案:

答案 0 :(得分:4)

Scala的隐式解析系统的局限性意味着Poly定义需要是一个稳定的标识符,这使得这种事情比它应该更痛苦。正如我在Gitter上提到的,我知道有几种解决方法(可能还有其他解决方法)。

一种方法是将Poly1设为PolyN,其中额外参数用于someIntsomeWord值。如果您要映射HList,则可以使用mapConstzip来使输入HList具有正确的形状。我从来没有为副产品做过这个,但类似的东西可能会起作用。

另一种方法是使用自定义类型类。在你的情况下,可能看起来像这样:

import shapeless._

trait IntFolder[C <: Coproduct] {
  def apply(i: Int, w: String)(c: C): Int
}

object IntFolder {
  implicit val cnilIntFolder: IntFolder[CNil] = new IntFolder[CNil] {
    def apply(i: Int, w: String)(c: CNil): Int = sys.error("Impossible")
  }

  def instance[H, T <: Coproduct](f: (H, Int, String) => Int)(implicit
    tif: IntFolder[T]
  ): IntFolder[H :+: T] = new IntFolder[H :+: T] {
    def apply(i: Int, w: String)(c: H :+: T): Int = c match {
      case Inl(h) => f(h, i, w)
      case Inr(t) => tif(i, w)(t)
    }
  }

  implicit def iif[T <: Coproduct: IntFolder]: IntFolder[Int :+: T] =
    instance((h, i, w) => h + i + w.length)

  implicit def sif[T <: Coproduct: IntFolder]: IntFolder[String :+: T] =
    instance((h, i, w) => h.length + i + w.length)

  implicit def pif[T <: Coproduct: IntFolder]: IntFolder[(String, Int) :+: T] =
    instance((h, i, w) => h._1.length + i + w.length)
}

然后您可以编写run的更通用版本:

def run[C <: Coproduct](
  someList: List[C],
  someInt: Int,
  someWord: String
)(implicit cif: IntFolder[C]): List[Int] = someList.map(cif(someInt, someWord))

并像这样使用它:

scala> run(List(Coproduct[SomeType](1)), 10, "foo")
res0: List[Int] = List(14)

scala> run(List(Coproduct[SomeType](("bar", 1))), 10, "foo")
res1: List[Int] = List(16)

操作的特殊性使得这种方法看起来有点奇怪,但如果我真的需要为不同的副产品做这样的事情,这可能是我选择的解决方案。