Merge pull request #1128 from trheyi/main

Implement job management functionality and update job model

Author: trheyi

data/bindata.go

@@ -0,0 +1,116 @@
+package job
+
+import (
+	"github.com/yaoapp/gou/model"
+	"github.com/yaoapp/kun/maps"
+)
+
+// ========================
+// Jobs methods
+// ========================
+
+// ListJobs list jobs
+func ListJobs(param model.QueryParam, page int, pagesize int) (maps.MapStrAny, error) {
+	return nil, nil
+}
+
+// GetActiveJobs get active jobs
+func GetActiveJobs() ([]*Job, error) {
+	return nil, nil
+}
+
+// CountJobs count jobs
+func CountJobs(param model.QueryParam) (int, error) {
+	return 0, nil
+}
+
+// SaveJob save job
+func SaveJob(job *Job) error {
+	return nil
+}
+
+// RemoveJobs remove jobs
+func RemoveJobs(ids []string) error {
+	return nil
+}
+
+// GetJob get job
+func GetJob(id string) (*Job, error) {
+	return nil, nil
+}
+
+// ========================
+// Categories methods
+// ========================
+
+// GetCategories get categories
+func GetCategories(param model.QueryParam) ([]*Category, error) {
+	return nil, nil
+}
+
+// CountCategories count categories
+func CountCategories(param model.QueryParam) (int, error) {
+	return 0, nil
+}
+
+// RemoveCategories remove categories
+func RemoveCategories(ids []string) error {
+	return nil
+}
+
+// SaveCategory save category
+func SaveCategory(category *Category) error {
+	return nil
+}
+
+// ========================
+// Logs methods
+// ========================
+
+// ListLogs get logs
+func ListLogs(id string, param model.QueryParam, page int, pagesize int) (maps.MapStrAny, error) {
+	return nil, nil
+}
+
+// SaveLog save log
+func SaveLog(log *Log) error {
+	return nil
+}
+
+// RemoveLogs remove logs
+func RemoveLogs(ids []string) error {
+	return nil
+}
+
+// ========================
+// Executions methods
+// ========================
+
+// GetExecutions get executions
+func GetExecutions(id string) ([]*Execution, error) {
+	return nil, nil
+}
+
+// CountExecutions count executions
+func CountExecutions(id string, param model.QueryParam) (int, error) {
+	return 0, nil
+}
+
+// RemoveExecutions remove executions
+func RemoveExecutions(ids []string) error {
+	return nil
+}
+
+// GetExecution get execution
+func GetExecution(id string, param model.QueryParam) (*Execution, error) {
+	return nil, nil
+}
+
+// ========================
+// Live progress methods
+// ========================
+
+// GetProgress get progress with callback
+func GetProgress(id string, cb func(progress *Progress)) (*Progress, error) {
+	return nil, nil
+}

job/data.go

@@ -0,0 +1,63 @@
+package job
+
+// Add add a new execution to the job
+func (j *Job) Add(priority int, handler HandlerFunc) error {
+	return nil
+}
+
+// GetExecutions get executions
+func (j *Job) GetExecutions() ([]*Execution, error) {
+	return nil, nil
+}
+
+// GetExecution get execution
+func (j *Job) GetExecution(id string) (*Execution, error) {
+	return nil, nil
+}
+
+// Log log
+func (e *Execution) Log(level LogLevel, format string, args ...interface{}) error {
+	return nil
+}
+
+// Info info log
+func (e *Execution) Info(format string, args ...interface{}) error {
+	return e.Log(Info, format, args...)
+}
+
+// Debug debug log
+func (e *Execution) Debug(format string, args ...interface{}) error {
+	return e.Log(Debug, format, args...)
+}
+
+// Warn warn log
+func (e *Execution) Warn(format string, args ...interface{}) error {
+	return e.Log(Warn, format, args...)
+}
+
+// Error error log
+func (e *Execution) Error(format string, args ...interface{}) error {
+	return e.Log(Error, format, args...)
+}
+
+// Fatal fatal log
+func (e *Execution) Fatal(format string, args ...interface{}) error {
+	return e.Log(Fatal, format, args...)
+}
+
+// Panic panic log
+func (e *Execution) Panic(format string, args ...interface{}) error {
+	return e.Log(Panic, format, args...)
+}
+
+// Trace trace log
+func (e *Execution) Trace(format string, args ...interface{}) error {
+	return e.Log(Trace, format, args...)
+}
+
+// SetProgress set the progress
+func (e *Execution) SetProgress(progress int, message string) error {
+	p := e.Job.Progress()
+	p.Set(progress, message)
+	return nil
+}

