anorm.ToStatement.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of anorm_3 Show documentation

anorm

The newest version!

/*
 * Copyright (C) from 2022 The Play Framework Contributors , 2011-2021 Lightbend Inc. 
 */

package anorm

import java.sql.PreparedStatement

/** Sets value as statement parameter. */
@annotation.implicitNotFound(
  "Cannot set value of type ${A} as parameter on statement: `anorm.ToStatement[${A}] required`"
)
trait ToStatement[A] extends ToStatementBase[A] { self =>

  /**
   * {{{
   * import anorm.ToStatement
   *
   * sealed trait MyEnum
   * case object Foo extends MyEnum
   * case object Bar extends MyEnum
   *
   * implicit val to: ToStatement[MyEnum] =
   *   ToStatement.of[Int].contramap[MyEnum] {
   *     case Foo => 1
   *     case Bar => 2
   *   }
   * }}}
   */
  final def contramap[B](f: B => A): ToStatement[B] =
    ToStatement[B] { (s, i, b) => self.set(s, i, f(b)) }

}

object ToStatement extends ToStatementConversions {
  private class FunctionalToStatement[T](f: (PreparedStatement, Int, T) => Unit) extends ToStatement[T] {
    def set(s: PreparedStatement, index: Int, v: T): Unit = { f(s, index, v) }
  }

  /** Functional factory */
  def apply[T](set: (PreparedStatement, Int, T) => Unit): ToStatement[T] =
    new FunctionalToStatement[T](set)
}