Scala中非案例类的模式匹配

Question

假设我有一个普通的第三方（即我不能修改它）类定义如下：

class Price(var value: Int)

这是否可以将此类的实例与某些模式匹配？

比如我要实现函数：

def printPrice(price: Price) = {
    // implementation here
}

...在所有其他情况下，为每个具有value <= 9000 和price is over 9000 的price 打印price is {some value}。

例如调用：

printPrice(new Price(10))
printPrice(new Price(9001))

应该打印：

price is 10
price is over 9000

如何使用模式匹配实现printPrice？

Answer 1

您可以创建自定义提取器：

package external {
    class Price(var value: Int)
}

object Price {
    def unapply(price: Price): Option[Int] = Some(price.value)
}

def printPrice(price: Price) = price match {
    case Price(v) if v <= 9000 => println(s"price is $v")
    case _ => println("price is over 9000")
}

printPrice(new Price(10))
printPrice(new Price(9001))

对于案例类，编译器会自动生成它。我认为在你的情况下提取器是矫枉过正的，但可能只是简化的样本。

Answer 2

考虑过接受弗拉维安的解决方案，但我自己想出了更好的解决方案。

以下是如何实现printPrice（无需使用包装对象和修改原始类）：

def printPrice(price: Price) = price match {
    case p: Price if (p.value <= 9000) => println("price is " + p.value)
    case p: Price => println("price is over 9000")
}

PS：感谢 flavian 证明您可以在模式中使用 if。对此表示赞同。

Answer 3

您可以使用 PIMP 我的库模式：

case class RichPrice(value: Int) {}
  implicit def priceToRichPrice(price: Price): RichPrice = RichPrice(price.value)

  def printPrice(x: RichPrice): Unit = {
    x match {
      case RichPrice(value) if (value <= 9000) => println("below 9000")
      case RichPrice(value) if (value > 9000) => println("over 9000")
      case _ => println("wtf")
    }
  }
println(printPrice(new Price(10)))
println(printPrice(new Price(9001)))

使用case class 的目的是让Scala 定义apply 方法和unapply 用于模式匹配的魔法。