Question

创建表查询时，我想通过映射默认表查询来修改我的select语句。但是，我找不到一种方法来映射列的值并仍然映射到我的case类

case class MyRecord(id: Int, name: String, value: Int)

class MyTable(tag: Tag) extends Table[MyRecord](tag, "MYTABLE") {
    def id = column[Int]("id")
    def name = column[String]("name")
    def value = column[Int]("value")

    def * = (id, name, value) <> (MyRecord.tupled, MyRecord.unapply)
  }

lazy val tableQuery = TableQuery[MyTable]

我想用this function修改name的值：

def trimLeading0: (Rep[String]) => Rep[String] = SimpleExpression.unary[String, String] {
    (str, queryBuilder) =>
      import slick.util.MacroSupport._
      import queryBuilder._
      b"TRIM(LEADING 0 FROM $str)"
  }

现在我不知道该怎么做：

val trimmedTableQuery: Query[MyTable, MyRecord, Seq] = tableQuery.map(s => ???)

我已尝试映射Rep，就像我对案例类所做的那样：

val trimmedTableQuery = tableQuery.map(s => s.copy(name = trimLeading0(s.name)))

这拒绝使用value copy is not a member of MyTable

进行编译

我目前的解决方法是使用自定义函数代替MyRecord.tupled进行默认投影：

def trimming(t: (Int, String, Int)) = MyRecord(t._1, t._2.dropWhile(_ == "0"), t._3)
def * = (id, name, value) <> (trimming, MyRecord.unapply)

或者，我可以将返回元组的DBIOAction的返回结果映射到案例类，这不太优雅：

val action = tableQuery.map{ s => (s.id, trimLeading0(s.name), s.value)}.result
val futureTuples: Future[Seq[(Int, String, Int)]] = db.run(action)
val records = futureTuples map (s => s.map(MyRecord.tupled))

但是在构建查询时如何在map方法中执行此操作？或者更改def name列描述会更好吗？

Answer 1

你不能搞砸MyTable中的默认投影（即def *），因为它需要是对称的。它用于查询和插入。但是您可以基于MyTable的特化创建一个trimmedTableQuery，并使用重写的默认投影。然后，您还可以基于对称默认投影获得tableQuery。如果您尝试基于trimmedTableQuery进行插入，则会出现错误（但您不需要这样做，只需使用tableQuery进行插入）。

lazy val tableQuery = TableQuery[MyTable]
lazy val trimmedTableQuery = new TableQuery(new MyTable(_) {
  override def * = (id, trimLeading0(name), value) <> (MyRecord.tupled, MyRecord.unapply)
})

修改列值后，将Slick查询映射到默认投影

1 个答案: