InsertIfNotExists.scala

InsertIfNotExists.scala

trait InsertIfNotExists[T <: AbstractTable[_]] {
  val driver: JdbcProfile

  import driver.api._

  val tableQuery: TableQuery[T]

  def exists(e: T#TableElementType): Rep[Boolean]

  def insertIfNotExists(e: T#TableElementType)(implicit executionContext: ExecutionContext): DBIO[Unit] = for {
    ex <- exists(e).result
    _ <- if(ex) DBIO.successful(()) else (tableQuery += e).map(_ => ())
  } yield ()
}

  import driver.api._

  trait InsertIfNotEmptyOps[T <: AbstractTable[_]] {
    def insertIfNotExists(e: T#TableElementType): DBIO[Unit]
    def insertIfNotExists(es: Seq[T#TableElementType]): DBIO[Unit]
    def +=?(e: T#TableElementType): DBIO[Unit]
    def ++=?(e: Seq[T#TableElementType]): DBIO[Unit]
  }

  implicit def upsertQuery[T <: AbstractTable[_]](
      tableQuery: TableQuery[T])(implicit u: InsertIfNotExists[T], ec: ExecutionContext) =
    new InsertIfNotEmptyOps[T] {
      def insertIfNotExists(e: T#TableElementType): DBIO[Unit] = u.insertIfNotExists(e)
      def insertIfNotExists(es: Seq[T#TableElementType]): DBIO[Unit] = DBIO.sequence(es.map(insertIfNotExists)).map(_ => ())
      def +=?(e: T#TableElementType): DBIO[Unit] = insertIfNotExists(e)
      def ++=?(es: Seq[T#TableElementType]): DBIO[Unit] = insertIfNotExists(es)
    }