Add Dispatcher Version 1.0

dispatcher.py

@@ -0,0 +1,152 @@
+# =====================================================
+# Apckeyl Framework
+# Version 1.0
+# dispatcher.py
+# =====================================================
+
+"""
+Apckeyl Dispatcher.
+
+Version 1.0
+
+Dispatcher отвечает за передачу задачи
+из Control Plane в Compute Plane.
+
+На этом этапе Dispatcher НЕ выполняет
+реальный HTTP/API-запрос.
+
+Он только:
+
+- принимает задачу;
+- проверяет выбранный Compute Module;
+- формирует dispatch request;
+- переводит задачу в статус dispatched.
+
+Реальное подключение внешнего Space
+будет добавлено через Compute Adapter.
+"""
+
+from task_manager import TaskManager
+
+
+# =====================================================
+# Dispatcher
+# =====================================================
+
+class Dispatcher:
+
+    def __init__(
+        self,
+        task_manager=None,
+    ):
+
+        if task_manager is None:
+
+            task_manager = TaskManager()
+
+        self.task_manager = task_manager
+
+
+    # =================================================
+    # Prepare Dispatch
+    # =================================================
+
+    def prepare_dispatch(
+        self,
+        task_id,
+    ):
+
+        task = self.task_manager.get_task(
+            task_id
+        )
+
+        if task is None:
+
+            raise KeyError(
+                f"Unknown task: {task_id}"
+            )
+
+
+        # -------------------------------------------------
+        # Task must be created or queued
+        # -------------------------------------------------
+
+        allowed_statuses = {
+
+            "created",
+
+            "queued",
+        }
+
+        if task["status"] not in allowed_statuses:
+
+            raise RuntimeError(
+
+                f"Task {task_id} cannot be "
+                f"dispatched from status "
+                f"'{task['status']}'"
+            )
+
+
+        # -------------------------------------------------
+        # Build dispatch request
+        # -------------------------------------------------
+
+        dispatch_request = {
+
+            "task_id": task[
+                "task_id"
+            ],
+
+            "task_type": task[
+                "task_type"
+            ],
+
+            "module_id": task[
+                "module_id"
+            ],
+
+            "module_name": task[
+                "module_name"
+            ],
+
+            "payload": task[
+                "payload"
+            ],
+        }
+
+
+        # -------------------------------------------------
+        # Update task status
+        # -------------------------------------------------
+
+        self.task_manager.update_status(
+
+            task_id,
+
+            "dispatched",
+        )
+
+
+        return dispatch_request
+
+
+    # =================================================
+    # Get Task
+    # =================================================
+
+    def get_task(
+        self,
+        task_id,
+    ):
+
+        return self.task_manager.get_task(
+            task_id
+        )
+
+
+# =====================================================
+# Default Dispatcher
+# =====================================================
+
+dispatcher = Dispatcher()