mirror of
https://github.com/oarkflow/mq.git
synced 2025-09-29 13:22:10 +08:00
update
This commit is contained in:
sujit
436
workflow/api.go
Normal file
436
workflow/api.go
Normal file
@@ -0,0 +1,436 @@
|
||||
package workflow
|
||||
|
||||
import (
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/gofiber/fiber/v2"
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
// WorkflowAPI provides HTTP handlers for workflow management
|
||||
type WorkflowAPI struct {
|
||||
engine *WorkflowEngine
|
||||
}
|
||||
|
||||
// NewWorkflowAPI creates a new workflow API handler
|
||||
func NewWorkflowAPI(engine *WorkflowEngine) *WorkflowAPI {
|
||||
return &WorkflowAPI{
|
||||
engine: engine,
|
||||
}
|
||||
}
|
||||
|
||||
// RegisterRoutes registers all workflow routes with Fiber app
|
||||
func (api *WorkflowAPI) RegisterRoutes(app *fiber.App) {
|
||||
v1 := app.Group("/api/v1/workflows")
|
||||
|
||||
// Workflow definition routes
|
||||
v1.Post("/", api.CreateWorkflow)
|
||||
v1.Get("/", api.ListWorkflows)
|
||||
v1.Get("/:id", api.GetWorkflow)
|
||||
v1.Put("/:id", api.UpdateWorkflow)
|
||||
v1.Delete("/:id", api.DeleteWorkflow)
|
||||
v1.Get("/:id/versions", api.GetWorkflowVersions)
|
||||
|
||||
// Execution routes
|
||||
v1.Post("/:id/execute", api.ExecuteWorkflow)
|
||||
v1.Get("/:id/executions", api.ListWorkflowExecutions)
|
||||
v1.Get("/executions", api.ListAllExecutions)
|
||||
v1.Get("/executions/:executionId", api.GetExecution)
|
||||
v1.Post("/executions/:executionId/cancel", api.CancelExecution)
|
||||
v1.Post("/executions/:executionId/suspend", api.SuspendExecution)
|
||||
v1.Post("/executions/:executionId/resume", api.ResumeExecution)
|
||||
|
||||
// Management routes
|
||||
v1.Get("/health", api.HealthCheck)
|
||||
v1.Get("/metrics", api.GetMetrics)
|
||||
}
|
||||
|
||||
// CreateWorkflow creates a new workflow definition
|
||||
func (api *WorkflowAPI) CreateWorkflow(c *fiber.Ctx) error {
|
||||
var definition WorkflowDefinition
|
||||
if err := c.BodyParser(&definition); err != nil {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": "Invalid request body",
|
||||
})
|
||||
}
|
||||
|
||||
// Set ID if not provided
|
||||
if definition.ID == "" {
|
||||
definition.ID = uuid.New().String()
|
||||
}
|
||||
|
||||
// Set version if not provided
|
||||
if definition.Version == "" {
|
||||
definition.Version = "1.0.0"
|
||||
}
|
||||
|
||||
if err := api.engine.RegisterWorkflow(c.Context(), &definition); err != nil {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.Status(fiber.StatusCreated).JSON(definition)
|
||||
}
|
||||
|
||||
// ListWorkflows lists workflow definitions with filtering
|
||||
func (api *WorkflowAPI) ListWorkflows(c *fiber.Ctx) error {
|
||||
filter := &WorkflowFilter{
|
||||
Limit: 10,
|
||||
Offset: 0,
|
||||
}
|
||||
|
||||
// Parse query parameters
|
||||
if limit := c.Query("limit"); limit != "" {
|
||||
if l, err := strconv.Atoi(limit); err == nil {
|
||||
filter.Limit = l
|
||||
}
|
||||
}
|
||||
|
||||
if offset := c.Query("offset"); offset != "" {
|
||||
if o, err := strconv.Atoi(offset); err == nil {
|
||||
filter.Offset = o
|
||||
}
|
||||
}
|
||||
|
||||
if status := c.Query("status"); status != "" {
|
||||
filter.Status = []WorkflowStatus{WorkflowStatus(status)}
|
||||
}
|
||||
|
||||
if category := c.Query("category"); category != "" {
|
||||
filter.Category = []string{category}
|
||||
}
|
||||
|
||||
if owner := c.Query("owner"); owner != "" {
|
||||
filter.Owner = []string{owner}
|
||||
}
|
||||
|
||||
if search := c.Query("search"); search != "" {
|
||||
filter.Search = search
|
||||
}
|
||||
|
||||
if sortBy := c.Query("sort_by"); sortBy != "" {
|
||||
filter.SortBy = sortBy
|
||||
}
|
||||
|
||||
if sortOrder := c.Query("sort_order"); sortOrder != "" {
|
||||
filter.SortOrder = sortOrder
|
||||
}
|
||||
|
||||
workflows, err := api.engine.ListWorkflows(c.Context(), filter)
|
||||
if err != nil {
|
||||
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.JSON(fiber.Map{
|
||||
"workflows": workflows,
|
||||
"total": len(workflows),
|
||||
"limit": filter.Limit,
|
||||
"offset": filter.Offset,
|
||||
})
|
||||
}
|
||||
|
||||
// GetWorkflow retrieves a specific workflow definition
|
||||
func (api *WorkflowAPI) GetWorkflow(c *fiber.Ctx) error {
|
||||
id := c.Params("id")
|
||||
version := c.Query("version")
|
||||
|
||||
workflow, err := api.engine.GetWorkflow(c.Context(), id, version)
|
||||
if err != nil {
|
||||
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.JSON(workflow)
|
||||
}
|
||||
|
||||
// UpdateWorkflow updates an existing workflow definition
|
||||
func (api *WorkflowAPI) UpdateWorkflow(c *fiber.Ctx) error {
|
||||
id := c.Params("id")
|
||||
|
||||
var definition WorkflowDefinition
|
||||
if err := c.BodyParser(&definition); err != nil {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": "Invalid request body",
|
||||
})
|
||||
}
|
||||
|
||||
// Ensure ID matches
|
||||
definition.ID = id
|
||||
|
||||
if err := api.engine.RegisterWorkflow(c.Context(), &definition); err != nil {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.JSON(definition)
|
||||
}
|
||||
|
||||
// DeleteWorkflow removes a workflow definition
|
||||
func (api *WorkflowAPI) DeleteWorkflow(c *fiber.Ctx) error {
|
||||
id := c.Params("id")
|
||||
|
||||
if err := api.engine.DeleteWorkflow(c.Context(), id); err != nil {
|
||||
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.Status(fiber.StatusNoContent).Send(nil)
|
||||
}
|
||||
|
||||
// GetWorkflowVersions retrieves all versions of a workflow
|
||||
func (api *WorkflowAPI) GetWorkflowVersions(c *fiber.Ctx) error {
|
||||
id := c.Params("id")
|
||||
|
||||
versions, err := api.engine.registry.GetVersions(c.Context(), id)
|
||||
if err != nil {
|
||||
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.JSON(fiber.Map{
|
||||
"workflow_id": id,
|
||||
"versions": versions,
|
||||
})
|
||||
}
|
||||
|
||||
// ExecuteWorkflow starts workflow execution
|
||||
func (api *WorkflowAPI) ExecuteWorkflow(c *fiber.Ctx) error {
|
||||
id := c.Params("id")
|
||||
|
||||
var request struct {
|
||||
Input map[string]interface{} `json:"input"`
|
||||
Priority Priority `json:"priority"`
|
||||
Owner string `json:"owner"`
|
||||
TriggeredBy string `json:"triggered_by"`
|
||||
ParentExecution string `json:"parent_execution"`
|
||||
Delay int `json:"delay"` // seconds
|
||||
}
|
||||
|
||||
if err := c.BodyParser(&request); err != nil {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": "Invalid request body",
|
||||
})
|
||||
}
|
||||
|
||||
options := &ExecutionOptions{
|
||||
Priority: request.Priority,
|
||||
Owner: request.Owner,
|
||||
TriggeredBy: request.TriggeredBy,
|
||||
ParentExecution: request.ParentExecution,
|
||||
Delay: time.Duration(request.Delay) * time.Second,
|
||||
}
|
||||
|
||||
execution, err := api.engine.ExecuteWorkflow(c.Context(), id, request.Input, options)
|
||||
if err != nil {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.Status(fiber.StatusCreated).JSON(execution)
|
||||
}
|
||||
|
||||
// ListWorkflowExecutions lists executions for a specific workflow
|
||||
func (api *WorkflowAPI) ListWorkflowExecutions(c *fiber.Ctx) error {
|
||||
workflowID := c.Params("id")
|
||||
|
||||
filter := &ExecutionFilter{
|
||||
WorkflowID: []string{workflowID},
|
||||
Limit: 10,
|
||||
Offset: 0,
|
||||
}
|
||||
|
||||
// Parse query parameters
|
||||
if limit := c.Query("limit"); limit != "" {
|
||||
if l, err := strconv.Atoi(limit); err == nil {
|
||||
filter.Limit = l
|
||||
}
|
||||
}
|
||||
|
||||
if offset := c.Query("offset"); offset != "" {
|
||||
if o, err := strconv.Atoi(offset); err == nil {
|
||||
filter.Offset = o
|
||||
}
|
||||
}
|
||||
|
||||
if status := c.Query("status"); status != "" {
|
||||
filter.Status = []ExecutionStatus{ExecutionStatus(status)}
|
||||
}
|
||||
|
||||
executions, err := api.engine.ListExecutions(c.Context(), filter)
|
||||
if err != nil {
|
||||
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.JSON(fiber.Map{
|
||||
"executions": executions,
|
||||
"total": len(executions),
|
||||
"limit": filter.Limit,
|
||||
"offset": filter.Offset,
|
||||
})
|
||||
}
|
||||
|
||||
// ListAllExecutions lists all executions with filtering
|
||||
func (api *WorkflowAPI) ListAllExecutions(c *fiber.Ctx) error {
|
||||
filter := &ExecutionFilter{
|
||||
Limit: 10,
|
||||
Offset: 0,
|
||||
}
|
||||
|
||||
// Parse query parameters
|
||||
if limit := c.Query("limit"); limit != "" {
|
||||
if l, err := strconv.Atoi(limit); err == nil {
|
||||
filter.Limit = l
|
||||
}
|
||||
}
|
||||
|
||||
if offset := c.Query("offset"); offset != "" {
|
||||
if o, err := strconv.Atoi(offset); err == nil {
|
||||
filter.Offset = o
|
||||
}
|
||||
}
|
||||
|
||||
if status := c.Query("status"); status != "" {
|
||||
filter.Status = []ExecutionStatus{ExecutionStatus(status)}
|
||||
}
|
||||
|
||||
if owner := c.Query("owner"); owner != "" {
|
||||
filter.Owner = []string{owner}
|
||||
}
|
||||
|
||||
if priority := c.Query("priority"); priority != "" {
|
||||
filter.Priority = []Priority{Priority(priority)}
|
||||
}
|
||||
|
||||
executions, err := api.engine.ListExecutions(c.Context(), filter)
|
||||
if err != nil {
|
||||
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.JSON(fiber.Map{
|
||||
"executions": executions,
|
||||
"total": len(executions),
|
||||
"limit": filter.Limit,
|
||||
"offset": filter.Offset,
|
||||
})
|
||||
}
|
||||
|
||||
// GetExecution retrieves a specific execution
|
||||
func (api *WorkflowAPI) GetExecution(c *fiber.Ctx) error {
|
||||
executionID := c.Params("executionId")
|
||||
|
||||
execution, err := api.engine.GetExecution(c.Context(), executionID)
|
||||
if err != nil {
|
||||
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.JSON(execution)
|
||||
}
|
||||
|
||||
// CancelExecution cancels a running execution
|
||||
func (api *WorkflowAPI) CancelExecution(c *fiber.Ctx) error {
|
||||
executionID := c.Params("executionId")
|
||||
|
||||
if err := api.engine.CancelExecution(c.Context(), executionID); err != nil {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.Status(fiber.StatusOK).JSON(fiber.Map{
|
||||
"message": "Execution cancelled",
|
||||
})
|
||||
}
|
||||
|
||||
// SuspendExecution suspends a running execution
|
||||
func (api *WorkflowAPI) SuspendExecution(c *fiber.Ctx) error {
|
||||
executionID := c.Params("executionId")
|
||||
|
||||
if err := api.engine.SuspendExecution(c.Context(), executionID); err != nil {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.Status(fiber.StatusOK).JSON(fiber.Map{
|
||||
"message": "Execution suspended",
|
||||
})
|
||||
}
|
||||
|
||||
// ResumeExecution resumes a suspended execution
|
||||
func (api *WorkflowAPI) ResumeExecution(c *fiber.Ctx) error {
|
||||
executionID := c.Params("executionId")
|
||||
|
||||
if err := api.engine.ResumeExecution(c.Context(), executionID); err != nil {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.Status(fiber.StatusOK).JSON(fiber.Map{
|
||||
"message": "Execution resumed",
|
||||
})
|
||||
}
|
||||
|
||||
// HealthCheck returns the health status of the workflow engine
|
||||
func (api *WorkflowAPI) HealthCheck(c *fiber.Ctx) error {
|
||||
return c.JSON(fiber.Map{
|
||||
"status": "healthy",
|
||||
"timestamp": time.Now(),
|
||||
"version": "1.0.0",
|
||||
})
|
||||
}
|
||||
|
||||
// GetMetrics returns workflow engine metrics
|
||||
func (api *WorkflowAPI) GetMetrics(c *fiber.Ctx) error {
|
||||
// In a real implementation, collect actual metrics
|
||||
metrics := map[string]interface{}{
|
||||
"total_workflows": 0,
|
||||
"total_executions": 0,
|
||||
"running_executions": 0,
|
||||
"completed_executions": 0,
|
||||
"failed_executions": 0,
|
||||
"average_execution_time": "0s",
|
||||
"uptime": "0s",
|
||||
"memory_usage": "0MB",
|
||||
"cpu_usage": "0%",
|
||||
}
|
||||
|
||||
return c.JSON(metrics)
|
||||
}
|
||||
|
||||
// Error handling middleware
|
||||
func ErrorHandler(c *fiber.Ctx, err error) error {
|
||||
code := fiber.StatusInternalServerError
|
||||
|
||||
if e, ok := err.(*fiber.Error); ok {
|
||||
code = e.Code
|
||||
}
|
||||
|
||||
return c.Status(code).JSON(fiber.Map{
|
||||
"error": true,
|
||||
"message": err.Error(),
|
||||
"timestamp": time.Now(),
|
||||
})
|
||||
}
|
||||
|
||||
// CORS middleware configuration
|
||||
func CORSConfig() fiber.Config {
|
||||
return fiber.Config{
|
||||
ErrorHandler: ErrorHandler,
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user