typo.dsl.DeleteBuilder.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of typo-dsl-zio-jdbc_3 Show documentation
Typed postgres boilerplate generation
The newest version!
package typo.dsl

import zio.ZIO
import zio.jdbc.*

import scala.annotation.nowarn

trait DeleteBuilder[Fields, Row] {
  protected def params: DeleteParams[Fields]
  protected def withParams(sqlParams: DeleteParams[Fields]): DeleteBuilder[Fields, Row]

  final def where[N[_]: Nullability](v: Fields => SqlExpr[Boolean, N]): DeleteBuilder[Fields, Row] =
    whereStrict(f => v(f).?.coalesce(false))

  final def whereStrict(v: Fields => SqlExpr[Boolean, Required]): DeleteBuilder[Fields, Row] =
    withParams(params.where(v))

  def sql: Option[SqlFragment]
  def execute: ZIO[ZConnection, Throwable, Long]
}

object DeleteBuilder {
  def apply[Fields, Row](name: String, structure: Structure.Relation[Fields, Row]): DeleteBuilderSql[Fields, Row] =
    DeleteBuilderSql(name, structure, DeleteParams.empty)

  final case class DeleteBuilderSql[Fields, Row](
      name: String,
      structure: Structure.Relation[Fields, Row],
      params: DeleteParams[Fields]
  ) extends DeleteBuilder[Fields, Row] {
    override def withParams(sqlParams: DeleteParams[Fields]): DeleteBuilder[Fields, Row] =
      copy(params = sqlParams)

    def mkSql(ctx: RenderCtx): SqlFragment = {
      List[Iterable[SqlFragment]](
        Some(SqlFragment.deleteFrom(name)),
        params.where
          .map(w => w(structure.fields))
          .reduceLeftOption(_.and(_))
          .map { where => sql" where " ++ where.render(ctx) }
      ).flatten.reduce(_ ++ _)
    }

    override def sql: Option[SqlFragment] =
      Some(mkSql(RenderCtx.Empty))

    override def execute: ZIO[ZConnection, Throwable, Long] =
      mkSql(RenderCtx.Empty).update
  }

  final case class DeleteBuilderMock[Id, Fields, Row](
      params: DeleteParams[Fields],
      structure: Structure[Fields, Row],
      map: scala.collection.mutable.Map[Id, Row]
  ) extends DeleteBuilder[Fields, Row] {
    override def withParams(sqlParams: DeleteParams[Fields]): DeleteBuilder[Fields, Row] =
      copy(params = sqlParams)

    override def sql: Option[SqlFragment] = None

    override def execute: ZIO[ZConnection, Throwable, Long] = ZIO.succeed {
      var changed: Long = 0L
      map.foreach { case (id, row) =>
        if (params.where.forall(w => structure.untypedEval(w(structure.fields), row))) {
          map.remove(id): @nowarn
          changed += 1
        }
      }
      changed
    }
  }
}