Update ZioMetrics.scala

varshith257 · Dec 11, 2024 · 9da5621 · 9da5621
1 parent c81aee7
commit 9da5621
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/metrics/zio-metrics/src/main/scala/sttp/tapir/server/metrics/zio/ZioMetrics.scala b/metrics/zio-metrics/src/main/scala/sttp/tapir/server/metrics/zio/ZioMetrics.scala
@@ -87,8 +87,8 @@ object ZioMetrics {
 
   /** Convert into zio metric labels */
   private def asZioLabel(l: MetricLabels, res: Either[Throwable, ServerResponse[_]], phase: Option[String]): Set[MetricLabel] = {
-    val responseLabels = l.forResponse.flatMap { case (key, valueFn) =>
-      valueFn(res).map(value => MetricLabel(key, value))
+    val responseLabels = l.forResponse.map { case (key, valueFn) =>
+      MetricLabel(key, valueFn(res).getOrElse("unknown"))
     }
     val phaseLabel = phase.map(v => MetricLabel(l.forResponsePhase.name, v))
     (responseLabels ++ phaseLabel).toSet