nucliadb-telemetry 6.3.5.post3972__py3-none-any.whl → 6.3.5.post3980__py3-none-any.whl

nucliadb_telemetry/jetstream.py

@@ -32,7 +32,6 @@ from opentelemetry.trace import SpanKind, Tracer
 
 from nucliadb_telemetry import logger, metrics
 from nucliadb_telemetry.common import set_span_exception
-from nucliadb_telemetry.utils import get_telemetry
 
 msg_consume_time_histo = metrics.Histogram(
     # time it takes from when msg was queue to when it finished processing
@@ -210,7 +209,7 @@ class JetStreamContextTelemetry:
         return await self.js.consumer_info(stream, consumer, timeout)
 
 
-class NatsClientTelemetry(Client):
+class NatsClientTelemetry:
     def __init__(self, nc: Client, service_name: str, tracer_provider: TracerProvider):
         self.nc = nc
         self.service_name = service_name
@@ -239,17 +238,17 @@ class NatsClientTelemetry(Client):
     async def publish(
         self,
         subject: str,
-        body: bytes = b"",
-        reply: str = "",
+        body: bytes,
         headers: Optional[Dict[str, str]] = None,
-    ) -> None:
+        **kwargs,
+    ):
         tracer = self.tracer_provider.get_tracer(f"{self.service_name}_nc_publisher")
         headers = {} if headers is None else headers
         inject(headers)
 
         with start_span_message_publisher(tracer, subject) as span:
             try:
-                await self.nc.publish(subject, body, reply, headers)
+                await self.nc.publish(subject, body, headers=headers, **kwargs)
                 msg_sent_counter.inc({"subject": subject, "status": metrics.OK})
             except Exception as error:
                 set_span_exception(span, error)
@@ -275,24 +274,3 @@ class NatsClientTelemetry(Client):
                 raise error
 
         return result
-
-
-def get_traced_nats_client(nc: Client, service_name: str) -> Union[Client, NatsClientTelemetry]:
-    tracer_provider = get_telemetry(service_name)
-    if tracer_provider is not None:
-        return NatsClientTelemetry(nc, service_name, tracer_provider)
-    else:
-        return nc
-
-
-def get_traced_jetstream(
-    nc: Union[Client, NatsClientTelemetry], service_name: str
-) -> Union[JetStreamContext, JetStreamContextTelemetry]:
-    jetstream = nc.jetstream()
-    tracer_provider = get_telemetry(service_name)
-
-    if tracer_provider is not None and jetstream is not None:  # pragma: no cover
-        logger.info(f"Configuring {service_name} jetstream with telemetry")
-        return JetStreamContextTelemetry(jetstream, service_name, tracer_provider)
-    else:
-        return jetstream

{nucliadb_telemetry-6.3.5.post3972.dist-info → nucliadb_telemetry-6.3.5.post3980.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb_telemetry
-Version: 6.3.5.post3972
+Version: 6.3.5.post3980
 Summary: NucliaDB Telemetry Library Python process
 Author-email: Nuclia <nucliadb@nuclia.com>
 License: AGPL

{nucliadb_telemetry-6.3.5.post3972.dist-info → nucliadb_telemetry-6.3.5.post3980.dist-info}/RECORD

@@ -7,7 +7,7 @@ nucliadb_telemetry/grpc.py,sha256=1CdueeJedxManSUQKwNOvqUuFiVhnMiZViXUJm81bIY,14
 nucliadb_telemetry/grpc_metrics.py,sha256=L2XMcJDxioNUYPX_IbKl_zvrG8uOFpF_a67c6yJq-Jw,5174
 nucliadb_telemetry/grpc_sentry.py,sha256=3eS_DVIX5MzvbJBkykf8JyZUf1b_b9w1KLhjoiW7oV0,2659
 nucliadb_telemetry/jaeger.py,sha256=HMQK8x_oaol5tCx3hCjaMG5Uzvsy_WWhKnD-FEZoYA4,7036
-nucliadb_telemetry/jetstream.py,sha256=P5lz5EothzO7BbGiSa7gXnfm57fY7O9SX-OchBoeqUU,10834
+nucliadb_telemetry/jetstream.py,sha256=E2_SY5-8ljhZAY0n_q-qlvewUmi2kF2En3mcqyA26kY,9977
 nucliadb_telemetry/logs.py,sha256=ag1e7tKEoNFgB1FuAmyTxdIHtTFwOJlGxPl2s1bhUic,9311
 nucliadb_telemetry/metrics.py,sha256=jSGecUI7qNbPm3OEzo11v4Zo-lUhXhZrhkWPtX1eGf8,8022
 nucliadb_telemetry/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -30,7 +30,7 @@ nucliadb_telemetry/tests/grpc/helloworld_pb2.py,sha256=ht4dmi0pAy6qDrwcjkbtSf_hO
 nucliadb_telemetry/tests/grpc/helloworld_pb2.pyi,sha256=mBZCQE6z7riQw6fOegJgZ5lHmXqO23_zxL8iQMqSqms,1191
 nucliadb_telemetry/tests/grpc/helloworld_pb2_grpc.py,sha256=_jxUNxl4Fx-JztK9RO5R6osjNP2sVNVPAxLnmczEYOc,2677
 nucliadb_telemetry/tests/grpc/helloworld_pb2_grpc.pyi,sha256=Y6teCx-PhPU-rI6w5ItLBKaTb34FLpngPnuDVWtNve4,958
-nucliadb_telemetry-6.3.5.post3972.dist-info/METADATA,sha256=m8uZWXhINq7VsJvq7Q65ajOZyofwURRdNWnovFhZyy8,11022
-nucliadb_telemetry-6.3.5.post3972.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-nucliadb_telemetry-6.3.5.post3972.dist-info/top_level.txt,sha256=3qEHI_5ttqQIL2gkNYwSlKsFyBBiEzDiIy9UKISzOaQ,19
-nucliadb_telemetry-6.3.5.post3972.dist-info/RECORD,,
+nucliadb_telemetry-6.3.5.post3980.dist-info/METADATA,sha256=7e6JsmxzyJvLRO_M1I_HNHvXFB5IWZefMI1MIb9erDo,11022
+nucliadb_telemetry-6.3.5.post3980.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+nucliadb_telemetry-6.3.5.post3980.dist-info/top_level.txt,sha256=3qEHI_5ttqQIL2gkNYwSlKsFyBBiEzDiIy9UKISzOaQ,19
+nucliadb_telemetry-6.3.5.post3980.dist-info/RECORD,,