Otel : Don't read response body size if the content-type is text/even…

…t-stream, fixing SSE
gofiber · Jun 4, 2024 · eaae8b4 · eaae8b4
1 parent bb80ed2
commit eaae8b4
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/otelfiber/fiber.go b/otelfiber/fiber.go
@@ -132,10 +132,13 @@ func Middleware(opts ...Option) fiber.Handler {
 			semconv.HTTPAttributesFromHTTPStatusCode(c.Response().StatusCode()),
 			semconv.HTTPRouteKey.String(c.Route().Path), // no need to copy c.Route().Path: route strings should be immutable across app lifecycle
 		)
-
+
+		var responseSize int64
 		requestSize := int64(len(c.Request().Body()))
-		responseSize := int64(len(c.Response().Body()))
-
+		if c.GetRespHeader("Content-Type") != "text/event-stream" {
+			responseSize = int64(len(c.Response().Body()))
+		}
+
 		defer func() {
 			responseMetricAttrs = append(
 				responseMetricAttrs,