Experimenting with ZIO & sockets

zio_socket_server.scala

import scalaz.nio._
import scalaz.nio.channels.{AsynchronousServerSocketChannel, AsynchronousSocketChannel}
import scalaz.zio.console._
import scalaz.zio.{App, ZIO}


object Bracket extends App {

  override def run(args: List[String]): ZIO[Environment, Nothing, Int] = {
    theSocket.foldM(
      failure = err => putStrLn(s"Execution Failed with: $err") *> ZIO.succeed(1),
      success = _ => ZIO.succeed(0)
    )
  }

  val theSocket =
    for {
      address <- SocketAddress.inetSocketAddress("127.0.0.1", 1337)
      socket <- AsynchronousServerSocketChannel()
      _ <- socket.bind(address)
      connections <- socket.accept.bracket(_.close.orDie)(doWork).forever.fork
      worker <- connections.join
    } yield worker

  def doWork(work: AsynchronousSocketChannel): ZIO[Console, Exception, Unit] = {
    val w = for {
      chunk <- work.read(16)
      str = chunk.toArray.map(_.toChar).mkString
      _ <- putStrLn("content: " + str)
    } yield ()
    w.forever
  }
}

This program allows multiple connections to be created, but only the first connection is read from.