首页 文章

使用Slick String插值进行条件UPDATE字段

提问于
浏览
2

我有两个 Option

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

如果上面的值是 Some ,我想构建 UPDATE 这些字段的 UPDATE 查询,但如果它们是 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注入,我可以做一些事情:

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

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

Question: 使用Slick的普通SQL /字符串插值是否有更好的方法可以做到这一点,或者是我唯一可以选择移植到嵌入式嵌入式?

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
     }
    

    我将字段名称和值对放入列表作为 Tuple 的选项,然后展平以删除 None . 我正在折叠以产生最终陈述,考虑到在第一篇文章之后的任何一条查询都需要一个逗号 .

相关问题