From b331c9990a0396301e934daffe095f99d62d1c89 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期三, 27 九月 2023 15:58:49 +0800
Subject: [PATCH] 任务获取修改,状态2只返回已开始和已结束的

---
 service/task.go |   23 +++++++++--------------
 1 files changed, 9 insertions(+), 14 deletions(-)

diff --git a/service/task.go b/service/task.go
index 9b7eb16..dd1c11c 100644
--- a/service/task.go
+++ b/service/task.go
@@ -23,16 +23,8 @@
 	return &TaskService{}
 }
 
-type TaskMode int
-
-const (
-	TaskModeUnStarted    TaskMode = 1 //鏈紑濮嬬殑
-	TaskModeCurrent      TaskMode = 2
-	TaskModeLastFinished TaskMode = 3 //涓婁竴涓粨鏉熺殑
-)
-
 // GetTask 鑾峰彇浠诲姟锛屾湭瀹屾垚鐨勫紑濮嬫椂闂村皬浜庣瓑浜庡綋鍓嶆椂闂达紝缁撴潫鏃堕棿澶т簬褰撳墠鏃堕棿鐨勪换鍔�
-func (slf TaskService) GetTask(page, pageSize int, mode TaskMode, channels []int32) (taskResp *response.TaskResponse, code int) {
+func (slf TaskService) GetTask(page, pageSize int, mode constvar.TaskMode, channels []int32) (taskResp *response.TaskResponse, code int) {
 	var taskList []*response.TaskData
 	var count int64
 	var workers []*common.ProcedureWorker
@@ -54,17 +46,17 @@
 		SetDeviceId(conf.Conf.System.DeviceId).
 		SetPage(page, pageSize)
 
-	if mode == TaskModeUnStarted {
+	if mode == constvar.TaskModeUnStarted {
 		search.SetStatus(model.ProcedureStatusWaitProcess).
 			SetStartTimeMax(nowTs).
 			SetEndTimeMin(nowTs).
 			SetOrder("start_time asc")
-	} else if mode == TaskModeCurrent {
-		search.SetStatusNot(model.ProcedureStatusFinished).
+	} else if mode == constvar.TaskModeCurrent {
+		search.SetStatus(model.ProcedureStatusProcessing).
 			SetStartTimeMax(nowTs).
 			SetEndTimeMin(nowTs).
 			SetOrder("status desc, start_time asc")
-	} else if mode == TaskModeLastFinished {
+	} else if mode == constvar.TaskModeLastFinished {
 		search.SetStatus(model.ProcedureStatusFinished).SetOrder("updated_at desc")
 		if len(channels) > 0 {
 			search.SetChannels(channels)
@@ -75,7 +67,7 @@
 	if err != nil {
 		return taskResp, ecode.DBErr
 	}
-	count, err = search.Count()
+
 	if err != nil {
 		return taskResp, ecode.DBErr
 	}
@@ -114,6 +106,9 @@
 		taskList = append(taskList, &taskData)
 		workers = append(workers, procedure.ProceduresInfo.Workers...)
 	}
+
+	count, _ = search.SetStatus(model.ProcedureStatusWaitProcess).SetStartTimeMax(nowTs).SetEndTimeMin(nowTs).Count()
+
 	taskResp = &response.TaskResponse{
 		Tasks:     taskList,
 		TaskCount: count,

--
Gitblit v1.8.0