add response status codes

Author: miguelgrinberg

elastic_transport/_async_transport.py

@@ -366,6 +366,7 @@ async def perform_request(  # type: ignore[override, return]
                 # We either got a response we're happy with or
                 # we've exhausted all of our retries so we return it.
                 if not retry or attempt >= max_retries:
+                    otel_span.set_db_response(resp.meta.status)
                     return TransportApiResponse(resp.meta, body)
                 else:
                     _logger.warning(

elastic_transport/_otel.py

@@ -92,3 +92,9 @@ def set_db_statement(self, serialized_body: bytes) -> None:
             self.otel_span.set_attribute(
                 "db.query.text", serialized_body.decode("utf-8")
             )
+
+    def set_db_response(self, status_code: int) -> None:
+        if self.otel_span is None:
+            return
+
+        self.otel_span.set_attribute("db.response.status_code", str(status_code))

elastic_transport/_transport.py

@@ -444,6 +444,7 @@ def perform_request(  # type: ignore[return]
                 # We either got a response we're happy with or
                 # we've exhausted all of our retries so we return it.
                 if not retry or attempt >= max_retries:
+                    otel_span.set_db_response(resp.meta.status)
                     return TransportApiResponse(resp.meta, body)
                 else:
                     _logger.warning(

tests/test_otel.py

@@ -74,6 +74,7 @@ def test_detailed_span():
                 "X-Found-Handling-Instance": "instance-0000000001",
             }
         )
+        span.set_db_response(202)
 
     spans = memory_exporter.get_finished_spans()
     assert len(spans) == 1
@@ -88,6 +89,7 @@ def test_detailed_span():
         "http.request.method": "POST",
         "db.namespace": "e9106fc68e3044f0b1475b04bf4ffd5f",
         "elasticsearch.node.name": "instance-0000000001",
+        "db.response.status_code": "202",
     }