Slick：如何将 SQL LIKE 语句与 SQL IN 语句结合起来

Question

我基本上想用更多的东西替换下面的代码"slicky":

final case class User(firstName: String, lastName: String)

def dbAction(lastNameParts: Seq[String]): SqlStreamingAction[Vector[User], User, Effect]
implicit val getUserResult =
  GetResult((r: PositionedResult) => {
    val resultSet: ResultSet = r.rs
    User(
      resultSet.getString(1),
      resultSet.getString(2)
    )
  })
val pattern = orgIds.mkString("|")
sql"""SELECT u.first_name, u.last_name
      FROM users u
      WHERE last_name ~* $pattern""".as[User]

所以结果 SQL 将是：

SELECT u.first_name, u.last_name
FROM users u
WHERE last_name ~* '%bar|baz%';

所以这个 dbAction 将 return 一个操作，我可以用它来查询列表中包含某些名称部分的所有用户。

所以

dbAction(Seq("bar", "baz"))

将 return 查询包含字符串 "bar" 或 "baz"（不区分大小写）的所有姓氏的操作。我找到了查询单个模式的方法

val query = for {
  user <- users if user.lastName like "%bar%"
} yield (user.firstName, user.lastName)

我找到了查询列表包含的方法

u <- users if u.lastName.inSet(Seq("bar", "baz"))

但找不到合并的方法

编辑：另一种可能解决该问题的方法是通过正则表达式。有没有办法实现类似下面的 SQL 语句：

select * from users where last_name ~ '[\w]*bar[\w]*|[\w]*baz[\w]*';

因为这在某种程度上是一个不同的问题如何使用正则表达式我为此创建了一个不同的问题：

Answer 1

只需结合这2个查询条件：

import slick.lifted.Tag
import slick.jdbc.H2Profile.api._
import scala.concurrent.duration._
import scala.concurrent.Await

object Test {
  final case class User(firstName:String, lastName:String, id:Long = 0l)

  class UserTable(tag: Tag) extends Table[User](tag, "user"){
    def id = column[Long]("id", O.PrimaryKey, O.AutoInc)
    def firstName = column[String]("first_name")
    def lastName = column[String]("last_name")
    def * = (firstName, lastName, id).mapTo[User]
  }



  def main(args:Array[String]):Unit = {
    val db = Database.forConfig("h2config")
    val users = TableQuery[UserTable]

    val initialData = Seq(
      User("Alex", "Arendar"),
      User("David", "Arora"),
      User("Dude", "Stoecki"),
      User("Alexander", "the Great")
    )

    Await.result(
      db.run(
        users.schema.create  andThen (users ++= initialData)
      ), 3 seconds
    )

    val query1 = for {
      user <- users if user.firstName like "%Alex%"
    } yield (user.firstName, user.lastName)

    println(query1.result.statements.head)
    println(Await.result(db.run(query1.result), 3 seconds))

    val query2 = for {
      user <- users if (user.firstName like "%Alex%") && user.firstName.inSet(Seq("Alex", "David"))
    } yield (user.firstName, user.lastName)

    println(query2.result.statements.head)
    println(Await.result(db.run(query2.result), 3 seconds))

    db.close()
  }

我的控制台输出是：

select "first_name", "last_name" from "user" where "first_name" like '%Alex%'
Vector((Alex,Arendar), (Alexander,the Great))
select "first_name", "last_name" from "user" where ("first_name" like '%Alex%') and ("first_name" in ('Alex', 'David'))
Vector((Alex,Arendar))

Process finished with exit code 0

如您所见，生成的 SQL 确实包含 like 和 in 部分。

我在这里使用了内存中的 H2 数据库，但我认为这适用于任何 RDBMS。

Answer 2

查看您发布的代码，我认为您不需要字面上将 IN 与 LIKE 结合起来。我读到这个问题是想做一个正则表达式查询。尽管 Slick 不支持开箱即用的 ~* 运算符，但您可以自己添加。这将为您提供一种使用 lifted embedded 样式的 Slick 查询来执行查询的方法。

为此，您可以使用 SimpleExpression 生成器。关于它的文档不多，但起点是参考手册的 Scalar Database Functions 页。

我们想要做的是按照这些思路编写一个方法：

def find(names: Seq[String]): DBIO[Seq[String]] = {
  val pattern = names.mkString("|")
  users.filter(_.lastName regexLike pattern).map(_.lastName).result
}

要获得 regexLike 我们可以丰富（增强，"pimp"）字符串列以具有 regexLike 方法：

implicit class RegexLikeOps(s: Rep[String]) {
  def regexLike(p: Rep[String]): Rep[Boolean] = {
    val expr = SimpleExpression.binary[String,String,Boolean] { (s, p, qb) =>
      qb.expr(s)
      qb.sqlBuilder += " ~* "
      qb.expr(p)
    }
    expr.apply(s,p)
  }
}

implicit class 部分允许编译器在任何时候构造 RegexLikeOps class 它有一个 Rep[String] 调用一个 Rep[String] 没有的方法'已经有（即，当要求 regexLike 时）。

我们的 regexLike 方法将另一个 Rep[String] 参数作为模式，然后使用 SimpleExpression 构建器安全地构建我们要使用的 SQL。

把它们放在一起我们可以写成：

val program = for {
  _ <- users.schema.create
  _ <- users ++= User("foo") :: User("baz") :: User("bar") :: Nil
  result <- find( Seq("baz","bar") )
} yield result

println( Await.result(db.run(program), 2.seconds) )

生成的 SQL（在我使用 H2 的测试中）是：

select "last_name" from "app_user" where "last_name" ~* 'baz|bar'

完整代码为：https://github.com/d6y/so46199828

Slick：如何将 SQL LIKE 语句与 SQL IN 语句结合起来

Slick: How can I combine a SQL LIKE statement with a SQL IN statement

scala

slick

sql-like

sql-in