From c6173b62e0fea3116075d56692b73cc0bc624815 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期日, 27 八月 2023 00:54:56 +0800
Subject: [PATCH] 逻辑修改

---
 service/task.go |   28 +++++++++++++++++-----------
 1 files changed, 17 insertions(+), 11 deletions(-)

diff --git a/service/task.go b/service/task.go
index 5b7be1c..ac5fdca 100644
--- a/service/task.go
+++ b/service/task.go
@@ -19,9 +19,9 @@
 func (slf TaskService) GetTask() (taskData *response.TaskData, code int) {
 	nowTs := time.Now().Unix()
 	orderSearch := model.NewOrderSearch(nil)
-	orderSearch.SetOrder("created_at asc").
-		SetStartTimeMax(nowTs).
-		SetStatus(model.OrderStatusUnFinished)
+	orderSearch.SetOrder("start_time asc").
+		SetStartTimeMin(nowTs).
+		SetStatusList([]model.OrderStatus{model.OrderStatusWaitProcess, model.OrderStatusProcessing})
 	var (
 		err   error
 		order *model.Order
@@ -34,10 +34,10 @@
 	if err != nil {
 		return nil, ecode.DBErr
 	}
-	procedure, err := model.NewProceduresSearch(nil).
+	procedure, err := model.NewProceduresSearch(nil).SetOrder("start_time asc").
 		SetWorkOrderId(order.WorkOrderID).
 		SetStartTimeMax(nowTs).
-		SetStatus(model.ProcedureStatusUnFinished).
+		SetStatusNot(model.ProcedureStatusFinished).
 		First()
 	if err == gorm.ErrRecordNotFound { //璇ヨ鍗曟湰璁惧鍏ㄩ儴宸ュ簭瀹屾垚锛屾妸璁㈠崟缃负瀹屾垚鐘舵�侊紙鍙槸瀵规湰璁惧瀹屾垚锛�
 		err = nil
@@ -66,8 +66,14 @@
 	return procedure, ecode.OK
 }
 
-func (slf TaskService) UpdateProcedureStatus(id int, status model.ProcedureStatus) error {
-	return model.NewProceduresSearch(nil).SetId(id).UpdateByMap(map[string]interface{}{
+func (slf TaskService) UpdateProcedureStatus(db *gorm.DB, id int, status model.ProcedureStatus) error {
+	return model.NewProceduresSearch(db).SetId(id).UpdateByMap(map[string]interface{}{
+		"status": status,
+	})
+}
+
+func (slf TaskService) UpdateOrderStatus(db *gorm.DB, id uint, status model.OrderStatus) error {
+	return model.NewOrderSearch(db).SetId(id).UpdateByMap(map[string]interface{}{
 		"status": status,
 	})
 }
@@ -76,11 +82,11 @@
 	return model.NewOrderSearch(nil).SetWorkOrderId(workOrderId).First()
 }
 
-// GetComingTask 鑾峰彇鏃堕棿鏈埌鐨勬渶鏃╀换鍔�
-func (slf TaskService) GetComingTask() (workOrder *model.Order, err error) {
+// GetCurrentTask 鑾峰彇寰呭畬鎴愭垨杩涜涓殑浠诲姟锛� 姣忎釜宸ュ崟鍙湁涓�涓伐搴�
+func (slf TaskService) GetCurrentTask() (workOrder *model.Order, err error) {
 	nowTs := time.Now().Unix()
 	orderSearch := model.NewOrderSearch(nil)
-	return orderSearch.SetOrder("created_at asc").
+	return orderSearch.SetOrder("start_time asc").
 		SetStartTimeMin(nowTs).
-		SetStatus(model.OrderStatusUnFinished).First()
+		SetStatusList([]model.OrderStatus{model.OrderStatusWaitProcess, model.OrderStatusProcessing}).First()
 }

--
Gitblit v1.8.0