zioCollaboratorFull.scala

package com.zio.pitfalls

import zio.console.putStrLn
import zio.{DefaultRuntime, Task, UIO, ZIO}

import scala.collection.mutable

case class NewOrderRequest() {
  def toOrder: Order = {Order()}
}
case class NewOrderResponse()
case class Order()

trait DAO {
  def insert(order: Order): Unit
  def size(): Int
}

object DAO {
  def make(): Task[DAO] = Task(new DAO {
    val orders = new mutable.ArrayBuffer[Order]()

    override def insert(order: Order): Unit = {
      orders.+=(order)}

    override def size(): Int = orders.size
  })
}

trait MyRpcService {
  def newOrder(request: NewOrderRequest): NewOrderResponse
}

object MyRpcService {
  def apply(daoZ: Task[DAO]): MyRpcService = {
    new MyRpcService{
      override def newOrder(request: NewOrderRequest): NewOrderResponse = Runtime.unsafeRun {
        for {
          dao <- daoZ
          _ <- ZIO.effect(dao.insert(request.toOrder))
          _ <- putStrLn(s"Added an Order.")
          currentSize <- ZIO.effect(dao.size())
          _ <- putStrLn(s"Amount of orders: $currentSize.")
        } yield NewOrderResponse()
      }
    }
  }
}

object RPCServerSimulator {
  def callEndpoints(rpcService: MyRpcService): Task[Unit] = {
    ZIO.effect(rpcService.newOrder(NewOrderRequest())) *>
      ZIO.effect(rpcService.newOrder(NewOrderRequest()))
  }
}

object MyApp extends App with DefaultRuntime {
  unsafeRun {
    for {
      service <- UIO(MyRpcService(DAO.make()))
      _ <- RPCServerSimulator.callEndpoints(service)
    } yield ()
  }
}

object Runtime extends DefaultRuntime