如何为模式匹配定义`StrContains.unapply`？

Question

在我的代码中有一个方法，它需要检查传递的字符串是否包含一些指定的字符，然后执行一些操作。

代码看起来是：

def check(str: String) = {
  if(str.contains("A")) {
    doSomething()
  } else if(str.contains("B")) {
    doSomething()
  } else if(str.contains("C")) {
    doSomething()
  } else {
    doSomething()
  }
}

我想试一试模式匹配，但不太满意：

def check(str: String) = str match {
  case s if s.contains("A") => doSomething()
  case s if s.contains("B") => doSomething()
  case s if s.contains("C") => doSomething()
  case _ => doSomething()
}

我希望我可以定义一个StrContains.unapply来像这样使用它：

def check(str: String) = str match {
  case StrContains("A") => doSomething()
  case StrContains("B") => doSomething()
  case StrContains("C") => doSomething()
  case _ => doSomething()
}

但现在可以确定该怎么做了。有什么想法吗？

Answer 1

问题是，当您执行case StrContains("A")时，编译器将首先调用StrContains.unapply/unapplySeq (无论以何种方式定义)，然后尝试将返回的结果与"A"匹配。"A"文本本身永远不会传递给StrContains.unapply/unapplySeq，因此它无法在unapply/unapplySeq中执行调用s.contains("A")。

简单地说，这意味着您需要定义不同的对象，比如StrContainsA/StrContainsB/StrContainsC，这显然比简单地执行case s if s.contains("A")更糟糕。

但是，有一种替代的(而且有些人为的)解决方案，它允许定义单个提取器，同时仍然能够指定要匹配的子字符串内联，这就是利用scala支持基于字符串内插定义提取器的事实：

implicit class ContainsContext (val sc : StringContext) {
  object StrContains {
    def unapply(str: String): Boolean = {
      val substr: String = sc.parts.mkString
      str.contains(substr)
    }
  }
}

用法：

def check(str: String) = str match {
  case StrContains"A" => doSomething()
  case StrContains"B" => doSomething()
  case StrContains"C" => doSomething()
  case _ => doSomething()
}