From 8dfb8feb32bb5e4e460e23dcde42612a26fa2bcb Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期日, 27 八月 2023 01:11:28 +0800
Subject: [PATCH] fix

---
 service/task.go |   29 ++++++++++++++++++++++-------
 1 files changed, 22 insertions(+), 7 deletions(-)

diff --git a/service/task.go b/service/task.go
index 3cb816f..5a1634e 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").
+	orderSearch.SetOrder("start_time asc").
 		SetStartTimeMax(nowTs).
-		SetStatus(model.OrderStatusUnFinished)
+		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
@@ -47,7 +47,7 @@
 		if err != nil {
 			return nil, ecode.DBErr
 		}
-		return slf.GetTask() //褰撳墠璁㈠崟宸插畬鍏紝缁х画鏌ヨ鏈畬鎴愯鍗�
+		return slf.GetTask() //褰撳墠璁㈠崟宸插畬鎴愶紝缁х画鏌ヨ鏈畬鎴愯鍗�
 	}
 	if err != nil {
 		return nil, ecode.DBErr
@@ -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,
 	})
 }
@@ -75,3 +81,12 @@
 func (slf TaskService) GetOrderByWorkOrderId(workOrderId string) (order *model.Order, err error) {
 	return model.NewOrderSearch(nil).SetWorkOrderId(workOrderId).First()
 }
+
+// GetCurrentTask 鑾峰彇寰呭畬鎴愭垨杩涜涓殑浠诲姟锛� 姣忎釜宸ュ崟鍙湁涓�涓伐搴�
+func (slf TaskService) GetCurrentTask() (workOrder *model.Order, err error) {
+	nowTs := time.Now().Unix()
+	orderSearch := model.NewOrderSearch(nil)
+	return orderSearch.SetOrder("start_time asc").
+		SetStartTimeMin(nowTs).
+		SetStatusList([]model.OrderStatus{model.OrderStatusWaitProcess, model.OrderStatusProcessing}).First()
+}

--
Gitblit v1.8.0