为无形记录定义类型类

时间:2015-03-06 01:59:48

标签: scala record typeclass shapeless

我试图学习无形,我想定义一个将无形记录实例加在一起的幺半群。请注意,我使用的是algebird monoids(不是scalaz),但我确定它们非常相似。以下是我希望能够做到的一个例子:

val result = Monoid.sum(
  ('a ->> 1) :: ('b ->> 1) :: HNil,
  ('a ->> 4) :: ('b ->> 3) :: HNil,
  ('a ->> 2) :: ('b ->> 6) :: HNil)
// result should be: ('a ->> 7) :: ('b ->> 10) :: HNil

我想出了如何为HList编写monoid实例,如下所示:

  implicit val HNilGroup: Group[HNil] = new ConstantGroup[HNil](HNil)
  implicit val HNilMonoid: Monoid[HNil] = HNilGroup
  class HListMonoid[H, T <: HList](implicit hmon: Monoid[H], tmon: Monoid[T]) extends Monoid[::[H, T]] {
    def zero = hmon.zero :: tmon.zero
    def plus(a: ::[H, T], b: ::[H, T]) = 
      hmon.plus(a.head, b.head) :: tmon.plus(a.tail, b.tail)
  }
  implicit def hListMonoid[H, T <: HList](implicit hmon: Monoid[H], tmon: Monoid[T]) = new HListMonoid[H, T]

这允许我写:

val result = Monoid.sum(
  1 :: 1 :: HNil,
  4 :: 3 :: HNil,
  2 :: 6 :: HNil)
// result is 7 :: 10 :: HNil

既然我可以对HList实例求和,那么缺少的部分似乎是定义了monoid实例,它可以对形式为('name ->> 1)的字段求和,我的IDE告诉我它具有以下类型:Int with record.KeyTag[Symbol with tag.Tagged[Constant(name).type] { .. }, Int] { .. }。在这一点上,我被困住了,因为我不知道如何去做这件事。

1 个答案:

答案 0 :(得分:11)

你非常接近 - 你只需要在每个归纳步骤中添加FieldType[K, H]而不是H并使用field[K]来适当地输入从Monoid[H]获得的值:

import com.twitter.algebird._
import shapeless._, labelled._, record._, syntax.singleton._

implicit val hnilGroup: Group[HNil] = new ConstantGroup[HNil](HNil)
implicit val hnilMonoid: Monoid[HNil] = hnilGroup
implicit def hconsMonoid[K, H, T <: HList](implicit
  hm: Monoid[H],
  tm: Monoid[T]
): Monoid[FieldType[K, H] :: T] =
  Monoid.from(field[K](hm.zero) :: tm.zero) {
    case (hx :: tx, hy :: ty) => field[K](hm.plus(hx, hy)) :: tm.plus(tx, ty)
  }

或者您可以使用Shapeless的TypeClass机制,它还为您提供案例类的实例等:

import com.twitter.algebird._
import shapeless._, ops.hlist._, ops.record._, record._, syntax.singleton._

object MonoidHelper extends ProductTypeClassCompanion[Monoid] {
  object typeClass extends ProductTypeClass[Monoid] {
    def emptyProduct: Monoid[HNil] = Monoid.from[HNil](HNil)((_, _) => HNil)
    def product[H, T <: HList](hm: Monoid[H], tm: Monoid[T]): Monoid[H :: T] =
      Monoid.from(hm.zero :: tm.zero) {
        case (hx :: tx, hy :: ty) => hm.plus(hx, hy) :: tm.plus(tx, ty)
      }

    def project[F, G](m: => Monoid[G], to: F => G, from: G => F): Monoid[F] =
      Monoid.from(from(m.zero))((x, y) => from(m.plus(to(x), to(y))))
  }

  implicit def deriveRecordInstance[
    R <: HList,
    K <: HList,
    H,
    T <: HList
  ](implicit
    vs: Values.Aux[R, H :: T],        
    vm: Lazy[Monoid[H :: T]],
    ks: Keys.Aux[R, K],
    zk: ZipWithKeys.Aux[K, H :: T, R]
  ): Monoid[R] = typeClass.project(vm.value, vs(_), zk(_: H :: T))
}

import MonoidHelper._

我在这里提供了一种derivedRecordInstance方法,可以将这项工作记录在案,但我很遗憾有必要 - 你可能会这样做&#39 ; ll将在未来的Shapeless版本中免费获取记录实例。