dify

dify/api/tasks/workflow_cfs_scheduler/cfs_scheduler.py

@@ -0,0 +1,32 @@
+from services.workflow.entities import WorkflowScheduleCFSPlanEntity
+from services.workflow.scheduler import CFSPlanScheduler, SchedulerCommand
+from tasks.workflow_cfs_scheduler.entities import AsyncWorkflowQueue
+
+
+class AsyncWorkflowCFSPlanEntity(WorkflowScheduleCFSPlanEntity):
+    """
+    Trigger workflow CFS plan entity.
+    """
+
+    queue: AsyncWorkflowQueue
+
+
+class AsyncWorkflowCFSPlanScheduler(CFSPlanScheduler):
+    """
+    Trigger workflow CFS plan scheduler.
+    """
+
+    plan: AsyncWorkflowCFSPlanEntity
+
+    def can_schedule(self) -> SchedulerCommand:
+        """
+        Check if the workflow can be scheduled.
+        """
+        if self.plan.queue in [AsyncWorkflowQueue.PROFESSIONAL_QUEUE, AsyncWorkflowQueue.TEAM_QUEUE]:
+            """
+            permitted all paid users to schedule the workflow any time
+            """
+            return SchedulerCommand.NONE
+
+        # FIXME: avoid the sandbox user's workflow at a running state for ever
+        return SchedulerCommand.RESOURCE_LIMIT_REACHED

dify/api/tasks/workflow_cfs_scheduler/entities.py

@@ -0,0 +1,25 @@
+from enum import StrEnum
+
+from configs import dify_config
+from services.workflow.entities import WorkflowScheduleCFSPlanEntity
+
+# Determine queue names based on edition
+if dify_config.EDITION == "CLOUD":
+    # Cloud edition: separate queues for different tiers
+    _professional_queue = "workflow_professional"
+    _team_queue = "workflow_team"
+    _sandbox_queue = "workflow_sandbox"
+    AsyncWorkflowSystemStrategy = WorkflowScheduleCFSPlanEntity.Strategy.TimeSlice
+else:
+    # Community edition: single workflow queue (not dataset)
+    _professional_queue = "workflow"
+    _team_queue = "workflow"
+    _sandbox_queue = "workflow"
+    AsyncWorkflowSystemStrategy = WorkflowScheduleCFSPlanEntity.Strategy.Nop
+
+
+class AsyncWorkflowQueue(StrEnum):
+    # Define constants
+    PROFESSIONAL_QUEUE = _professional_queue
+    TEAM_QUEUE = _team_queue
+    SANDBOX_QUEUE = _sandbox_queue