ZIO learning - zhttp - playing with http server / published by https://github.com/dacr/code-examples-manager #df382d02-6f01-417e-bf2a-b8c4a7e401d9/2d99b1ee26ceee29f960a96ebfb191dfd212561d

zio-learning-zhttp-server-2.sc

// summary : ZIO learning - zhttp - playing with http server
// keywords : scala, zio, learning, pure-functional, async, http-server, zhttp, @testable, @exclusive
// publish : gist
// authors : David Crosson
// license : Apache NON-AI License Version 2.0 (https://raw.githubusercontent.com/non-ai-licenses/non-ai-licenses/main/NON-AI-APACHE2)
// id : df382d02-6f01-417e-bf2a-b8c4a7e401d9
// created-on : 2022-01-15T08:48:21+01:00
// managed-by : https://github.com/dacr/code-examples-manager
// run-with : scala-cli $file
// test-with : curl http://127.0.0.1:8080/uuid

// ---------------------
//> using scala "3.4.2"
//> using dep "dev.zio::zio-http:3.0.0-RC1"
// ---------------------

import zio.*, zio.http.*, zio.ZIO.succeed

object Main extends ZIOAppDefault {

  val routes = Http.collectZIO[Request] {
    case Method.GET -> root          => succeed(Response.text("Hello World!"))
    case Method.GET -> root / "json" => succeed(Response.json("""{"greetings": "Hello World!"}"""))
    case Method.GET -> root / "uuid" => Random.nextUUID.map(uuid => Response.text(uuid.toString))
  }

  val serverLogic = for {
    config <- ZIO.config(Server.Config.config)
    port    = config.address.getPort
    _      <- Console.printLine(s"Server listening on http://127.0.0.1:$port/")
    _      <- Server.serve(routes.withDefaultErrorResponse)
  } yield ()

  override def run =
    for {
      args         <- getArgs
      listenOn      = args.headOption.map(_.toInt).getOrElse(8080)
      autoExitDelay = args.drop(1).headOption.map(_.toInt).getOrElse(5)
      _            <- Console.printLine(s"Auto exiting in $autoExitDelay seconds")
      _            <- serverLogic.timeout(autoExitDelay.seconds).provide(Server.defaultWithPort(listenOn))
    } yield ()
}

Main.main(args)