3

我有两个Option

val name: Option[String] = ...
val shared: Option[Boolean] = ...

如果上述值是,我想构建一个UPDATE查询这些字段,但如果它们是,则保持它们不变。SETSomeNone

我已经设法做到了这一点,但我不太热衷于检查逗号,如果我需要添加额外的列,它不会按比例缩放。我也不太热衷于使用var.

 var query = Q.u + "UPDATE wishlist SET"
 if(name.isDefined)  query = query + " name = " +? name + (if (share.isDefined) ", " else "")
 if(shared.isDefined) query = query + " shared = " +? share

如果我不关心 SQL 注入,我可以执行以下操作:

 val fields = List(
   name.map(n => s"name = '$n'"),
   shared.map(e => s"shared = $e")
 ).flatten

 s"UPDATE wishlist SET ${fields.mkString(", ")}"

问题:有没有更好的方法来使用 Slick 的普通 SQL/String 插值来做到这一点,或者是我唯一的选择来提升嵌入式?

4

1 回答 1

1

它不是超级优雅,但我认为它至少是灵活的,因为它将扩展到支持许多输入而不改变底层逻辑:

 val name:Option[String] = Some("foo")
 val share:Option[String] = Some("bar")

 val sets = List(
   name.map(n => (" name = ", n)),
   share.map(s => (" shared = ", s))
 ).flatten

 val query2 = sets.zipWithIndex.foldLeft(Q.u + "UPDATE wishlist SET"){
   case (q,(p,0)) => q + p._1 +? p._2
   case (q,(p,i)) => q + "," + p._1 +? p._2
 }

我将字段名称和值对作为Tuples 的选项放入列表中,然后展平以删除Nones。然后,我将折叠以生成最终语句,并考虑到超过第一部分的任何查询部分都需要一个逗号。

于 2013-08-09T17:52:58.243 回答