add BAM

server/controller/CronController.go

@@ -0,0 +1,217 @@
+package controller
+
+import (
+	"errors"
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"server/logic"
+	"server/model/system"
+	"server/plugin/spider"
+	"strings"
+)
+
+// ------------------------------------------------------ 定时任务管理 ------------------------------------------------------
+
+// FilmCronTaskList 获取所有的定时任务信息
+func FilmCronTaskList(c *gin.Context) {
+	tl := logic.CL.GetFilmCrontab()
+	if len(tl) <= 0 {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  StatusFailed,
+			"message": "暂无任务定时任务信息",
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"status": StatusOk,
+		"data":   tl,
+	})
+}
+
+// GetFilmCronTask 通过Id获取对应的定时任务信息
+func GetFilmCronTask(c *gin.Context) {
+	id := c.DefaultQuery("id", "")
+	if id == "" {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  StatusFailed,
+			"message": "定时任务信息获取失败,任务Id不能为空",
+		})
+		return
+	}
+	task, err := logic.CL.GetFilmCrontabById(id)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  StatusFailed,
+			"message": fmt.Sprint("定时任务信息获取失败", err.Error()),
+		})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"status": StatusOk,
+		"data":   task,
+	})
+}
+
+// FilmCronAdd 添加定时任务
+func FilmCronAdd(c *gin.Context) {
+	var vo = system.FilmCronVo{}
+	// 获取请求参数
+	if err := c.ShouldBindJSON(&vo); err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  StatusFailed,
+			"message": "请求参数异常",
+		})
+		return
+	}
+	// 校验请求参数
+	if err := validTaskAddVo(vo); err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  StatusFailed,
+			"message": err.Error(),
+		})
+		return
+	}
+	// 去除cron表达式左右空格
+	vo.Spec = strings.TrimSpace(vo.Spec)
+	// 执行 定时任务信息保存逻辑
+	if err := logic.CL.AddFilmCrontab(vo); err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  StatusFailed,
+			"message": fmt.Sprint("定时任务添加失败: ", err.Error()),
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"status":  StatusOk,
+		"message": "定时任务添加成功",
+	})
+
+}
+
+// FilmCronUpdate 更新定时任务信息
+func FilmCronUpdate(c *gin.Context) {
+	var t = system.FilmCollectTask{}
+	// 获取请求参数
+	if err := c.ShouldBindJSON(&t); err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  StatusFailed,
+			"message": "请求参数异常",
+		})
+		return
+	}
+	// 校验必要参数
+	if err := validTaskInfo(t); err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  StatusFailed,
+			"message": err.Error(),
+		})
+		return
+	}
+	// 获取未更新的task信息
+	task, err := logic.CL.GetFilmCrontabById(t.Id)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  StatusFailed,
+			"message": fmt.Sprint("更新失败: ", err.Error()),
+		})
+		return
+	}
+	// 将task的可变更属性进行变更
+	task.Ids = t.Ids
+	task.Time = t.Time
+	task.State = t.State
+	task.Remark = t.Remark
+	// 将变更后的task更新到系统中
+	logic.CL.UpdateFilmCron(task)
+	c.JSON(http.StatusOK, gin.H{
+		"status":  StatusOk,
+		"message": fmt.Sprintf("定时任务[%s]更新成功", task.Id),
+	})
+}
+
+// ChangeTaskState 开启 | 关闭Id 对应的定时任务
+func ChangeTaskState(c *gin.Context) {
+	var t = system.FilmCollectTask{}
+	// 获取请求参数
+	if err := c.ShouldBindJSON(&t); err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  StatusFailed,
+			"message": "请求参数异常",
+		})
+		return
+	}
+	// 获取未更新的task信息
+	task, err := logic.CL.GetFilmCrontabById(t.Id)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  StatusFailed,
+			"message": fmt.Sprint("更新失败: ", err.Error()),
+		})
+		return
+	}
+	// 修改task的状态
+	task.State = t.State
+	// 将变更后的task更新到系统中
+	logic.CL.UpdateFilmCron(task)
+	c.JSON(http.StatusOK, gin.H{
+		"status":  StatusOk,
+		"message": fmt.Sprintf("定时任务[%s]更新成功", task.Id),
+	})
+}
+
+// DelFilmCron 删除定时任务
+func DelFilmCron(c *gin.Context) {
+	id := c.DefaultQuery("id", "")
+	if id == "" {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  StatusFailed,
+			"message": "删除失败,任务Id不能为空",
+		})
+		return
+	}
+	// 如果Id不为空则执行删除逻辑
+	if err := logic.CL.DelFilmCrontab(id); err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  StatusFailed,
+			"message": err.Error(),
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"status":  StatusOk,
+		"message": fmt.Sprintf("定时任务[%s]已删除", id),
+	})
+
+}
+
+// -------------------------------------------------- 参数校验 --------------------------------------------------
+
+// 定时任务必要属性校验
+func validTaskInfo(t system.FilmCollectTask) error {
+	if len(t.Id) <= 0 {
+		return errors.New("参数校验失败, 任务Id信息不能为空")
+	}
+	if t.Time == 0 {
+		return errors.New("参数校验失败, 采集时长不能为零值")
+	}
+	return nil
+}
+
+// 任务添加参数校验
+func validTaskAddVo(vo system.FilmCronVo) error {
+	if vo.Model != 0 && vo.Model != 1 {
+		return errors.New("参数校验失败, 未定义的任务类型")
+	}
+	if vo.Time == 0 {
+		return errors.New("参数校验失败, 采集时长不能为零值")
+	}
+	if err := spider.ValidSpec(vo.Spec); err != nil {
+		return errors.New(fmt.Sprint("参数校验失败 cron表达式校验失败: ", err.Error()))
+	}
+	if vo.Model == 1 && (vo.Ids == nil || len(vo.Ids) <= 0) {
+		return errors.New("参数校验失败, 自定义更新未绑定任何资源站点")
+	}
+	return nil
+}