Initial commit

cleanup_consumers.py

@@ -0,0 +1,39 @@
+
+import asyncio
+import os
+import nats
+
+async def main():
+    # Use environment variables for connection logic to match the worker pod's context
+    nats_url = os.getenv("NATS_URL", "nats://nats-server:4222")
+    nats_user = os.getenv("NATS_USER", "admin")
+    nats_password = os.getenv("NATS_PASSWORD", "package@321#")
+    
+    print(f"Connecting to NATS at {nats_url}...")
+    
+    try:
+        nc = await nats.connect(nats_url, user=nats_user, password=nats_password)
+        js = nc.jetstream()
+        print("Connected!")
+        
+        # List of streams to clean
+        streams = ["CUSTOMERS", "DELIVERIES", "ORDERS", "RIDER", "PRODUCTS"]
+        
+        for stream in streams:
+            print(f"Checking stream: {stream}...")
+            try:
+                consumers = await js.consumers_info(stream)
+                for c in consumers:
+                    # We want to delete the stuck consumers to let them be recreated properly
+                    print(f" - Deleting consumer: {c.name}")
+                    await js.delete_consumer(stream, c.name)
+            except Exception as e:
+                print(f"  Note: Stream {stream} check skipped/failed: {e}")
+
+        await nc.close()
+        
+    except Exception as e:
+        print(f"Fatal connection error: {e}")
+
+if __name__ == "__main__":
+    asyncio.run(main())