134 lines
2.8 KiB
Go
134 lines
2.8 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"oci-portal/internal/service"
|
|
)
|
|
|
|
// taskHandler 处理后台任务（定时测活、抢机）相关请求。
|
|
type taskHandler struct {
|
|
svc *service.TaskService
|
|
}
|
|
|
|
type createTaskRequest struct {
|
|
Name string `json:"name" binding:"required"`
|
|
Type string `json:"type" binding:"required"`
|
|
CronExpr string `json:"cronExpr" binding:"required"`
|
|
Payload json.RawMessage `json:"payload"`
|
|
}
|
|
|
|
type updateTaskRequest struct {
|
|
Name *string `json:"name"`
|
|
CronExpr *string `json:"cronExpr"`
|
|
Payload json.RawMessage `json:"payload"`
|
|
Status *string `json:"status"`
|
|
}
|
|
|
|
func (h *taskHandler) create(c *gin.Context) {
|
|
var req createTaskRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
task, err := h.svc.CreateTask(c.Request.Context(), service.CreateTaskInput{
|
|
Name: req.Name,
|
|
Type: req.Type,
|
|
CronExpr: req.CronExpr,
|
|
Payload: req.Payload,
|
|
})
|
|
if err != nil {
|
|
respondError(c, err)
|
|
return
|
|
}
|
|
c.JSON(http.StatusCreated, task)
|
|
}
|
|
|
|
func (h *taskHandler) list(c *gin.Context) {
|
|
tasks, err := h.svc.ListTasks(c.Request.Context())
|
|
if err != nil {
|
|
respondError(c, err)
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, tasks)
|
|
}
|
|
|
|
func (h *taskHandler) get(c *gin.Context) {
|
|
id, ok := pathID(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
task, err := h.svc.GetTask(c.Request.Context(), id)
|
|
if err != nil {
|
|
respondError(c, err)
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, task)
|
|
}
|
|
|
|
func (h *taskHandler) update(c *gin.Context) {
|
|
id, ok := pathID(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
var req updateTaskRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
task, err := h.svc.UpdateTask(c.Request.Context(), id, service.UpdateTaskInput{
|
|
Name: req.Name,
|
|
CronExpr: req.CronExpr,
|
|
Payload: req.Payload,
|
|
Status: req.Status,
|
|
})
|
|
if err != nil {
|
|
respondError(c, err)
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, task)
|
|
}
|
|
|
|
func (h *taskHandler) remove(c *gin.Context) {
|
|
id, ok := pathID(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
if err := h.svc.DeleteTask(c.Request.Context(), id); err != nil {
|
|
respondError(c, err)
|
|
return
|
|
}
|
|
c.Status(http.StatusNoContent)
|
|
}
|
|
|
|
func (h *taskHandler) logs(c *gin.Context) {
|
|
id, ok := pathID(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
limit, _ := strconv.Atoi(c.DefaultQuery("limit", "50"))
|
|
logs, err := h.svc.TaskLogs(c.Request.Context(), id, limit)
|
|
if err != nil {
|
|
respondError(c, err)
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, logs)
|
|
}
|
|
|
|
func (h *taskHandler) run(c *gin.Context) {
|
|
id, ok := pathID(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
entry, err := h.svc.RunTaskNow(c.Request.Context(), id)
|
|
if err != nil {
|
|
respondError(c, err)
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, entry)
|
|
}
|