81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
|
package api
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"devops/agent/tasks"
|
|||
|
|
"devops/server/apps/task"
|
|||
|
|
|
|||
|
|
restfulspec "github.com/emicklei/go-restful-openapi/v2"
|
|||
|
|
"github.com/emicklei/go-restful/v3"
|
|||
|
|
"github.com/google/uuid"
|
|||
|
|
"github.com/infraboard/mcube/v2/exception"
|
|||
|
|
"github.com/infraboard/mcube/v2/http/restful/response"
|
|||
|
|
"github.com/infraboard/mcube/v2/ioc"
|
|||
|
|
"github.com/infraboard/mcube/v2/ioc/config/gorestful"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func init() {
|
|||
|
|
ioc.Api().Registry(&TaskApiHandler{})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 一个API, 提供给外部调用的接口，来触发一个Task的执行
|
|||
|
|
// 参数 TaskSpec, 返回 Task
|
|||
|
|
type TaskApiHandler struct {
|
|||
|
|
ioc.ObjectImpl
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (h *TaskApiHandler) Name() string {
|
|||
|
|
return "tasks"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (i *TaskApiHandler) Version() string {
|
|||
|
|
return "v1"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (h *TaskApiHandler) Init() error {
|
|||
|
|
ws := gorestful.ObjectRouter(h)
|
|||
|
|
|
|||
|
|
// restfulspec "github.com/emicklei/go-restful-openapi/v2"
|
|||
|
|
// 这个库是 go-restful 的一个扩展库，用于生成 OpenAPI 规范的文档
|
|||
|
|
tags := []string{"Task管理"}
|
|||
|
|
ws.Route(ws.POST("").To(h.RunTask).
|
|||
|
|
Doc("运行任务").
|
|||
|
|
Metadata(restfulspec.KeyOpenAPITags, tags).
|
|||
|
|
Reads(task.TaskSpec{}).
|
|||
|
|
Writes(task.Task{}).
|
|||
|
|
Returns(200, "OK", task.Task{}))
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RunTask 处理运行任务的API请求
|
|||
|
|
// Body 参数是 TaskSpec, 返回 Task
|
|||
|
|
func (h *TaskApiHandler) RunTask(r *restful.Request, w *restful.Response) {
|
|||
|
|
// 解析请求参数，构造 TaskSpec
|
|||
|
|
req := &task.TaskSpec{}
|
|||
|
|
if err := r.ReadEntity(req); err != nil {
|
|||
|
|
// "github.com/infraboard/mcube/v2/http/restful/response"
|
|||
|
|
// Failed 封装的 GoRestful框架的 异常返回
|
|||
|
|
response.Failed(w, err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 补充TaskId
|
|||
|
|
if req.Id == "" {
|
|||
|
|
req.Id = uuid.NewString()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 找到对应的 TaskRunner, 这里我们通过 TaskSpec 中的 Name 字段来找到对应的 TaskRunner
|
|||
|
|
taskDebugRunner := tasks.GetTaskRunner(req.Name)
|
|||
|
|
if taskDebugRunner == nil {
|
|||
|
|
response.Failed(w, exception.NewInternalServerError("%s not found", req.Name))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 处理请求
|
|||
|
|
taskResp, err := taskDebugRunner.Run(r.Request.Context(), req)
|
|||
|
|
if err != nil {
|
|||
|
|
response.Failed(w, err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
response.Success(w, taskResp)
|
|||
|
|
}
|