提交

modules/task/controller/admin/admin.go

@@ -0,0 +1 @@
+package admin

modules/task/controller/admin/task_info.go

@@ -0,0 +1,94 @@
+package admin
+
+import (
+	"github.com/cool-team-official/cool-admin-go/cool"
+	"github.com/cool-team-official/cool-admin-go/modules/task/service"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/net/ghttp"
+	"github.com/gogf/gf/v2/util/gconv"
+)
+
+type TaskInfoController struct {
+	*cool.Controller
+}
+
+func init() {
+	var task_info_controller = &TaskInfoController{
+		&cool.Controller{
+			Prefix:  "/admin/task/info",
+			Api:     []string{"Add", "Delete", "Update", "Info", "List", "Page", "Start", "Stop"},
+			Service: service.NewTaskInfoService(),
+		},
+	}
+	// 注册路由
+	cool.RegisterController(task_info_controller)
+}
+
+// TaskInfoStopReq 请求参数
+type TaskInfoStopReq struct {
+	g.Meta `path:"/stop" method:"GET"`
+	ID     int64 `json:"id" v:"required#请输入id"`
+}
+
+// Stop 停止任务
+func (c *TaskInfoController) Stop(ctx g.Ctx, req *TaskInfoStopReq) (res *cool.BaseRes, err error) {
+
+	err = cool.ClusterRunFunc(ctx, "TaskStopFunc("+gconv.String(req.ID)+")")
+	if err != nil {
+		return cool.Fail(err.Error()), err
+	}
+	res = cool.Ok("停止成功")
+	return
+}
+
+// TaskInfoStartReq 请求参数
+type TaskInfoStartReq struct {
+	g.Meta `path:"/start" method:"GET"`
+	ID     int64 `json:"id" v:"required#请输入id"`
+}
+
+// Start 启动任务
+func (c *TaskInfoController) Start(ctx g.Ctx, req *TaskInfoStartReq) (res *cool.BaseRes, err error) {
+
+	err = cool.ClusterRunFunc(ctx, "TaskStartFunc("+gconv.String(req.ID)+")")
+	if err != nil {
+		return cool.Fail(err.Error()), err
+	}
+	res = cool.Ok("启动成功")
+	return
+}
+
+// TaskInfoOnceReq 请求参数
+type TaskInfoOnceReq struct {
+	g.Meta `path:"/once" method:"POST"`
+	ID     int64 `json:"id" v:"required#请输入id"`
+}
+
+// Once 执行一次
+func (c *TaskInfoController) Once(ctx g.Ctx, req *TaskInfoOnceReq) (res *cool.BaseRes, err error) {
+	err = c.Service.(*service.TaskInfoService).Once(ctx, req.ID)
+	if err != nil {
+		return cool.Fail(err.Error()), err
+	}
+	res = cool.Ok("执行成功")
+	return
+}
+
+// TaskInfoLogReq 请求参数
+type TaskInfoLogReq struct {
+	g.Meta `path:"/log" method:"GET"`
+	ID     int64 `json:"id"`
+	Status int   `json:"status"`
+}
+
+// Log 任务日志
+func (c *TaskInfoController) Log(ctx g.Ctx, req *TaskInfoLogReq) (res *cool.BaseRes, err error) {
+	r := ghttp.RequestFromCtx(ctx)
+	param := r.GetQueryMapStrStr()
+	data, err := c.Service.(*service.TaskInfoService).Log(ctx, param)
+	if err != nil {
+		return cool.Fail(err.Error()), err
+	}
+	res = cool.Ok(data)
+	return
+}

modules/task/controller/app/app.go

@@ -0,0 +1 @@
+package app

modules/task/controller/controller.go

@@ -0,0 +1,7 @@
+package controller
+
+import (
+	_ "github.com/cool-team-official/cool-admin-go/modules/task/controller/admin"
+	_ "github.com/cool-team-official/cool-admin-go/modules/task/controller/app"
+	_ "github.com/cool-team-official/cool-admin-go/modules/task/service"
+)