scala - Scala 中的类型推断和模式匹配

Question

这不会输入:

sealed trait BinOp[-InA, -InB, +Out]
case object Add extends BinOp[Int, Int, Int]

sealed trait Expression[+A]
final case class IntegerAtom(value: Int) extends Expression[Int]
final case class BinaryExp[-A, -B, +C](op: BinOp[A, B, C], lhs: Expression[A], rhs: Expression[B]) extends Expression[C]

def optimizeStep[A](x: Expression[A]): Expression[A] = x match {
  case BinaryExp(Add, IntegerAtom(a), IntegerAtom(b)) => IntegerAtom(a + b)
}

最直接的是在模式匹配中使用 case 对象:

[error] (...) pattern type is incompatible with expected type;
[error]  found   : minimumexample.Add.type
[error]  required: minimumexample.BinOp[Any,Any,A]

看来这可以通过引入eye-bleeding来解决:

val AddOp = Add

进而:

case BinaryExp(AddOp, IntegerAtom(a), IntegerAtom(b)) => IntegerAtom(a + b)

但是之后:

[error] (...) type mismatch;
[error]  found   : minimumexample.IntegerAtom
[error]  required: minimumexample.Expression[A]
[error]     case BinaryExp(AddOp, IntegerAtom(a), IntegerAtom(b)) => IntegerAtom(a + b)
[error]                                                                         ^

我想尽可能安全地解决这个问题，而不是求助于 .asInstanceOf[] .想法？

Answer 1

您的代码的主要问题是 BinaryExp 定义中的差异问题。 ，但这似乎不在问题的范围内。一旦确定了方差，唯一的不便就是 case object不引入新类型。

解决此问题的典型模式是声明 sealed trait然后有一个 case object延长它。

这是一个编译的例子

sealed trait BinOp[-InA, -InB, +Out]
sealed trait Add extends BinOp[Int, Int, Int]
case object Add extends Add

sealed trait Expression[+A]
final case class IntegerAtom(value: Int) extends Expression[Int]
final case class BinaryExp[A, B, C](op: BinOp[A, B, C], lhs: Expression[A], rhs: Expression[B]) extends Expression[C]

def optimizeStep[A](x: Expression[A]): Expression[A] = x match {
  case BinaryExp((_: Add), IntegerAtom(a), IntegerAtom(b)) => IntegerAtom(a + b)
}

在哪里:

差异以天真的方式“固定”(删除它)

Add由于 sealed trait 现在是一种类型定义

匹配使用 (_: Add) 执行