zio.dynamodb.DynamoDBExecutor.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of zio-dynamodb_2.13 Show documentation
zio-dynamodb
The newest version!
package zio.dynamodb

import zio.aws.dynamodb.DynamoDb
import zio.stm.{ STM, TMap }
import zio.{ Ref, ULayer, URLayer, ZIO, ZLayer }

trait DynamoDBExecutor {
  def execute[A](query: DynamoDBQuery[_, A]): ZIO[Any, DynamoDBError, A]
}

object DynamoDBExecutor {
  val live: URLayer[DynamoDb, DynamoDBExecutor] =
    ZLayer.fromZIO(for {
      db <- ZIO.service[DynamoDb]
    } yield DynamoDBExecutorImpl(db))

  lazy val test: ULayer[DynamoDBExecutor with TestDynamoDBExecutor] = {
    val effect = for {
      ref  <- Ref.make(List.empty[DynamoDBQuery[_, _]])
      test <- (for {
                  tableMap       <- TMap.empty[TableName, TMap[PrimaryKey, Item]]
                  tablePkNameMap <- TMap.empty[TableName, String]
                } yield TestDynamoDBExecutorImpl(ref, tableMap, tablePkNameMap)).commit
    } yield test
    ZLayer.fromZIO(effect)
  }

  def test(tableAndPKNames: (String, String)*): ULayer[DynamoDBExecutor with TestDynamoDBExecutor] = {
    val effect = for {
      ref  <- Ref.make(List.empty[DynamoDBQuery[_, _]])
      test <- (for {
                  tableMap       <- TMap.empty[TableName, TMap[PrimaryKey, Item]]
                  tablePkNameMap <- TMap.empty[TableName, String]
                  _              <- STM.foreach(tableAndPKNames) {
                                      case (tableName, pkFieldName) =>
                                        for {
                                          _           <- tablePkNameMap.put(TableName(tableName), pkFieldName)
                                          pkToItemMap <- TMap.empty[PrimaryKey, Item]
                                          _           <- tableMap.put(TableName(tableName), pkToItemMap)
                                        } yield ()
                                    }
                } yield TestDynamoDBExecutorImpl(ref, tableMap, tablePkNameMap)).commit
    } yield test
    ZLayer.fromZIO(effect)
  }

}