Question

我有一个HList，其中每列代表一个表的列。 HList中的每个列表具有相同的长度。

我希望能够编写一个函数，将该表的各行选取为元组或值的HList。最终我会把它转换成更合理的东西（例如Case Case）。

import shapeless.PolyDefns.~>
import shapeless.{HList, HNil}
val a = List(1,2,3) :: List("a", "b", "c") :: List(true, false, true) :: HNil

object broken extends (HList ~> HList) {
  def apply[T](n:Int, l:HList): HList = {
    // I want to pick out the nth element of each HList
    // so in the above example, if n==1
    // I want to return
    // 2 :: "b" :: false :: HNil
    ???
  }
}

broken(1,a)

我可以修复此功能，使其按照我在评论中描述的内容工作吗？

加分点：我可以将它写为迭代器，将上面的HList“a”转换为（Int，String，Boolean）或等效HList的序列吗？

Answer 1

有很多方法可以做到这一点，但我会选择自定义类型：

import shapeless._

trait RowSelect[L <: HList] extends DepFn2[L, Int] {
  type Row <: HList
  type Out = Option[Row]
}

object RowSelect {
  def select[L <: HList](l: L, i: Int)(implicit rs: RowSelect[L]): rs.Out = rs(l, i)

  type Aux[L <: HList, Row0 <: HList] = RowSelect[L] { type Row = Row0 }

  implicit val hnilRowSelect: Aux[HNil, HNil] = new RowSelect[HNil] {
    type Row = HNil
    def apply(l: HNil, i: Int): Option[HNil] = Some(HNil)
  }

  implicit def hconsRowSelect[A, T <: HList](implicit
    trs: RowSelect[T]
  ): Aux[List[A] :: T, A :: trs.Row] = new RowSelect[List[A] :: T] {
    type Row = A :: trs.Row
    def apply(l: List[A] :: T, i: Int): Option[A :: trs.Row] = for {
      h <- l.head.lift(i)
      t <- trs(l.tail, i)
    } yield h :: t
  }
}

其中的工作原理如下：

scala> println(RowSelect.select(a, 0))
Some(1 :: a :: true :: HNil)

scala> println(RowSelect.select(a, 1))
Some(2 :: b :: false :: HNil)

scala> println(RowSelect.select(a, 2))
Some(3 :: c :: true :: HNil)

scala> println(RowSelect.select(a, 3))
None

RowSelect L个L个实例，List是一个包含所有List个元素的hlist，并提供一个操作，可以选择从每个元素的指定索引处选择项目NatTRel。

您应该可以使用ConstMapper或ZipWith和Poly2以及RowSelect的组合来完成同样的事情，但是自定义类型类可以很好地将所有内容捆绑在一起并且在许多情况下允许更方便的组合性。

例如，在这种情况下，您的奖金问题的解决方案可以非常简单地用def allRows[L <: HList](l: L)(implicit rs: RowSelect[L]): List[rs.Row] = Stream.from(0).map(rs(l, _).toList).takeWhile(_.nonEmpty).flatten.toList写成：

scala> allRows(a).foreach(println)
1 :: a :: true :: HNil
2 :: b :: false :: HNil
3 :: c :: true :: HNil

然后：

def allRows[L <: HList, R <: HList](l: L)(implicit
  rs: RowSelect.Aux[L, R],
  tp: ops.hlist.Tupler[R]
): List[tp.Out] =
  Stream.from(0).map(rs(l, _).map(tp(_)).toList).takeWhile(_.nonEmpty).flatten.toList

同样如果你想将这些hlists转换为元组：

scala> allRows(a)
res7: List[(Int, String, Boolean)] = List((1,a,true), (2,b,false), (3,c,true))

这给了你：

<ul>
    <li id="test1">a</li>
    <li id="test2">b</li>
    <li id="test_test">c</li>
</ul>

等等。

选取Lists的HList的第N个元素，并将该值作为值的HList返回

1 个答案: