UncountablePythonSDK 0.0.82__py3-none-any.whl → 0.0.83__py3-none-any.whl

{UncountablePythonSDK-0.0.82.dist-info → UncountablePythonSDK-0.0.83.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.82
+Version: 0.0.83
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{UncountablePythonSDK-0.0.82.dist-info → UncountablePythonSDK-0.0.83.dist-info}/RECORD

@@ -106,7 +106,7 @@ uncountable/integration/queue_runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeu
 uncountable/integration/queue_runner/job_scheduler.py,sha256=lhvcl11jyk_wg9BSc1Xyh5u8iEtsdqQFW83FPLupjQI,6189
 uncountable/integration/queue_runner/queue_runner.py,sha256=0BmYu5zHdothTevGsB-nXg6MBd1UD-WkP3h1WCKMdQg,710
 uncountable/integration/queue_runner/types.py,sha256=8qTq29BTSa5rmW6CBlBntP0pNIiDcwu1wHa78pjroS0,219
-uncountable/integration/queue_runner/worker.py,sha256=EMDtdObowaq0aHndH1bBmlSIOih0jF3Gcn7Yl6NNeOM,4265
+uncountable/integration/queue_runner/worker.py,sha256=WRh_EeGyCPr72v4QL17m6Iu-ipB3T0mhc7nulILf59E,4535
 uncountable/integration/queue_runner/command_server/__init__.py,sha256=gQPVILGpWzCr2i5GJyoqna7AOSFvtn4tav69gB78mTQ,571
 uncountable/integration/queue_runner/command_server/command_client.py,sha256=DJb0TUVFkiiLBEQzHSN94sTRnuEbutNEgdN39XmnOXI,2046
 uncountable/integration/queue_runner/command_server/command_server.py,sha256=yyXryhiEC2eGS0yFElLGsVzSKwOuYvj-zp22jQorkv0,2138
@@ -286,7 +286,7 @@ uncountable/types/api/triggers/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr
 uncountable/types/api/triggers/run_trigger.py,sha256=-oZgPyn43xEKSCs81DVNzwaYMCdRJxbM9GY6fsqKwf4,1090
 uncountable/types/api/uploader/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
 uncountable/types/api/uploader/invoke_uploader.py,sha256=6mwVG136oLp9JcbB2I-kZnrcm3aeZzYZB-SFjEImY2o,1314
-UncountablePythonSDK-0.0.82.dist-info/METADATA,sha256=T6G_cXsagqxX7gJQ9ftG_gjKrByKf9ojWxpxFxHy1ZI,2051
-UncountablePythonSDK-0.0.82.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
-UncountablePythonSDK-0.0.82.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
-UncountablePythonSDK-0.0.82.dist-info/RECORD,,
+UncountablePythonSDK-0.0.83.dist-info/METADATA,sha256=ZyKbekQPHy3xFv8qqoVkAB5XWGyi8pX8XD15NY1GjIc,2051
+UncountablePythonSDK-0.0.83.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+UncountablePythonSDK-0.0.83.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
+UncountablePythonSDK-0.0.83.dist-info/RECORD,,

uncountable/integration/queue_runner/worker.py

@@ -2,6 +2,8 @@ import asyncio
 from concurrent.futures import ProcessPoolExecutor
 from dataclasses import dataclass
 
+from opentelemetry.trace import get_current_span
+
 from uncountable.core.async_batch import AsyncBatchProcessor
 from uncountable.integration.construct_client import construct_uncountable_client
 from uncountable.integration.executors.executors import execute_job
@@ -9,7 +11,7 @@ from uncountable.integration.job import JobArguments
 from uncountable.integration.queue_runner.datastore.interface import Datastore
 from uncountable.integration.queue_runner.types import ListenQueue, ResultQueue
 from uncountable.integration.scan_profiles import load_profiles
-from uncountable.integration.telemetry import JobLogger, get_otel_tracer
+from uncountable.integration.telemetry import JobLogger, Logger, get_otel_tracer
 from uncountable.types import base_t, job_definition_t, queued_job_t
 
 
@@ -28,19 +30,24 @@ class Worker:
         self.datastore = datastore
 
     async def run_worker_loop(self) -> None:
+        logger = Logger(get_current_span())
         while True:
-            queued_job = await self.listen_queue.get()
-            self.datastore.increment_num_attempts(queued_job.queued_job_uuid)
-            loop = asyncio.get_event_loop()
-            result = await loop.run_in_executor(
-                self.process_pool, run_queued_job, queued_job
-            )
-            assert isinstance(result, job_definition_t.JobResult)
-            await self.result_queue.put(
-                queued_job_t.QueuedJobResult(
-                    job_result=result, queued_job_uuid=queued_job.queued_job_uuid
+            try:
+                queued_job = await self.listen_queue.get()
+                self.datastore.increment_num_attempts(queued_job.queued_job_uuid)
+                loop = asyncio.get_event_loop()
+                result = await loop.run_in_executor(
+                    self.process_pool, run_queued_job, queued_job
                 )
-            )
+                assert isinstance(result, job_definition_t.JobResult)
+                await self.result_queue.put(
+                    queued_job_t.QueuedJobResult(
+                        job_result=result, queued_job_uuid=queued_job.queued_job_uuid
+                    )
+                )
+            except BaseException as e:
+                logger.log_exception(e)
+                raise e
 
 
 @dataclass(kw_only=True)