GoFilm/server/controller/CronController.go

package controller

import (
	"errors"
	"fmt"
	"github.com/gin-gonic/gin"
	"server/logic"
	"server/model/system"
	"server/plugin/spider"
	"strings"
)

// ------------------------------------------------------ 定时任务管理 ------------------------------------------------------

// FilmCronTaskList 获取所有的定时任务信息
func FilmCronTaskList(c *gin.Context) {
	tl := logic.CL.GetFilmCrontab()
	if len(tl) <= 0 {
		system.Failed("暂无任务定时任务信息", c)
		return
	}
	system.Success(tl, "定时任务列表获取成功!!!", c)
}

// GetFilmCronTask 通过Id获取对应的定时任务信息
func GetFilmCronTask(c *gin.Context) {
	id := c.DefaultQuery("id", "")
	if id == "" {
		system.Failed("定时任务信息获取失败,任务Id不能为空", c)
		return
	}
	task, err := logic.CL.GetFilmCrontabById(id)
	if err != nil {
		system.Failed(fmt.Sprint("定时任务信息获取失败", err.Error()), c)
		return
	}
	system.Success(task, "定时任务详情获取成功!!!", c)
}

// FilmCronAdd 添加定时任务
func FilmCronAdd(c *gin.Context) {
	var vo = system.FilmCronVo{}
	// 获取请求参数
	if err := c.ShouldBindJSON(&vo); err != nil {
		system.Failed("请求参数异常!!!", c)
		return
	}
	// 校验请求参数
	if err := validTaskAddVo(vo); err != nil {
		system.Failed(err.Error(), c)
		return
	}
	// 去除cron表达式左右空格
	vo.Spec = strings.TrimSpace(vo.Spec)
	// 执行 定时任务信息保存逻辑
	if err := logic.CL.AddFilmCrontab(vo); err != nil {
		system.Failed(fmt.Sprint("定时任务添加失败: ", err.Error()), c)
		return
	}
	system.SuccessOnlyMsg("定时任务添加成功", c)
}

// FilmCronUpdate 更新定时任务信息
func FilmCronUpdate(c *gin.Context) {
	var t = system.FilmCollectTask{}
	// 获取请求参数
	if err := c.ShouldBindJSON(&t); err != nil {
		system.Failed("请求参数异常!!!", c)
		return
	}
	// 校验必要参数
	if err := validTaskInfo(t); err != nil {
		system.Failed(err.Error(), c)
		return
	}
	// 获取未更新的task信息
	task, err := logic.CL.GetFilmCrontabById(t.Id)
	if err != nil {
		system.Failed(fmt.Sprint("更新失败: ", err.Error()), c)
		return
	}
	// 将task的可变更属性进行更新
	task.Ids = t.Ids
	task.Time = t.Time
	task.State = t.State
	task.Remark = t.Remark
	// 将变更后的task更新到系统中
	logic.CL.UpdateFilmCron(task)
	system.SuccessOnlyMsg(fmt.Sprintf("定时任务[%s]更新成功", task.Id), c)
}

// ChangeTaskState 开启 | 关闭Id 对应的定时任务
func ChangeTaskState(c *gin.Context) {
	var t = system.FilmCollectTask{}
	// 获取请求参数
	if err := c.ShouldBindJSON(&t); err != nil {
		system.Failed("请求参数异常!!!", c)
		return
	}
	// 获取未更新的task信息
	task, err := logic.CL.GetFilmCrontabById(t.Id)
	if err != nil {
		system.Failed(fmt.Sprint("更新失败: ", err.Error()), c)
		return
	}
	// 修改task的状态
	task.State = t.State
	// 将变更后的task更新到系统中
	logic.CL.UpdateFilmCron(task)
	system.SuccessOnlyMsg(fmt.Sprintf("定时任务[%s]更新成功", task.Id), c)
}

// DelFilmCron 删除定时任务
func DelFilmCron(c *gin.Context) {
	id := c.DefaultQuery("id", "")
	if id == "" {
		system.Failed("定时任务清除失败, 任务ID不能为空", c)
		return
	}
	// 如果Id不为空则执行删除逻辑
	if err := logic.CL.DelFilmCrontab(id); err != nil {
		system.Failed(err.Error(), c)
		return
	}
	system.SuccessOnlyMsg(fmt.Sprintf("定时任务[%s]已删除", id), c)
}

// -------------------------------------------------- 参数校验 --------------------------------------------------

// 定时任务必要属性校验
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 {
	switch vo.Model {
	case 0:
		if vo.Time == 0 {
			return errors.New("参数校验失败, 采集时长不能为零值")
		}
	case 1:
		if vo.Time == 0 {
			return errors.New("参数校验失败, 采集时长不能为零值")
		}
		if vo.Ids == nil || len(vo.Ids) <= 0 {
			return errors.New("参数校验失败, 自定义更新未绑定任何资源站点")
		}
	case 2:
		break
	default:
		return errors.New("参数校验失败, 未定义的任务类型")
	}
	//if vo.Model != 0 && vo.Model != 1 && vo.Model != 2 {
	//	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
}