nucliadb-utils 6.0.0.post2324__py3-none-any.whl → 6.0.0.post2340__py3-none-any.whl

nucliadb_utils/transaction.py

@@ -26,6 +26,7 @@ from typing import Any, Optional, Union
 import nats
 import nats.errors
 from nats.aio.client import Client
+from nats.aio.errors import ErrConnectionClosed
 from nats.js.client import JetStreamContext
 
 from nucliadb_protos.writer_pb2 import (
@@ -58,7 +59,7 @@ class MaxTransactionSizeExceededError(Exception):
     pass
 
 
-class StreamingServerTimeoutError(Exception):
+class StreamingServerError(Exception):
     pass
 
 
@@ -199,7 +200,13 @@ class TransactionUtility:
         request_id = uuid.uuid4().hex
 
         if wait:
-            waiting_event = await self.wait_for_commited(writer.kbid, waiting_for, request_id=request_id)
+            try:
+                waiting_event = await self.wait_for_commited(
+                    writer.kbid, waiting_for, request_id=request_id
+                )
+            except ErrConnectionClosed:
+                logger.error("Connection to NATS is closed")
+                raise StreamingServerError()
 
         if target_subject is None:
             target_subject = const.Streams.INGEST.subject.format(partition=partition)
@@ -217,7 +224,7 @@ class TransactionUtility:
                     "resource": writer.uuid,
                 },
             )
-            raise StreamingServerTimeoutError() from ex
+            raise StreamingServerError() from ex
 
         waiting_for.seq = res.seq
 

{nucliadb_utils-6.0.0.post2324.dist-info → nucliadb_utils-6.0.0.post2340.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nucliadb_utils
-Version: 6.0.0.post2324
+Version: 6.0.0.post2340
 Home-page: https://nuclia.com
 License: BSD
 Classifier: Development Status :: 4 - Beta
@@ -24,8 +24,8 @@ Requires-Dist: PyNaCl
 Requires-Dist: pyjwt>=2.4.0
 Requires-Dist: memorylru>=1.1.2
 Requires-Dist: mrflagly>=0.2.9
-Requires-Dist: nucliadb-protos>=6.0.0.post2324
-Requires-Dist: nucliadb-telemetry>=6.0.0.post2324
+Requires-Dist: nucliadb-protos>=6.0.0.post2340
+Requires-Dist: nucliadb-telemetry>=6.0.0.post2340
 Provides-Extra: cache
 Requires-Dist: redis>=4.3.4; extra == "cache"
 Requires-Dist: orjson>=3.6.7; extra == "cache"

{nucliadb_utils-6.0.0.post2324.dist-info → nucliadb_utils-6.0.0.post2340.dist-info}/RECORD

@@ -15,7 +15,7 @@ nucliadb_utils/run.py,sha256=HpAIM8xbR7UpVC2_7xOjB4fYbUVykyPP6yHrv2RD3DI,1707
 nucliadb_utils/settings.py,sha256=BOuYTh08cJe7UgBgU9ALHCJm6eDezbdPsdAXM4HGEgo,8144
 nucliadb_utils/signals.py,sha256=JRNv2y9zLtBjOANBf7krGfDGfOc9qcoXZ6N1nKWS2FE,2674
 nucliadb_utils/store.py,sha256=kQ35HemE0v4_Qg6xVqNIJi8vSFAYQtwI3rDtMsNy62Y,890
-nucliadb_utils/transaction.py,sha256=6N2JyUKqH5MCaWnWSR7_fFV4ky_OAbIEtY5hdOw6Z7Q,7740
+nucliadb_utils/transaction.py,sha256=GjM754TU9940pfQY3vxAQAYlQdBpqfwrO-_gYkSjWrI,7979
 nucliadb_utils/utilities.py,sha256=1C5o6WcOupoTfBRn6dgHc-XTZ2bsG397JVQCKy6swBo,15832
 nucliadb_utils/aiopynecone/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb_utils/aiopynecone/client.py,sha256=MPyHnDXwhukJr7U3CJh7BpsekfSuOkyM4g5b9LLtzc8,22941
@@ -59,8 +59,8 @@ nucliadb_utils/tests/indexing.py,sha256=YW2QhkhO9Q_8A4kKWJaWSvXvyQ_AiAwY1VylcfVQ
 nucliadb_utils/tests/local.py,sha256=fXIBasrvdaFJM-sw2wk1_oiFzBcm9O10iCyC-OiXwY8,1914
 nucliadb_utils/tests/nats.py,sha256=xqpww4jZjTKY9oPGlJdDJG67L3FIBQsa9qDHxILR8r8,7687
 nucliadb_utils/tests/s3.py,sha256=pl-RJFjA4MH6iXkqhsh5g8gDuEhrYu1nPZ-laxlrMlE,3704
-nucliadb_utils-6.0.0.post2324.dist-info/METADATA,sha256=uWKhL1TjSgnv74rbdL8tpow_SCy-QRtMs0_SIo75GMQ,2055
-nucliadb_utils-6.0.0.post2324.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-nucliadb_utils-6.0.0.post2324.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-6.0.0.post2324.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-6.0.0.post2324.dist-info/RECORD,,
+nucliadb_utils-6.0.0.post2340.dist-info/METADATA,sha256=MbR48tlmViGnpwl5Hjhb3jXCV2NQZ3ohvzZCgvS6yQs,2055
+nucliadb_utils-6.0.0.post2340.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+nucliadb_utils-6.0.0.post2340.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-6.0.0.post2340.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-6.0.0.post2340.dist-info/RECORD,,