job/execution.go

@@ -0,0 +1,6 @@
+package job
+
+// ProgressManager the progress manager
+type ProgressManager interface {
+	Set(progress int, message string) error
+}

job/interfaces.go

@@ -1 +1,114 @@
 package job
+
+import jsoniter "github.com/json-iterator/go"
+
+// Once create a new job
+func Once(mode ModeType, data map[string]interface{}) (*Job, error) {
+	data["mode"] = mode
+	data["schedule_type"] = ScheduleTypeOnce
+	raw, err := jsoniter.Marshal(data)
+	if err != nil {
+		return nil, err
+	}
+	return makeJob(raw)
+}
+
+// Cron create a new job
+func Cron(mode ModeType, data map[string]interface{}, expression string) (*Job, error) {
+	data["mode"] = mode
+	data["schedule_type"] = ScheduleTypeCron
+	data["schedule_expression"] = expression
+	raw, err := jsoniter.Marshal(data)
+	if err != nil {
+		return nil, err
+	}
+	return makeJob(raw)
+}
+
+// Daemon create a new job
+func Daemon(mode ModeType, data map[string]interface{}) (*Job, error) {
+	data["mode"] = mode
+	data["schedule_type"] = ScheduleTypeDaemon
+	raw, err := jsoniter.Marshal(data)
+	if err != nil {
+		return nil, err
+	}
+	return makeJob(raw)
+}
+
+// Start start the job
+func (j *Job) Start() error {
+	return nil
+}
+
+// Cancel cancel the job
+func (j *Job) Cancel() error {
+	return nil
+}
+
+// SetData set the data of the job
+func (j *Job) SetData(data map[string]interface{}) *Job {
+	return j
+}
+
+// SetConfig set the config of the job
+func (j *Job) SetConfig(config map[string]interface{}) *Job {
+	j.Config = config
+	return j
+}
+
+// SetName set the name of the job
+func (j *Job) SetName(name string) *Job {
+	j.Name = name
+	return j
+}
+
+// SetDescription set the description of the job
+func (j *Job) SetDescription(description string) *Job {
+	j.Description = &description
+	return j
+}
+
+// SetCategory set the category of the job
+func (j *Job) SetCategory(category string) *Job {
+	j.CategoryID = category
+	return j
+}
+
+// SetMaxWorkerNums set the max worker nums of the job
+func (j *Job) SetMaxWorkerNums(maxWorkerNums int) *Job {
+	j.MaxWorkerNums = maxWorkerNums
+	return j
+}
+
+// SetStatus set the status of the job
+func (j *Job) SetStatus(status string) *Job {
+	j.Status = status
+	return j
+}
+
+// SetMaxRetryCount set the max retry count of the job
+func (j *Job) SetMaxRetryCount(maxRetryCount int) *Job {
+	j.MaxRetryCount = maxRetryCount
+	return j
+}
+
+// SetDefaultTimeout set the default timeout of the job
+func (j *Job) SetDefaultTimeout(defaultTimeout int) *Job {
+	j.DefaultTimeout = &defaultTimeout
+	return j
+}
+
+// SetMode set the mode of the job
+func (j *Job) SetMode(mode ModeType) {
+	j.Mode = mode
+}
+
+func makeJob(data []byte) (*Job, error) {
+	var job Job
+	err := jsoniter.Unmarshal(data, &job)
+	if err != nil {
+		return nil, err
+	}
+	return &job, nil
+}