Armeria Bug

bug.scala

  test("requestLog never completed with empty streaming response") {
    import scala.concurrent.ExecutionContext.Implicits.global
    val closed = new AtomicBoolean(false)
    val service = new HttpService {
      override def serve(ctx: ServiceRequestContext, req: HttpRequest): HttpResponse = {
        val rep = HttpResponse.streaming()
        rep.close()
        rep
      }
    }

    val server = Server.builder()
      .port(0, SessionProtocol.HTTP)
      .decorator { (d, ctx, r) =>
        ctx.log().whenComplete().toScala.onComplete { _ =>
          closed.set(true)
        }
        d.serve(ctx, r)
      }
      .serviceUnder("/empty", service)
      .build()

    Await.ready(server.start().toScala, 100.seconds)

    val uri = s"h2://localhost:${server.activeLocalPort()}"

    val client = WebClient
      .builder(uri)
      .build()

    val req = HttpRequest.of(HttpMethod.GET, "/empty/")
    Await.ready(client.execute(req).aggregate().toScala, 100.seconds)
    assert(closed.get() == true)
  }