映射到具有多态函数的泛型元组



Scala 3提供了类似于无形HList的多态函数和元组:

scala> 1 *: "foo" *: Tuple()
val res0: (Int, String) = (1,foo)
scala> val f: ([T] => T => Option[T]) = [T] => (v: T) => Some(v)
val f: PolyFunction{apply: [T](x$1: T): Option[T]} = <function1>
scala> res0.map(f)
val res1: Option[Int] *: Option[String] *: EmptyTuple = (Some(1),Some(foo))

我们如何使用Scala 3功能重新实现下面的无形状示例?

import poly._
object choose extends (Set ~> Option) {
def apply[T](s : Set[T]) = s.headOption
}
scala> val sets = Set(1) :: Set("foo") :: HNil
sets: Set[Int] :: Set[String] :: HNil = Set(1) :: Set(foo) :: HNil
scala> val opts = sets map choose
opts: Option[Int] :: Option[String] :: HNil = Some(1) :: Some(foo) :: HNil

换句话说,我们怎么做这样的编译

scala> val choose: ([T] => Set[T] => Option[T]) = [T] => (s: Set[T]) => s.headOption
val choose: PolyFunction{apply: [T](x$1: Set[T]): Option[T]} = <function1>
scala> val sets = Set(1) *: Set("foo") *: Tuple()
val sets: (Set[Int], Set[String]) = (Set(1),Set(foo))
scala> sets.map(choose)
1 |sets.map(choose)
|         ^^^^^^
|     Found:    (choose : PolyFunction{apply: [T](x$1: Set[T]): Option[T]})
|     Required: PolyFunction{apply: [t](x$1: t): Any}

Shapelessmap比Scala 3元组map更神奇,后者的签名是:

def map[F[_]](f: [t] => (x$1: t) => F[t]): Map[Tuple, F]

Map[Tuple, F]是一个特殊的匹配类型,它基本上是一个元组,每个参数类型都包装在F[_]中。问题是形状,t => F[t]阻止了太多的花哨。

当然,F本身也可以是匹配类型:

type Choose[T] = T match {
case Set[a] => Option[a]
}
def choose[T](t: T): Choose[T] = t match
case set: Set[a] => set.headOption
// messy, but it works
@main def run =
val sets = Set(1) *: Set("foo") *: Tuple()
println(sets.map([T] => (t: T) => choose(t)))

目前存在编译器无法从方法中推断多态函数的问题。匹配类型也不是完全类型安全的,例如执行choose("string")将编译但抛出MatchError。我还遇到了从多态函数值推断匹配类型的问题,因此使用def方法。

最新更新