项目目录优化，任务模块后端代码

apps/job/jobs/examples.go

@@ -0,0 +1,38 @@
+package jobs
+
+import (
+	"fmt"
+	"time"
+)
+
+// 需要将定义的struct 添加到字典中；
+// 字典 key 可以配置到 自动任务 调用目标 中；
+func InitJob() {
+	jobList = map[string]JobsExec{
+		"ExamplesOne": ExamplesOne{},
+		// ...
+	}
+}
+
+// 新添加的job 必须按照以下格式定义，并实现Exec函数
+type ExamplesOne struct {
+}
+
+func (t ExamplesOne) Exec(arg interface{}) error {
+	str := time.Now().Format(timeFormat) + " [INFO] JobCore ExamplesOne exec success"
+	// TODO: 这里需要注意 Examples 传入参数是 string 所以 arg.(string)；请根据对应的类型进行转化；
+	switch arg.(type) {
+
+	case string:
+		if arg.(string) != "" {
+			fmt.Println("string", arg.(string))
+			fmt.Println(str, arg.(string))
+		} else {
+			fmt.Println("arg is nil")
+			fmt.Println(str, "arg is nil")
+		}
+		break
+	}
+
+	return nil
+}

apps/job/jobs/jobbase.go

@@ -0,0 +1,195 @@
+package jobs
+
+import (
+	"fmt"
+	"pandax/apps/job/entity"
+	"pandax/apps/job/services"
+	"pandax/base/global"
+	"pandax/base/httpclient"
+	"sync"
+	"time"
+
+	"github.com/robfig/cron/v3"
+)
+
+var timeFormat = "2006-01-02 15:04:05"
+var retryCount = 3
+
+var jobList map[string]JobsExec
+var lock sync.Mutex
+
+var Crontab = new(cron.Cron)
+
+type JobCore struct {
+	InvokeTarget   string
+	Name           string
+	JobId          int64
+	EntryId        int
+	CronExpression string // 任务表达式
+	MisfirePolicy  string // 错误执行策略
+	Args           string
+}
+
+// 任务类型 http
+type HttpJob struct {
+	cron *cron.Cron
+	JobCore
+}
+
+type ExecJob struct {
+	cron *cron.Cron
+	JobCore
+}
+
+func (e *ExecJob) Run() {
+	startTime := time.Now()
+	var obj = jobList[e.InvokeTarget]
+	if obj == nil {
+		global.Log.Warn("[Job] ExecJob Run job nil")
+		if e.MisfirePolicy == "2" {
+			Remove(e.cron, e.EntryId)
+		}
+		return
+	}
+	err := CallExec(obj.(JobsExec), e.Args)
+	if err != nil {
+		// 如果失败暂停一段时间重试
+		fmt.Println(time.Now().Format(timeFormat), " [ERROR] mission failed! ", err)
+		if e.MisfirePolicy == "2" {
+			Remove(e.cron, e.EntryId)
+			return
+		}
+	}
+	// 结束时间
+	endTime := time.Now()
+
+	// 执行时间
+	latencyTime := endTime.Sub(startTime)
+	//TODO: 待完善部分
+	//str := time.Now().Format(timeFormat) + " [INFO] JobCore " + string(e.EntryId) + "exec success , spend :" + latencyTime.String()
+	//ws.SendAll(str)
+	global.Log.Info("[Job] JobCore %s exec success , spend :%v", e.Name, latencyTime)
+	// 执行一次
+	if e.MisfirePolicy == "1" {
+		Remove(e.cron, e.EntryId)
+	}
+	return
+}
+
+//http 任务接口
+func (h *HttpJob) Run() {
+
+	startTime := time.Now()
+	var count = 0
+LOOP:
+	if count < retryCount {
+		_, err := httpclient.NewRequest(h.InvokeTarget).Get().BodyToString()
+		if err != nil {
+			time.Sleep(time.Duration(count+1) * 5 * time.Second)
+			count = count + 1
+			goto LOOP
+		}
+	}
+	// 结束时间
+	endTime := time.Now()
+
+	// 执行时间
+	latencyTime := endTime.Sub(startTime)
+
+	global.Log.Infof("[Job] JobCore %s exec success , spend :%v", h.Name, latencyTime)
+	if h.MisfirePolicy == "1" {
+		Remove(h.cron, h.EntryId)
+	}
+	return
+}
+
+func Setup() {
+	Crontab = NewWithSeconds()
+	// 获取系统job 0是默认，1是系统
+	jl := services.JobModelDao.FindList(entity.SysJob{JobGroup: "1"})
+	jobList := *jl
+	if len(jobList) == 0 {
+		global.Log.Info(time.Now().Format(timeFormat), " [INFO] JobCore total:0")
+		return
+	}
+	err := services.JobModelDao.RemoveAllEntryID()
+	if err != nil {
+		global.Log.Info(time.Now().Format(timeFormat), " [ERROR] JobCore remove entry_id error", err)
+	}
+	sysJob := entity.SysJob{}
+	for i := 0; i < len(jobList); i++ {
+		if jobList[i].Status != "0" && jobList[i].EntryId > 0 {
+			continue
+		}
+		if jobList[i].JobType == "1" {
+			j := &HttpJob{}
+			j.InvokeTarget = jobList[i].InvokeTarget
+			j.CronExpression = jobList[i].CronExpression
+			j.JobId = jobList[i].JobId
+			j.Name = jobList[i].JobName
+			j.MisfirePolicy = jobList[i].MisfirePolicy
+			sysJob.EntryId, err = AddJob(Crontab, j)
+		} else if jobList[i].JobType == "2" {
+			j := &ExecJob{}
+			j.InvokeTarget = jobList[i].InvokeTarget
+			j.CronExpression = jobList[i].CronExpression
+			j.JobId = jobList[i].JobId
+			j.Name = jobList[i].JobName
+			j.Args = jobList[i].Args
+			j.MisfirePolicy = jobList[i].MisfirePolicy
+			sysJob.EntryId, err = AddJob(Crontab, j)
+		}
+		sysJob.JobId = jobList[i].JobId
+		services.JobModelDao.Update(sysJob)
+	}
+	// 其中任务
+	Crontab.Start()
+	global.Log.Info(time.Now().Format(timeFormat), " [INFO] JobCore start success.")
+	// 关闭任务
+	defer Crontab.Stop()
+	select {}
+}
+
+// 添加任务 AddJob(invokeTarget string, jobId int, jobName string, cronExpression string)
+func AddJob(c *cron.Cron, job Job) (int, error) {
+	if job == nil {
+		return 0, nil
+	}
+	return job.addJob(c)
+}
+
+func (h *HttpJob) addJob(c *cron.Cron) (int, error) {
+	id, err := c.AddJob(h.CronExpression, h)
+	if err != nil {
+		return 0, err
+	}
+	h.cron = c
+	EntryId := int(id)
+	h.EntryId = EntryId
+	return EntryId, nil
+}
+
+func (h *ExecJob) addJob(c *cron.Cron) (int, error) {
+	id, err := c.AddJob(h.CronExpression, h)
+	if err != nil {
+		return 0, err
+	}
+	h.cron = c
+	EntryId := int(id)
+	h.EntryId = EntryId
+	return EntryId, nil
+}
+
+// 移除任务(停止任务)
+func Remove(c *cron.Cron, entryID int) {
+	c.Remove(cron.EntryID(entryID))
+	var job entity.SysJob
+	job.EntryId = entryID
+	services.JobModelDao.RemoveEntryID(entryID)
+}
+
+func NewWithSeconds() *cron.Cron {
+	secondParser := cron.NewParser(cron.Second | cron.Minute |
+		cron.Hour | cron.Dom | cron.Month | cron.DowOptional | cron.Descriptor)
+	return cron.New(cron.WithParser(secondParser), cron.WithChain())
+}

apps/job/jobs/type.go

@@ -0,0 +1,16 @@
+package jobs
+
+import "github.com/robfig/cron/v3"
+
+type Job interface {
+	Run()
+	addJob(*cron.Cron) (int, error)
+}
+
+type JobsExec interface {
+	Exec(arg interface{}) error
+}
+
+func CallExec(e JobsExec, arg interface{}) error {
+	return e.Exec(arg)
+}