首页 > 解决方案 > Scala 从 DSL 到函数的自然转换

问题描述

我正在尝试在各种类型之间执行自然转换,* -> * -> * 所以我想F[A, B] => G[A, B] 具体来说我正在尝试定义一个 DSL,然后我可以将其转换为实际的函数定义,所以MyDSL[A, B] => Function[A, B]

这是自然变换的定义:

trait ~~>[F[_, _], G[_, _]] {
  def apply[A, B](fab: F[A, B]): G[A, B]
}

object ~~> {
  def apply[F[_, _], G[_, _]](implicit f2g: F ~~> G): F ~~> G = f2g
}

DSL 看起来像这样:

sealed trait MyDSL[A, B]

object MyDSL {

  case object Add1 extends MyDSL[Int, Int]
  case object Show extends MyDSL[Int, String]

  implicit def dsltoF: MyDSL ~~> Function = new ~~>[MyDSL, Function] {
    override def apply[A, B](fab: MyDSL[A, B]): Function[A, B] = fab match {
      case Add1 => i => i + 1
      case Show => i => i.toString
    }
  }
}

直接使用自然变换可以正常工作:

dsltoF(Add1)

输出:res0: Function[Int,Int] = MyDSL$$anon$2$$Lambda$1816/700824958@6f3aa425

它甚至适用于返回的函数是采用 2 个类型参数的方法的情况。

当我尝试定义一个使用一种类型参数的泛型方法进行转换的 DSL 对象时,它会出现问题。

case class Id[A]() extends MyDSL[A, A]

implicit def dsltoF: MyDSL ~~> Function = new ~~>[MyDSL, Function] {
    override def apply[A, B](fab: MyDSL[A, B]): Function[A, B] = fab match {
      case Id() => identity[A] _
      case Add1 => i => i + 1
      case Show => i => i.toString
    }
  }

我得到一个found A required B编译错误。在这种情况下,Scala 不承认 B 是 A。

我明白了为什么,因为类型参数 A 和 B 不一定与我返回的函数的定义正确相关,因此甚至写: case Add1 => i => i + 1 IntelliJ 中有红线,因为它没有意识到即使 Add "是" MyDSL[Int, Int]。尽管 Scala 对此表示满意。

类型参数对适用于自然变换的方法签名的所有可能性都是开放的,但在这种情况下,它需要某种限制。我的猜测是,由于 DSL 案例类中没有限制类型参数的值,它归结为模式匹配,这已经超过了 Scala 解释方法签名的位置,因此它需要不同的类型 B 并且它树皮。

我当然可以通过 .asInstanceOf 来解决这个问题,但我的意思是来吧。

任何关于不同策略以使其发挥作用的想法将不胜感激。

标签: scalagenericsfunctional-programmingpattern-matchingtype-inference

解决方案


这是当前语言版本中类型推断系统的一个已知限制,应该在未来版本中取消。

在这种情况下,您可以在模式匹配中使用类型变量来解决此限制:

import scala.language.higherKinds

trait ~~>[F[_, _], G[_, _]] {
  def apply[A, B](fab: F[A, B]): G[A, B]
}

object ~~> {
  def apply[F[_, _], G[_, _]](implicit f2g: F ~~> G): F ~~> G = f2g
}

sealed trait MyDSL[A, B]

object MyDSL {

  case class Id[A]() extends MyDSL[A, A]
  case class Const[A, B](constantResult: B) extends MyDSL[A, B]
  case object Add1 extends MyDSL[Int, Int]
  case object Show extends MyDSL[Int, String]

  implicit def dsltoF: MyDSL ~~> Function = new (MyDSL ~~> Function) {
    override def apply[A, B](fab: MyDSL[A, B]): Function[A, B] = fab match {
      case _: Id[x] => identity[x] _
      case c: Const[a, b] => (_ => c.constantResult)
      case Add1 => i => i + 1
      case Show => i => i.toString
    }
  }
}

本质上:如果没有编译器可以存放更多特定类型信息的“地方”,只需在模式中给它一个类型变量,以便它可以将推断的类型信息附加到它。


推荐阅读