You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
3.5 KiB
78 lines
3.5 KiB
package jobs_gorm
|
|
|
|
import (
|
|
"go.dtapp.net/gojobs/jobs_common"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// Task 任务
|
|
type Task struct {
|
|
Id uint `gorm:"primaryKey" json:"id"` // 记录编号
|
|
Status string `json:"status"` // 状态码
|
|
Params string `json:"params"` // 参数
|
|
ParamsType string `json:"params_type"` // 参数类型
|
|
StatusDesc string `json:"status_desc"` // 状态描述
|
|
Frequency int64 `json:"frequency"` // 频率（秒单位）
|
|
Number int64 `json:"number"` // 当前次数
|
|
MaxNumber int64 `json:"max_number"` // 最大次数
|
|
RunId string `json:"run_id"` // 执行编号
|
|
CustomId string `json:"custom_id"` // 自定义编号
|
|
CustomSequence int64 `json:"custom_sequence"` // 自定义顺序
|
|
Type string `json:"type"` // 类型
|
|
CreatedIp string `json:"created_ip"` // 创建外网IP
|
|
SpecifyIp string `json:"specify_ip"` // 指定外网IP
|
|
UpdatedIp string `json:"updated_ip"` // 更新外网IP
|
|
Result string `json:"result"` // 结果
|
|
CreatedAt string `gorm:"type:text" json:"created_at"` // 创建时间
|
|
UpdatedAt string `gorm:"type:text" json:"updated_at"` // 更新时间
|
|
DeletedAt gorm.DeletedAt `gorm:"type:text;index" json:"deleted_at"` // 删除时间
|
|
}
|
|
|
|
func (m *Task) TableName() string {
|
|
return "task"
|
|
}
|
|
|
|
// TaskTake 查询任务
|
|
func (jobsGorm *JobsGorm) TaskTake(tx *gorm.DB, customId string) (result Task) {
|
|
tx.Where("custom_id = ?", customId).Where("status = ?", jobs_common.TASK_IN).Take(&result)
|
|
return result
|
|
}
|
|
|
|
// TaskCustomIdTake 查询任务
|
|
func (jobsGorm *JobsGorm) TaskCustomIdTake(tx *gorm.DB, Type, customId string) (result Task) {
|
|
tx.Where("type = ?", Type).Where("custom_id = ?", customId).Take(&result)
|
|
return result
|
|
}
|
|
|
|
// TaskCustomIdTakeStatus 查询任务
|
|
func (jobsGorm *JobsGorm) TaskCustomIdTakeStatus(tx *gorm.DB, Type, customId, status string) (result Task) {
|
|
tx.Where("type = ?", Type).Where("custom_id = ?", customId).Where("status = ?", status).Take(&result)
|
|
return result
|
|
}
|
|
|
|
// TaskFind 查询任务
|
|
func (jobsGorm *JobsGorm) TaskFind(tx *gorm.DB, frequency int64) (results []Task) {
|
|
tx.Table("task").Select("task.*").Where("task.frequency = ?", frequency).Where("task.status = ?", jobs_common.TASK_IN).Where("task_ip.ips = ?", jobsGorm.outsideIp).Order("task.id asc").Joins("left join task_ip on task_ip.task_type = task.type").Find(&results)
|
|
return jobsGorm.taskFindCheck(results)
|
|
}
|
|
|
|
// TaskFindAll 查询任务
|
|
func (jobsGorm *JobsGorm) TaskFindAll(tx *gorm.DB, frequency int64) (results []Task) {
|
|
tx.Where("frequency = ?", frequency).Where("status = ?", jobs_common.TASK_IN).Order("id asc").Find(&results)
|
|
return results
|
|
}
|
|
|
|
// 检查任务
|
|
func (jobsGorm *JobsGorm) taskFindCheck(lists []Task) (results []Task) {
|
|
for _, v := range lists {
|
|
if v.SpecifyIp == "" {
|
|
results = append(results, v)
|
|
} else {
|
|
if jobsGorm.outsideIp == v.SpecifyIp {
|
|
results = append(results, v)
|
|
}
|
|
}
|
|
}
|
|
return results
|
|
}
|