nucliadb-utils 6.4.0.post4171__py3-none-any.whl → 6.4.0.post4176__py3-none-any.whl

nucliadb_utils/nats.py

@@ -21,7 +21,7 @@ import asyncio
 import logging
 import sys
 import time
-from functools import cached_property
+from functools import cached_property, partial
 from typing import Any, Awaitable, Callable, Optional, Union
 
 import nats
@@ -348,13 +348,18 @@ class NatsConnectionManager:
         durable: Optional[str] = None,
         config: Optional[nats.js.api.ConsumerConfig] = None,
     ) -> JetStreamContext.PullSubscription:
+        wrapped_cb: Callable[[Msg], Awaitable[None]]
+        if isinstance(self.js, JetStreamContextTelemetry):
+            wrapped_cb = partial(self.js.trace_pull_subscriber_message, cb)
+        else:
+            wrapped_cb = cb
+
         psub = await self.js.pull_subscribe(
             subject,
-            durable=durable,  # type: ignore
+            durable=durable,
             stream=stream,
-            config=config,  # type: ignore
+            config=config,
         )
-
         cancelled = asyncio.Event()
 
         async def consume(psub: JetStreamContext.PullSubscription, subject: str):
@@ -372,7 +377,7 @@ class NatsConnectionManager:
                         self.pull_utilization_metrics.inc({"status": "waiting"}, received - start_wait)
 
                     for message in messages:
-                        await cb(message)
+                        await wrapped_cb(message)
 
                     if self.pull_utilization_metrics:
                         processed = time.monotonic()

{nucliadb_utils-6.4.0.post4171.dist-info → nucliadb_utils-6.4.0.post4176.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb_utils
-Version: 6.4.0.post4171
+Version: 6.4.0.post4176
 Summary: NucliaDB util library
 Author-email: Nuclia <nucliadb@nuclia.com>
 License: AGPL
@@ -27,8 +27,8 @@ Requires-Dist: nats-py[nkeys]>=2.6.0
 Requires-Dist: PyNaCl
 Requires-Dist: pyjwt>=2.4.0
 Requires-Dist: mrflagly>=0.2.9
-Requires-Dist: nucliadb-protos>=6.4.0.post4171
-Requires-Dist: nucliadb-telemetry>=6.4.0.post4171
+Requires-Dist: nucliadb-protos>=6.4.0.post4176
+Requires-Dist: nucliadb-telemetry>=6.4.0.post4176
 Provides-Extra: cache
 Requires-Dist: redis>=4.3.4; extra == "cache"
 Requires-Dist: orjson>=3.6.7; extra == "cache"

{nucliadb_utils-6.4.0.post4171.dist-info → nucliadb_utils-6.4.0.post4176.dist-info}/RECORD

@@ -7,7 +7,7 @@ nucliadb_utils/exceptions.py,sha256=y_3wk77WLVUtdo-5FtbBsdSkCtK_DsJkdWb5BoPn3qo,
 nucliadb_utils/featureflagging.py,sha256=SYqr31e0_NeOMx6dZgSiDO9VJiV6xxYhM_gWkk8ITTQ,2386
 nucliadb_utils/grpc.py,sha256=apu0uePnkGHCAT7GRQ9YZfRYyFj26kJ440i8jitbM3U,3314
 nucliadb_utils/helpers.py,sha256=nPw8yod3hP-pxq80VF8QC36s7ygSg0dBUdfI-LatvCs,1600
-nucliadb_utils/nats.py,sha256=T5A_pH0koGPNH9IPNPv1LsYGnAS5hnJgN8VuNCqO_p8,15989
+nucliadb_utils/nats.py,sha256=uMs5dTaAplBuj0D0NFFmXGmobrnl4N3FzhCMa-YsnFc,16204
 nucliadb_utils/partition.py,sha256=jBgy4Hu5Iwn4gjbPPcthSykwf-qNx-GcLAIwbzPd1d0,1157
 nucliadb_utils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nucliadb_utils/run.py,sha256=Es0_Bu5Yc-LWczvwL6gzWqSwC85RjDCk-0oFQAJi9g4,1827
@@ -57,7 +57,7 @@ nucliadb_utils/tests/gcs.py,sha256=MBMzn_UHU5SU6iILuCsB5zU4umhNcaCw_MKrxZhwvOc,4
 nucliadb_utils/tests/local.py,sha256=cxIfPrKuqs5Ef0nbrVYQQAH2mwc4E0iD9bC2sWegS-c,1934
 nucliadb_utils/tests/nats.py,sha256=RWHjwqq5esuO7OFbP24yYX1cXnpPLcWJwDUdmwCpH28,1897
 nucliadb_utils/tests/s3.py,sha256=DACUh3HvgH3BchKFZ9R7RFUzsrg3v9A-cxTcXx4nmvA,3734
-nucliadb_utils-6.4.0.post4171.dist-info/METADATA,sha256=kcPzFKSAR8qvmsclZQgjIb33iOFL9WcO1k_zaYJ7m_s,2205
-nucliadb_utils-6.4.0.post4171.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
-nucliadb_utils-6.4.0.post4171.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-6.4.0.post4171.dist-info/RECORD,,
+nucliadb_utils-6.4.0.post4176.dist-info/METADATA,sha256=xc81R9dPXXlDQ26yTGi9u_XxsvpD9J4S8xavW7Snfs0,2205
+nucliadb_utils-6.4.0.post4176.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
+nucliadb_utils-6.4.0.post4176.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-6.4.0.post4176.dist-info/RECORD,,