我有两个选项
:
val name: Option[String] = ...
val shared: Option[Boolean] = ...
我想构建一个 UPDATE
查询,如果上述值为 Some
,则 SET
设置这些字段,但如果它们保持不变,则保持不变是 None
。
我已经设法做到了这一点,但我不太热衷于检查逗号,如果我需要添加额外的列,它不会扩展。我也不太热衷于使用 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 注入(inject),我可以按照以下方式做一些事情:
val fields = List(
name.map(n => s"name = '$n'"),
shared.map(e => s"shared = $e")
).flatten
s"UPDATE wishlist SET ${fields.mkString(", ")}"
问题:有没有更好的方法可以使用 Slick 的纯 SQL/字符串插值来实现这一点,或者是我转向提升嵌入式的唯一选择?
最佳答案
它不是非常优雅,但我认为它至少是灵活的,因为它可以扩展以支持许多输入而无需改变底层逻辑:
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
}
我将字段名称和值对作为 Tuple
的选项放入列表中,然后展平以删除 None
。然后我折叠生成最终语句,考虑到第一部分之后的任何查询部分都需要一个逗号。
关于scala - 使用 Slick String 插值有条件地更新字段,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/18152034/