jiacrontab/models/crontab.go

package models

import (
	"database/sql/driver"
	"encoding/json"
	"errors"
	"jiacrontab/pkg/util"
	"time"

	"gorm.io/gorm"
)

// JobStatus 任务状态
type JobStatus int

const (
	// StatusJobUnaudited 未审核
	StatusJobUnaudited JobStatus = 0
	// StatusJobOk 等待调度
	StatusJobOk JobStatus = 1
	// StatusJobTiming 定时中
	StatusJobTiming JobStatus = 2
	// StatusJobRunning 执行中
	StatusJobRunning JobStatus = 3
	// StatusJobStop 已停止
	StatusJobStop JobStatus = 4
)

type CrontabJob struct {
	gorm.Model
	Name             string      `json:"name" gorm:"index;not null"`
	GroupID          uint        `json:"groupID" grom:"index"`
	Command          StringSlice `json:"command" gorm:"type:varchar(1000)"`
	Code             string      `json:"code" gorm:"type:TEXT"`
	DependJobs       DependJobs  `json:"dependJobs" gorm:"type:TEXT"`
	LastCostTime     float64     `json:"lastCostTime"`
	LastExecTime     time.Time   `json:"lastExecTime"`
	NextExecTime     time.Time   `json:"nextExecTime"`
	Failed           bool        `json:"failed"`
	LastExitStatus   string      `json:"lastExitStatus" grom:"index"`
	CreatedUserID    uint        `json:"createdUserId"`
	CreatedUsername  string      `json:"createdUsername"`
	UpdatedUserID    uint        `json:"updatedUserID"`
	UpdatedUsername  string      `json:"updatedUsername"`
	WorkUser         string      `json:"workUser"`
	WorkIp           StringSlice `json:"workIp" gorm:"type:varchar(1000)"`
	WorkEnv          StringSlice `json:"workEnv" gorm:"type:varchar(1000)"`
	WorkDir          string      `json:"workDir"`
	KillChildProcess bool        `json:"killChildProcess"`
	Timeout          int         `json:"timeout"`
	ProcessNum       int         `json:"processNum"`
	ErrorMailNotify  bool        `json:"errorMailNotify"`
	ErrorAPINotify   bool        `json:"errorAPINotify"`
	ErrorDingdingNotify   bool   `json:"errorDingdingNotify"`
	RetryNum         int         `json:"retryNum"`
	Status           JobStatus   `json:"status"`
	IsSync           bool        `json:"isSync"` // 脚本是否同步执行
	MailTo           StringSlice `json:"mailTo" gorm:"type:varchar(1000)"`
	APITo            StringSlice `json:"APITo"  gorm:"type:varchar(1000)"`
	DingdingTo       StringSlice `json:"DingdingTo"  gorm:"type:varchar(1000)"`
	MaxConcurrent    uint        `json:"maxConcurrent"` // 脚本最大并发量
	TimeoutTrigger   StringSlice `json:"timeoutTrigger" gorm:"type:varchar(20)"`
	TimeArgs         TimeArgs    `json:"timeArgs" gorm:"type:TEXT"`
}

type StringSlice []string

func (s *StringSlice) Scan(v interface{}) error {

	switch val := v.(type) {
	case string:
		return json.Unmarshal([]byte(val), s)
	case []byte:
		return json.Unmarshal(val, s)
	default:
		return errors.New("not support")
	}
}

func (s StringSlice) MarshalJSON() ([]byte, error) {
	if s == nil {
		s = make(StringSlice, 0)
	}
	return json.Marshal([]string(s))
}

func (s StringSlice) Value() (driver.Value, error) {
	if s == nil {
		s = make(StringSlice, 0)
	}
	bts, err := json.Marshal(s)
	return string(bts), err
}

type DependJobs []DependJob

func (d *DependJobs) Scan(v interface{}) error {
	switch val := v.(type) {
	case string:
		return json.Unmarshal([]byte(val), d)
	case []byte:
		return json.Unmarshal(val, d)
	default:
		return errors.New("not support")
	}

}

func (d DependJobs) Value() (driver.Value, error) {

	if d == nil {
		d = make(DependJobs, 0)
	}

	for k, _ := range d {
		d[k].ID = util.UUID()
	}

	bts, err := json.Marshal(d)
	return string(bts), err
}

func (d DependJobs) MarshalJSON() ([]byte, error) {
	if d == nil {
		d = make(DependJobs, 0)
	}
	type m DependJobs
	return json.Marshal(m(d))
}

type TimeArgs struct {
	Weekday string `json:"weekday"`
	Month   string `json:"month"`
	Day     string `json:"day"`
	Hour    string `json:"hour"`
	Minute  string `json:"minute"`
	Second  string `json:"second"`
}

func (c *TimeArgs) Scan(v interface{}) error {
	switch val := v.(type) {
	case string:
		return json.Unmarshal([]byte(val), c)
	case []byte:
		return json.Unmarshal(val, c)
	default:
		return errors.New("not support")
	}

}

func (c TimeArgs) Value() (driver.Value, error) {
	bts, err := json.Marshal(c)
	return string(bts), err
}

type DependJob struct {
	Dest     string   `json:"dest"`
	From     string   `json:"from"`
	JobID    uint     `json:"jobID"`
	ID       string   `json:"id"`
	WorkUser string   `json:"user"`
	WorkDir  string   `json:"workDir"`
	Command  []string `json:"command"`
	Code     string   `json:"code"`
	Timeout  int64    `json:"timeout"`
}

type PipeComamnds [][]string

func (p *PipeComamnds) Scan(v interface{}) error {
	switch val := v.(type) {
	case string:
		return json.Unmarshal([]byte(val), p)
	case []byte:
		return json.Unmarshal(val, p)
	default:
		return errors.New("not support")
	}

}

func (p PipeComamnds) Value() (driver.Value, error) {
	if p == nil {
		p = make(PipeComamnds, 0)
	}
	bts, err := json.Marshal(p)
	return string(bts), err
}

func (d PipeComamnds) MarshalJSON() ([]byte, error) {
	if d == nil {
		d = make(PipeComamnds, 0)
	}
	type m PipeComamnds
	return json.Marshal(m(d))
}

type CrontabArgs struct {
	Weekday string
	Month   string
	Day     string
	Hour    string
	Minute  string
}

func (c *CrontabArgs) Scan(v interface{}) error {
	switch val := v.(type) {
	case string:
		return json.Unmarshal([]byte(val), c)
	case []byte:
		return json.Unmarshal(val, c)
	default:
		return errors.New("not support")
	}

}

func (c CrontabArgs) Value() (driver.Value, error) {
	bts, err := json.Marshal(c)
	return string(bts), err
}