diff --git a/models/migration.go b/models/migration.go index ba00ed0..2ce201d 100644 --- a/models/migration.go +++ b/models/migration.go @@ -53,11 +53,12 @@ func (migration *Migration) Upgrade(oldVersionId int) { return } - versionIds := []int{110, 122, 130} + versionIds := []int{110, 122, 130, 140} upgradeFuncs := []func(*xorm.Session) error{ migration.upgradeFor110, migration.upgradeFor122, migration.upgradeFor130, + migration.upgradeFor140, } startIndex := -1 @@ -157,7 +158,7 @@ func (migration *Migration) upgradeFor122(session *xorm.Session) error { return err } -// 升级到1.2.3版本 +// 升级到v1.3版本 func (migration *Migration) upgradeFor130(session *xorm.Session) error { logger.Info("开始升级到v1.3") @@ -167,5 +168,19 @@ func (migration *Migration) upgradeFor130(session *xorm.Session) error { logger.Info("已升级到v1.3\n") + return err +} + +// 升级到v1.4版本 +func (migration *Migration) upgradeFor140(session *xorm.Session) error { + logger.Info("开始升级到v1.4") + + tableName := TablePrefix + "task" + // task表增加字段 + // retry_interval 重试间隔时间(秒) + _, err := session.Exec(fmt.Sprintf("ALTER TABLE %s ADD COLUMN retry_interval SMALLINT NOT NULL DEFAULT 0", tableName)) + + logger.Info("已升级到v1.4\n") + return err } \ No newline at end of file diff --git a/models/task.go b/models/task.go index dffdba2..a9ab26f 100644 --- a/models/task.go +++ b/models/task.go @@ -32,17 +32,18 @@ const ( type Task struct { Id int `xorm:"int pk autoincr"` Name string `xorm:"varchar(32) notnull"` // 任务名称 - Level TaskLevel `xorm:"smallint notnull index default 1"` // 任务等级 1: 主任务 2: 依赖任务 + Level TaskLevel `xorm:"tinyint notnull index default 1"` // 任务等级 1: 主任务 2: 依赖任务 DependencyTaskId string `xorm:"varchar(64) notnull default ''"` // 依赖任务ID,多个ID逗号分隔 - DependencyStatus TaskDependencyStatus `xorm:"smallint notnull default 1"` // 依赖关系 1:强依赖 主任务执行成功, 依赖任务才会被执行 2:弱依赖 + DependencyStatus TaskDependencyStatus `xorm:"tinyint notnull default 1"` // 依赖关系 1:强依赖 主任务执行成功, 依赖任务才会被执行 2:弱依赖 Spec string `xorm:"varchar(64) notnull"` // crontab Protocol TaskProtocol `xorm:"tinyint notnull index"` // 协议 1:http 2:系统命令 Command string `xorm:"varchar(256) notnull"` // URL地址或shell命令 Timeout int `xorm:"mediumint notnull default 0"` // 任务执行超时时间(单位秒),0不限制 Multi int8 `xorm:"tinyint notnull default 1"` // 是否允许多实例运行 RetryTimes int8 `xorm:"tinyint notnull default 0"` // 重试次数 - NotifyStatus int8 `xorm:"smallint notnull default 1"` // 任务执行结束是否通知 0: 不通知 1: 失败通知 2: 执行结束通知 - NotifyType int8 `xorm:"smallint notnull default 0"` // 通知类型 1: 邮件 2: slack + RetryInterval int16 `xorm:"smallint notnull default 0"` // 重试间隔时间 + NotifyStatus int8 `xorm:"tinyint notnull default 1"` // 任务执行结束是否通知 0: 不通知 1: 失败通知 2: 执行结束通知 + NotifyType int8 `xorm:"tinyint notnull default 0"` // 通知类型 1: 邮件 2: slack NotifyReceiverId string `xorm:"varchar(256) notnull default '' "` // 通知接受者ID, setting表主键ID，多个ID逗号分隔 Tag string `xorm:"varchar(32) notnull default ''"` Remark string `xorm:"varchar(100) notnull default ''"` // 备注 @@ -83,7 +84,9 @@ func (task *Task) CreateTestTask() { func (task *Task) UpdateBean(id int) (int64, error) { return Db.ID(id). - Cols("name,spec,protocol,command,timeout,multi,retry_times,remark,notify_status,notify_type,notify_receiver_id, dependency_task_id, dependency_status, tag"). + Cols("name,spec,protocol,command,timeout,multi," + + "retry_times,retry_interval,remark,notify_status," + + "notify_type,notify_receiver_id, dependency_task_id, dependency_status, tag"). Update(task) } diff --git a/routers/task/task.go b/routers/task/task.go index eb4c7d8..74876c4 100644 --- a/routers/task/task.go +++ b/routers/task/task.go @@ -28,6 +28,7 @@ type TaskForm struct { Timeout int `binding:"Range(0,86400)"` Multi int8 `binding:"In(1,2)"` RetryTimes int8 + RetryInterval int16 HostId string Tag string Remark string @@ -135,6 +136,7 @@ func Store(ctx *macaron.Context, form TaskForm) string { taskModel.Remark = form.Remark taskModel.Multi = form.Multi taskModel.RetryTimes = form.RetryTimes + taskModel.RetryInterval = form.RetryInterval if taskModel.Multi != 1 { taskModel.Multi = 0 } @@ -162,6 +164,10 @@ func Store(ctx *macaron.Context, form TaskForm) string { return json.CommonFailure("任务重试次数取值0-10") } + if taskModel.RetryInterval > 3600 || taskModel.RetryInterval < 0 { + return json.CommonFailure("任务重试间隔时间取值0-3600") + } + if taskModel.DependencyStatus != models.TaskDependencyStatusStrong && taskModel.DependencyStatus != models.TaskDependencyStatusWeak { return json.CommonFailure("请选择依赖关系") diff --git a/service/task.go b/service/task.go index 2ce94f6..3c02620 100644 --- a/service/task.go +++ b/service/task.go @@ -404,8 +404,12 @@ func execJob(handler Handler, taskModel models.Task) TaskResult { i++ if i < execTimes { logger.Warnf("任务执行失败#任务id-%d#重试第%d次#输出-%s#错误-%s", taskModel.Id, i, output, err.Error()) - // 重试间隔时间，每次递增1分钟 - time.Sleep(time.Duration(i) * time.Minute) + if taskModel.RetryInterval > 0 { + time.Sleep(time.Duration(taskModel.RetryInterval) * time.Second) + } else { + // 默认重试间隔时间，每次递增1分钟 + time.Sleep(time.Duration(i) * time.Minute) + } } } diff --git a/templates/task/task_form.html b/templates/task/task_form.html index ab35db7..f3a6f69 100644 --- a/templates/task/task_form.html +++ b/templates/task/task_form.html @@ -118,15 +118,11 @@ -
+
-
- - -
-
+ +
+
+ + +
+
+ + +
+
+
@@ -423,6 +430,15 @@ } ] }, + retryInterval: { + identifier : 'retry_interval', + rules: [ + { + type : 'integer[0..3600]', + prompt : '重试间隔时间0-3600' + } + ] + }, remark: { identifier : 'remark', rules: [