使用巧妙的字符串插值法有条件地更新字段 [英] Conditonally UPDATE fields with Slick String interpolation

问题描述

我有两个Option：

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

我想构建一个UPDATE查询，如果上述值为Some，则SET这些字段保持不变。

我已经设法做到了这一点，但我对逗号的检查不是很感兴趣，如果我需要添加额外的列，它将不会扩展。我也不太喜欢使用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/字符串内插来实现这一点，或者我唯一的选择是移动到Lift嵌入吗？

推荐答案

它不是特别优雅，但我认为它至少是灵活的，因为它将扩展到支持许多输入，但不会更改底层逻辑：

 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的选项放入列表中，然后展开以删除Nones。然后我折叠以生成最后一条语句，同时考虑到查询的第一部分之后的任何部分都需要逗号。

这篇关于使用巧妙的字符串插值法有条件地更新字段的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！