Update Metric.scala

varshith257 · Dec 11, 2024 · 8252467 · 8252467
1 parent f5ccae4
commit 8252467
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/server/core/src/main/scala/sttp/tapir/server/metrics/Metric.scala b/server/core/src/main/scala/sttp/tapir/server/metrics/Metric.scala
@@ -36,8 +36,8 @@ case class MetricLabels(
   def namesForResponse: List[String] = forResponse.map { case (name, _) => name }
 
   def valuesForRequest(ep: AnyEndpoint, req: ServerRequest): List[String] = forRequest.map { case (_, f) => f(ep, req) }
-  def valuesForResponse(res: ServerResponse[_]): List[String] = forResponse.map { case (_, f) => f(Right(res)) }
-  def valuesForResponse(ex: Throwable): List[String] = forResponse.map { case (_, f) => f(Left(ex)) }
+  def valuesForResponse(res: ServerResponse[_]): List[String] = forResponse.flatMap { case (_, f) => f(Right(res)).toList }
+  def valuesForResponse(ex: Throwable): List[String] = forResponse.flatMap { case (_, f) => f(Left(ex)).toList }
 }
 
 object MetricLabels {
@@ -51,15 +51,15 @@ object MetricLabels {
     forResponse = List(
       "status" -> {
         case Right(r) =>
-          r.code match {
+          Some(r.code match {
             case c if c.isInformational => "1xx"
             case c if c.isSuccess       => "2xx"
             case c if c.isRedirect      => "3xx"
             case c if c.isClientError   => "4xx"
             case c if c.isServerError   => "5xx"
             case _                      => ""
-          }
-        case Left(_) => "5xx"
+          })
+        case Left(_) => Some("5xx")
       }
     )
   )