feat: trigger billing (#28335)

Signed-off-by: lyzno1 <yuanyouhuilyz@gmail.com>
Co-authored-by: lyzno1 <yuanyouhuilyz@gmail.com>
Co-authored-by: lyzno1 <92089059+lyzno1@users.noreply.github.com>
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>

api/services/trigger/app_trigger_service.py

@@ -0,0 +1,46 @@
+"""
+AppTrigger management service.
+
+Handles AppTrigger model CRUD operations and status management.
+This service centralizes all AppTrigger-related business logic.
+"""
+
+import logging
+
+from sqlalchemy import update
+from sqlalchemy.orm import Session
+
+from extensions.ext_database import db
+from models.enums import AppTriggerStatus
+from models.trigger import AppTrigger
+
+logger = logging.getLogger(__name__)
+
+
+class AppTriggerService:
+    """Service for managing AppTrigger lifecycle and status."""
+
+    @staticmethod
+    def mark_tenant_triggers_rate_limited(tenant_id: str) -> None:
+        """
+        Mark all enabled triggers for a tenant as rate limited due to quota exceeded.
+
+        This method is called when a tenant's quota is exhausted. It updates all
+        enabled triggers to RATE_LIMITED status to prevent further executions until
+        quota is restored.
+
+        Args:
+            tenant_id: Tenant ID whose triggers should be marked as rate limited
+
+        """
+        try:
+            with Session(db.engine) as session:
+                session.execute(
+                    update(AppTrigger)
+                    .where(AppTrigger.tenant_id == tenant_id, AppTrigger.status == AppTriggerStatus.ENABLED)
+                    .values(status=AppTriggerStatus.RATE_LIMITED)
+                )
+                session.commit()
+                logger.info("Marked all enabled triggers as rate limited for tenant %s", tenant_id)
+        except Exception:
+            logger.exception("Failed to mark all enabled triggers as rate limited for tenant %s", tenant_id)

api/services/trigger/webhook_service.py

@@ -18,6 +18,7 @@ from core.file.models import FileTransferMethod
 from core.tools.tool_file_manager import ToolFileManager
 from core.variables.types import SegmentType
 from core.workflow.enums import NodeType
+from enums.quota_type import QuotaType
 from extensions.ext_database import db
 from extensions.ext_redis import redis_client
 from factories import file_factory
@@ -27,6 +28,8 @@ from models.trigger import AppTrigger, WorkflowWebhookTrigger
 from models.workflow import Workflow
 from services.async_workflow_service import AsyncWorkflowService
 from services.end_user_service import EndUserService
+from services.errors.app import QuotaExceededError
+from services.trigger.app_trigger_service import AppTriggerService
 from services.workflow.entities import WebhookTriggerData
 
 logger = logging.getLogger(__name__)
@@ -98,6 +101,12 @@ class WebhookService:
                     raise ValueError(f"App trigger not found for webhook {webhook_id}")
 
                 # Only check enabled status if not in debug mode
+
+                if app_trigger.status == AppTriggerStatus.RATE_LIMITED:
+                    raise ValueError(
+                        f"Webhook trigger is rate limited for webhook {webhook_id}, please upgrade your plan."
+                    )
+
                 if app_trigger.status != AppTriggerStatus.ENABLED:
                     raise ValueError(f"Webhook trigger is disabled for webhook {webhook_id}")
 
@@ -729,6 +738,18 @@ class WebhookService:
                     user_id=None,
                 )
 
+                # consume quota before triggering workflow execution
+                try:
+                    QuotaType.TRIGGER.consume(webhook_trigger.tenant_id)
+                except QuotaExceededError:
+                    AppTriggerService.mark_tenant_triggers_rate_limited(webhook_trigger.tenant_id)
+                    logger.info(
+                        "Tenant %s rate limited, skipping webhook trigger %s",
+                        webhook_trigger.tenant_id,
+                        webhook_trigger.webhook_id,
+                    )
+                    raise
+
                 # Trigger workflow execution asynchronously
                 AsyncWorkflowService.trigger_workflow_async(
                     session,