api.go

// api.go
package main

import (
	"encoding/json"
	"net/http"
	"time"

	"github.com/gorilla/mux"
)

// APIHandler 封装了API的依赖，如工作池
type APIHandler struct {
	Pool *WorkerPool
}

// CreateTaskHandler 处理创建任务的请求
func (h *APIHandler) CreateTaskHandler(w http.ResponseWriter, r *http.Request) {
	sourceID := r.FormValue("source_id")
	url := r.FormValue("url")

	if sourceID == "" || url == "" {
		http.Error(w, "Missing 'source_id' or 'url'", http.StatusBadRequest)
		return
	}

	task := &Task{
		SourceID:       sourceID,
		URL:            url,
		ModelList:      "default_model",
		ReconnectCount: 5,
		OperationType:  1,
		Second:         10,
		IsCascade:      0,
		Status:         "FINISHED", // 初始状态为可运行
		CreatedAt:      time.Now(),
	}

	if err := CreateTask(task); err != nil {
		http.Error(w, "Failed to create task: "+err.Error(), http.StatusConflict)
		return
	}

	w.WriteHeader(http.StatusCreated)
	w.Write([]byte("Task created successfully."))
}

// GetTasksHandler 获取所有任务
func (h *APIHandler) GetTasksHandler(w http.ResponseWriter, r *http.Request) {
	tasks, err := GetAllTasks()
	if err != nil {
		http.Error(w, "Failed to retrieve tasks: "+err.Error(), http.StatusInternalServerError)
		return
	}
	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(tasks)
}

// DeleteTaskHandler 删除任务
func (h *APIHandler) DeleteTaskHandler(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)
	sourceID := vars["source_id"]

	if err := DeleteTask(sourceID); err != nil {
		http.Error(w, "Failed to delete task: "+err.Error(), http.StatusNotFound)
		return
	}
	w.WriteHeader(http.StatusNoContent)
}