字符串模式匹配最佳实践

Question

以下是不起作用的代码，但它描述了我想要做的事情。

您能推荐解决这个问题的最佳方法吗？

def resolveDriver(url: String) = {
  url match {
    case url.startsWith("jdbc:mysql:") => "com.mysql.jdbc.Driver"
    case url.startsWith("jdbc:postgresql:") => "org.postgresql.Driver"
    case url.startsWith("jdbc:h2:") => "org.h2.Driver"
    case url.startsWith("jdbc:hsqldb:") => "org.hsqldb.jdbcDriver"
    case _ => throw new IllegalArgumentException
  }
}

Answer 1

从 Scala 2.13 开始，可以通过 取消应用字符串插值器：

val s"jdbc:$dialect:$rest" = "jdbc:mysql:whatever"
// dialect: String = "mysql"
// rest: String = "whatever"

然后在我们的例子中，只需使用 Map 将提取的值（sql 方言）映射到适当的驱动程序即可：

val drivers = Map(
  "postgresql" -> "org.postgresql.Driver",
  "mysql"      -> "com.mysql.jdbc.Driver",
  "h2"         -> "org.h2.Driver"
)
val driver = drivers(dialect)
// driver: String = "com.mysql.jdbc.Driver"

---

如果您期望输入格式错误，您还可以使用匹配语句：

"jdbc:postgresql:something" match {
  case s"jdbc:$dialect:$rest" => Some(dialect)
  case _                      => None
}
// Option[String] = Some("postgresql")

Answer 2

这是另一种方法。将所有映射存储在映射中，然后使用 collectFirst 方法查找匹配项。 collectFirst 的类型签名是：

def TraversableOnce[A].collectFirst[B](pf: PartialFunction[A, B]): Option[B]

用法：

scala> val urlMappings = Map("jdbc:mysql:" -> "com.mysql.jdbc.Driver", "jdbc:postgresql:" -> "org.postgresql.Driver")
urlMappings: scala.collection.immutable.Map[java.lang.String,java.lang.String] = Map(jdbc:mysql: -> com.mysql.jdbc.Drive
r, jdbc:postgresql: -> org.postgresql.Driver)

scala> val url = "jdbc:mysql:somestuff"
url: java.lang.String = jdbc:mysql:somestuff

scala> urlMappings collectFirst { case(k, v) if url startsWith k => v }
res1: Option[java.lang.String] = Some(com.mysql.jdbc.Driver)

Answer 3

import PartialFunction._  // condOpt

val drivers = Map( 
  "mysql" -> "mysql driver",
  "h2" -> "h2 driver" 
  // ... 
)
val pattern = "^jdbc:(\\w+):.*".r

def resolveDriver(url: String) = condOpt(url) { 
  case pattern(dbms) => drivers.get(dbms) 
}.flatten.getOrElse(throw new IllegalArgumentException)

Answer 4

在语法方面，您可以稍微修改一下 case 语句：

case url if url.startsWith("jdbc:mysql:") => "com.mysql.jdbc.Driver"

这只是将值 url 绑定到模式表达式（也是 url）并添加一个保护表达与测试。这应该会使代码编译。

为了让它更像 scala，你可以返回一个 Option[String] （我删除了几个子句，因为它只是为了说明）：

def resolveDriver(url: String) = url match {
  case u if u.startsWith("jdbc:mysql:") => Some("com.mysql.jdbc.Driver")
  case u if u.startsWith("jdbc:postgresql:") => Some("org.postgresql.Driver")
  case _ => None
}

除非你想管理异常。