calasql-query_2.13.0.1.9.source-code.GetGeneratedKeys.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scalasql-query_2.13 Show documentation

scalasql-query

There is a newer version: 0.1.8

package scalasql.query

import scalasql.core.SqlStr.Renderable
import scalasql.core.{Context, Expr, Queryable, SqlStr, WithSqlExpr}

/**
 * Represents an [[Insert]] query that you want to call `JdbcStatement.getGeneratedKeys`
 * on to retrieve any auto-generated primary key values from the results
 */
trait GetGeneratedKeys[Q, R] extends Query[Seq[R]] {
  def single: Query.Single[R] = new Query.Single(this)
}

object GetGeneratedKeys {

  class Impl[Q, R](base: Returning.InsertBase[Q])(implicit qr: Queryable.Row[?, R])
      extends GetGeneratedKeys[Q, R] {

    def expr = WithSqlExpr.get(base)
    override protected def queryConstruct(args: Queryable.ResultSetIterator): Seq[R] = {
      Seq(qr.construct(args))
    }

    protected def queryWalkLabels(): Seq[List[String]] = Nil
    protected def queryWalkExprs(): Seq[Expr[?]] = Nil
    protected override def queryIsSingleRow = false
    protected override def queryIsExecuteUpdate = true

    override private[scalasql] def renderSql(ctx: Context): SqlStr = Renderable.renderSql(base)(ctx)

    override protected def queryGetGeneratedKeys: Option[Queryable.Row[?, ?]] = Some(qr)
  }
}