nedo-vision-worker 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

nedo_vision_worker/__init__.py

@@ -6,5 +6,5 @@ A library for running worker agents in the Nedo Vision platform.
 
 from .worker_service import WorkerService
 
-__version__ = "1.0.0"
+__version__ = "1.1.0"
 __all__ = ["WorkerService"] 

nedo_vision_worker/cli.py

@@ -115,7 +115,7 @@ Examples:
     parser.add_argument(
         "--version",
         action="version",
-        version="nedo-vision-worker 1.0.0"
+        version="nedo-vision-worker 1.1.0"
     )
     
     parser.add_argument(

nedo_vision_worker/worker/CoreActionWorker.py

@@ -72,12 +72,23 @@ class CoreActionWorker:
         """Main loop to manage RabbitMQ listener."""
         try:
             while not self.stop_event.is_set():
-                logger.info("📡 [APP] Waiting for core action messages...")
+                logger.info("📡 [APP] Starting core action message listener...")
                 self.listener.start_listening(
                     exchange_name="nedo.worker.core.action", 
                     queue_name=f"nedo.worker.core.{self.worker_id}"
                 )
-                safe_join_thread(self.listener.listener_thread)
+                
+                # Wait for the listener thread to finish (connection lost or stop requested)
+                while not self.stop_event.is_set() and self.listener.listener_thread and self.listener.listener_thread.is_alive():
+                    self.listener.listener_thread.join(timeout=5)  # Check every 5 seconds
+                
+                if not self.stop_event.is_set():
+                    logger.warning("⚠️ [APP] Core action listener disconnected. Attempting to reconnect in 10 seconds...")
+                    self.stop_event.wait(10)  # Wait 10 seconds before reconnecting
+                else:
+                    logger.info("📡 [APP] Core action listener stopped.")
+                    break
+                    
         except Exception as e:
             logger.error("🚨 [APP] Unexpected error in Core Action Worker loop.", exc_info=True)
 

nedo_vision_worker/worker/PipelineActionWorker.py

@@ -73,12 +73,23 @@ class PipelineActionWorker:
         """Main loop to manage RabbitMQ listener."""
         try:
             while not self.stop_event.is_set():
-                logger.info("📡 [APP] Waiting for Pipeline action messages...")
+                logger.info("📡 [APP] Starting pipeline action message listener...")
                 self.listener.start_listening(
                     exchange_name="nedo.worker.pipeline.action", 
                     queue_name=f"nedo.worker.pipeline.{self.worker_id}"
                 )
-                safe_join_thread(self.listener.listener_thread)
+                
+                # Wait for the listener thread to finish (connection lost or stop requested)
+                while not self.stop_event.is_set() and self.listener.listener_thread and self.listener.listener_thread.is_alive():
+                    self.listener.listener_thread.join(timeout=5)  # Check every 5 seconds
+                
+                if not self.stop_event.is_set():
+                    logger.warning("⚠️ [APP] Pipeline action listener disconnected. Attempting to reconnect in 10 seconds...")
+                    self.stop_event.wait(10)  # Wait 10 seconds before reconnecting
+                else:
+                    logger.info("📡 [APP] Pipeline action listener stopped.")
+                    break
+                    
         except Exception as e:
             logger.error("🚨 [APP] Unexpected error in Pipeline Action Worker loop.", exc_info=True)
 

nedo_vision_worker/worker/PipelineImageWorker.py

@@ -78,9 +78,20 @@ class PipelineImageWorker:
         """Main loop to manage RabbitMQ listener."""
         try:
             while not self.stop_event.is_set():
-                logger.info("📡 [APP] Waiting for image request messages...")
+                logger.info("📡 [APP] Starting image request message listener...")
                 self.listener.start_listening(exchange_name="nedo.pipeline.image.request", queue_name=f"nedo.pipeline.request.{self.worker_id}")
-                safe_join_thread(self.listener.listener_thread)
+                
+                # Wait for the listener thread to finish (connection lost or stop requested)
+                while not self.stop_event.is_set() and self.listener.listener_thread and self.listener.listener_thread.is_alive():
+                    self.listener.listener_thread.join(timeout=5)  # Check every 5 seconds
+                
+                if not self.stop_event.is_set():
+                    logger.warning("⚠️ [APP] Image request listener disconnected. Attempting to reconnect in 10 seconds...")
+                    self.stop_event.wait(10)  # Wait 10 seconds before reconnecting
+                else:
+                    logger.info("📡 [APP] Image request listener stopped.")
+                    break
+                    
         except Exception as e:
             logger.error("🚨 [APP] Unexpected error in Pipeline Image Worker loop.", exc_info=True)
 

nedo_vision_worker/worker/RabbitMQListener.py

@@ -91,6 +91,10 @@ class RabbitMQListener:
         if self.listener_thread and self.listener_thread.is_alive():
             logger.warning("⚠️ [APP] RabbitMQ listener is already running.")
             return
+            
+        # Clean up any dead thread reference
+        if self.listener_thread and not self.listener_thread.is_alive():
+            self.listener_thread = None
 
         def run():
             while not self.stop_event.is_set():
@@ -137,6 +141,13 @@ class RabbitMQListener:
             logger.error(f"🚨 [APP] Error during RabbitMQ shutdown: {e}")
 
         self._cleanup_connection()
+        
+        # Clean up the listener thread reference
+        if self.listener_thread:
+            if self.listener_thread.is_alive():
+                self.listener_thread.join(timeout=5)  # Wait up to 5 seconds for thread to finish
+            self.listener_thread = None
+            
         logger.info("🔌 [APP] RabbitMQ listener stopped.")
 
     def _cleanup_connection(self):

nedo_vision_worker/worker/VideoStreamWorker.py

@@ -80,9 +80,20 @@ class VideoStreamWorker:
         """Main loop to manage RabbitMQ listener."""
         try:
             while not self.stop_event.is_set():
-                logger.info("📡 [APP] Waiting for video stream messages...")
+                logger.info("📡 [APP] Starting video stream message listener...")
                 self.listener.start_listening(exchange_name="nedo.worker.stream.preview", queue_name=f"nedo.worker.preview.{self.worker_id}")
-                safe_join_thread(self.listener.listener_thread)
+                
+                # Wait for the listener thread to finish (connection lost or stop requested)
+                while not self.stop_event.is_set() and self.listener.listener_thread and self.listener.listener_thread.is_alive():
+                    self.listener.listener_thread.join(timeout=5)  # Check every 5 seconds
+                
+                if not self.stop_event.is_set():
+                    logger.warning("⚠️ [APP] Video stream listener disconnected. Attempting to reconnect in 10 seconds...")
+                    self.stop_event.wait(10)  # Wait 10 seconds before reconnecting
+                else:
+                    logger.info("📡 [APP] Video stream listener stopped.")
+                    break
+                    
         except Exception as e:
             logger.error("🚨 [APP] Unexpected error in Stream Worker loop.", exc_info=True)
 

{nedo_vision_worker-1.0.0.dist-info → nedo_vision_worker-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker
-Version: 1.0.0
+Version: 1.1.0
 Summary: Nedo Vision Worker Service Library for AI Vision Processing
 Author-email: Willy Achmat Fauzi <willy.achmat@gmail.com>
 Maintainer-email: Willy Achmat Fauzi <willy.achmat@gmail.com>

{nedo_vision_worker-1.0.0.dist-info → nedo_vision_worker-1.1.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
-nedo_vision_worker/__init__.py,sha256=2AhlsGmIh5Kv9ffxD5vwAW0oj6mHQvUKM0pcvPOd8Ow,203
-nedo_vision_worker/cli.py,sha256=LzdNu6Gv9aacxkY8z_hSvtZ5uPPi1S7mNiRwJAQ-zR8,6227
+nedo_vision_worker/__init__.py,sha256=vNz9DUyK7nrpItKgWomp9-IkZXjqbztGNhAa1m97sHQ,203
+nedo_vision_worker/cli.py,sha256=ZzkRXUvewieC81OBT_rFnUYCXkDiBZKoI3yV3jaQsvE,6227
 nedo_vision_worker/doctor.py,sha256=dO3eOOauAFrCwj1UX1A-1jj_zsE2Y0UycGqPXaEOKjg,16152
 nedo_vision_worker/worker_service.py,sha256=XLzaN-n8f4ysPcOmJpO51I9_MWuVTnFMJF08sVHT7IA,9882
 nedo_vision_worker/config/ConfigurationManager.py,sha256=QrQaQ9Cdjpkcr2JE_miyrWJIZmMgZwJYBz-wE45Zzes,8011
@@ -71,22 +71,22 @@ nedo_vision_worker/util/PlatformDetector.py,sha256=-iLPrKs7hp_oltkCI3hESJQkC2uRy
 nedo_vision_worker/util/SystemMonitor.py,sha256=2MWYaEXoL91UANT_-SuDWrFMq1ajPorh8co6Py9PV_c,11300
 nedo_vision_worker/util/VideoProbeUtil.py,sha256=r5P6-ecyrSlzJ7W05LrcfdMHlFKb6vg7asowNh1CJrE,4313
 nedo_vision_worker/util/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
-nedo_vision_worker/worker/CoreActionWorker.py,sha256=AVcHNBcNOye9ASICQIC7XftotKTVbyTybHyJlEMhAZw,4746
+nedo_vision_worker/worker/CoreActionWorker.py,sha256=lb7zPY3yui6I3F4rX4Ii7JwpWZahLEO72rh3iWOgFmg,5441
 nedo_vision_worker/worker/DataSenderWorker.py,sha256=o32MT28EqYwAPmd9NKhX3rfNlDKekNOI2n8mZ6s8CpU,7162
 nedo_vision_worker/worker/DataSyncWorker.py,sha256=2qkwef5cFfAZZa2lB5YZLkFVPselR3zdaaLXIBT7NEQ,6216
 nedo_vision_worker/worker/DatasetFrameSender.py,sha256=1SFYj8LJFNi-anBTapsbq8U_NGMM7mnoMKg9NeFAHys,8087
 nedo_vision_worker/worker/DatasetFrameWorker.py,sha256=Ni5gPeDPk9Rz4_cbg63u7Y6LVw_-Bz24OvfeY-6Yp44,19320
 nedo_vision_worker/worker/PPEDetectionManager.py,sha256=fAolWlrsY5SQAWygvjNBNU56IlC0HLlhPgpz7shL-gk,3588
-nedo_vision_worker/worker/PipelineActionWorker.py,sha256=OkP8pWicD2nezSSHOU4-LcKv8aJRIl-n5a0wNzKcJB4,4948
-nedo_vision_worker/worker/PipelineImageWorker.py,sha256=-d6Gzs9KQJegtvEItlGxoO_mLY2s9bBcYeGiDBz9rug,4916
-nedo_vision_worker/worker/RabbitMQListener.py,sha256=BGWqt_3VSGbYRPP6QW87A46FE0ZG9JJPEXnHl4qyo70,6489
+nedo_vision_worker/worker/PipelineActionWorker.py,sha256=xgvryjKtEsMj4BKqWzDIaK_lFny-DfMCj5Y2DxHnWww,5651
+nedo_vision_worker/worker/PipelineImageWorker.py,sha256=TN3r8WpgXg7K8knWf9y-3g1q1h-O-zDtxsxr0n2XVzI,5615
+nedo_vision_worker/worker/RabbitMQListener.py,sha256=9gR49MDplgpyb-D5HOH0K77-DJQFvhS2E7biL92SjSU,6950
 nedo_vision_worker/worker/RestrictedAreaManager.py,sha256=3yoXgQ459tV1bOa5choEzR9gE6LklrtHR_e0472U3L0,3521
 nedo_vision_worker/worker/SystemUsageManager.py,sha256=StutV4UyLUfduYfK20de4SbPd7wqkR7io0gsOajxWSU,4509
-nedo_vision_worker/worker/VideoStreamWorker.py,sha256=kLm_ke8UtRyzndxCB3MFa-mP4pXQ5d84vBFC2rdj9K0,5817
+nedo_vision_worker/worker/VideoStreamWorker.py,sha256=TxoMYoaU6IcwJDLILC71WUkYahqVAnNX05_5qPbeS_8,6514
 nedo_vision_worker/worker/WorkerManager.py,sha256=7G8ORHpcVAb73UbTUi2WhcQJ-A1XFSq9d_9dLxABHac,6357
 nedo_vision_worker/worker/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
-nedo_vision_worker-1.0.0.dist-info/METADATA,sha256=NBnVN1QJRmTzq0dX2mvuJ3Ww8GT3dEho9G-k_d_clAk,14642
-nedo_vision_worker-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nedo_vision_worker-1.0.0.dist-info/entry_points.txt,sha256=LrglS-8nCi8C_PL_pa6uxdgCe879hBETHDVXAckvs-8,60
-nedo_vision_worker-1.0.0.dist-info/top_level.txt,sha256=vgilhlkyD34YsEKkaBabmhIpcKSvF3XpzD2By68L-XI,19
-nedo_vision_worker-1.0.0.dist-info/RECORD,,
+nedo_vision_worker-1.1.0.dist-info/METADATA,sha256=kSwsqgaekOKQtuFLEk52zpSvSAqXygiwENX0d_X7YSI,14642
+nedo_vision_worker-1.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nedo_vision_worker-1.1.0.dist-info/entry_points.txt,sha256=LrglS-8nCi8C_PL_pa6uxdgCe879hBETHDVXAckvs-8,60
+nedo_vision_worker-1.1.0.dist-info/top_level.txt,sha256=vgilhlkyD34YsEKkaBabmhIpcKSvF3XpzD2By68L-XI,19
+nedo_vision_worker-1.1.0.dist-info/RECORD,,