Include response length by default in the network spans

nachoBonafonte · Sep 28, 2023 · 4798a27 · 4798a27
1 parent 8d6fa74
commit 4798a27
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/Sources/Instrumentation/URLSession/URLSessionLogger.swift b/Sources/Instrumentation/URLSession/URLSessionLogger.swift
@@ -107,9 +107,16 @@ class URLSessionLogger {
         }
 
         let statusCode = httpResponse.statusCode
-        span.setAttribute(key: SemanticAttributes.httpStatusCode.rawValue, value: AttributeValue.int(statusCode))
+        span.setAttribute(key: SemanticAttributes.httpStatusCode.rawValue, 
+                          value: AttributeValue.int(statusCode))
         span.status = statusForStatusCode(code: statusCode)
 
+        if let contentLengthHeader = httpResponse.value(forHTTPHeaderField: "Content-Length"),
+           let contentLength = Int(contentLengthHeader) {
+            span.setAttribute(key: SemanticAttributes.httpResponseContentLength.rawValue,
+                              value: AttributeValue.int(contentLength))
+        }
+
         instrumentation.configuration.receivedResponse?(response, dataOrFile, span)
         span.end()
     }