Initial public release

backend/app/services/aoi_operation_worker.py

@@ -0,0 +1,38 @@
+from __future__ import annotations
+
+import asyncio
+import logging
+
+from app.db.session import SessionLocal
+from app.models import AoiOperation
+from app.services.aoi_operation_executor import AoiOperationExecutor
+
+
+logger = logging.getLogger("geointel.aoi_worker")
+
+
+class AoiOperationWorker:
+    @staticmethod
+    def run_once() -> int:
+        db = SessionLocal()
+        try:
+            rows = db.query(AoiOperation).filter(AoiOperation.status.in_(("queued", "running"))).order_by(AoiOperation.created_at).limit(10).all()
+            for operation in rows:
+                try:
+                    AoiOperationExecutor.execute_next(db, operation.project_id, operation.id)
+                except Exception:
+                    db.rollback()
+                    logger.exception("AOI partition execution failed operation_id=%s", operation.id)
+            return len(rows)
+        finally:
+            db.close()
+
+    @staticmethod
+    async def run(stop_event: asyncio.Event, poll_seconds: float) -> None:
+        while not stop_event.is_set():
+            processed = await asyncio.to_thread(AoiOperationWorker.run_once)
+            if processed == 0:
+                try:
+                    await asyncio.wait_for(stop_event.wait(), timeout=poll_seconds)
+                except TimeoutError:
+                    pass