package/go_process_manager
1
0
Fork 0
mirror of https://github.com/lzh-1625/go_process_manager.git synced 2025-10-04 07:36:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

optimal task job get

Browse Source
This commit is contained in:
liuzhihang1
2025-02-13 11:24:28 +08:00
parent 1852f7ab4c
commit b2a1ef94e9
2 changed files with 26 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
internal/app/logic/task_logic.go
View File

@@ -21,6 +21,14 @@ type taskLogic struct {
var TaskLogic taskLogic
func (t *taskLogic) getTaskJob(id int) (*model.TaskJob, error) {
c, ok := t.taskJobMap.Load(id)
if !ok {
return nil, errors.New("don't exist this task id")
}
return c.(*model.TaskJob), nil
}
func (t *taskLogic) InitTaskJob() {
for _, v := range repository.TaskRepository.GetAllTask() {
tj := &model.TaskJob{
@@ -61,11 +69,10 @@ func (t *taskLogic) cronHandle(data *model.TaskJob) func() {
}
func (t *taskLogic) StopTaskJob(id int) error {
c, ok := t.taskJobMap.Load(id)
if !ok {
return errors.New("id不存在")
taskJob, err := t.getTaskJob(id)
if err != nil {
return errors.New("don't exist this task id")
}
taskJob := c.(*model.TaskJob)
if taskJob.Running {
taskJob.Cancel()
}
@@ -73,23 +80,21 @@ func (t *taskLogic) StopTaskJob(id int) error {
}
func (t *taskLogic) StartTaskJob(id int) error {
c, ok := t.taskJobMap.Load(id)
if !ok {
return errors.New("id不存在")
taskJob, err := t.getTaskJob(id)
if err != nil {
return errors.New("don't exist this task id")
}
TaskJob := c.(*model.TaskJob)
TaskJob.Cron.Run()
taskJob.Cron.Run()
return nil
}
func (t *taskLogic) GetAllTaskJob() []model.TaskVo {
result := repository.TaskRepository.GetAllTaskWithProcessName()
for i, v := range result {
item, ok := t.taskJobMap.Load(v.Id)
if !ok {
task, err := t.getTaskJob(v.Id)
if err != nil {
continue
}
task := item.(*model.TaskJob)
result[i].Id = task.Task.Id
result[i].Running = task.Running
result[i].Enable = task.Task.Enable
@@ -133,11 +138,10 @@ func (t *taskLogic) CreateTask(data model.Task) error {
}
func (t *taskLogic) EditTask(data model.Task) error {
v, ok := t.taskJobMap.Load(data.Id)
if !ok {
tj, err := t.getTaskJob(data.Id)
if err != nil {
return errors.New("don't exist this task id")
}
tj := v.(*model.TaskJob)
if tj.Running {
return errors.New("can't edit when task is running")
}
@@ -162,11 +166,10 @@ func (t *taskLogic) EditTask(data model.Task) error {
}
func (t *taskLogic) EditTaskEnable(id int, status bool) error {
v, ok := t.taskJobMap.Load(id)
if !ok {
tj, err := t.getTaskJob(id)
if err != nil {
return errors.New("don't exist this task id")
}
tj := v.(*model.TaskJob)
tj.Task.Enable = status
repository.TaskRepository.EditTaskEnable(id, status)
if tj.Cron != nil {