From 8324f872ef3a4d0c978a9b1d062800c6a1701c12 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 01 十二月 2023 09:58:17 +0800
Subject: [PATCH] fix

---
 service/task.go |  198 +++++++++++++++++++++++++++++++++++++------------
 1 files changed, 149 insertions(+), 49 deletions(-)

diff --git a/service/task.go b/service/task.go
index 113ebe6..66bf691 100644
--- a/service/task.go
+++ b/service/task.go
@@ -5,13 +5,13 @@
 	"apsClient/constvar"
 	"apsClient/model"
 	"apsClient/model/common"
+	"apsClient/model/request"
 	"apsClient/model/response"
-	"apsClient/nsq"
 	"apsClient/pkg/ecode"
 	"apsClient/pkg/logx"
-	"apsClient/pkg/structx"
+	"apsClient/pkg/timex"
 	"fmt"
-	"gorm.io/gorm"
+	"github.com/jinzhu/gorm"
 	"time"
 )
 
@@ -23,7 +23,7 @@
 }
 
 // GetTask 鑾峰彇浠诲姟锛屾湭瀹屾垚鐨勫紑濮嬫椂闂村皬浜庣瓑浜庡綋鍓嶆椂闂达紝缁撴潫鏃堕棿澶т簬褰撳墠鏃堕棿鐨勪换鍔�
-func (slf TaskService) GetTask(page, pageSize int, mode constvar.TaskMode, channels []int32) (taskResp *response.TaskResponse, code int) {
+func (slf TaskService) GetTask(deviceID string, page, pageSize int, mode constvar.TaskMode, channels []int32) (taskResp *response.TaskResponse, code int) {
 	var taskList []*response.TaskData
 	var count int64
 	var workers []*common.ProcedureWorker
@@ -31,7 +31,7 @@
 	taskResp = &response.TaskResponse{
 		Tasks:     taskList,
 		TaskCount: count,
-		Workers:   slf.WorkerDistinct(workers),
+		Workers:   workers,
 	}
 
 	var (
@@ -41,7 +41,7 @@
 		workOrderIds []string
 	)
 	search := model.NewProceduresSearch(nil).
-		SetDeviceId(conf.Conf.System.DeviceId).
+		SetDeviceId(deviceID).
 		SetPage(page, pageSize)
 
 	if mode == constvar.TaskModeUnStarted {
@@ -49,7 +49,7 @@
 			SetOrder("start_time asc")
 	} else if mode == constvar.TaskModeCurrent {
 		search.SetStatus(model.ProcedureStatusProcessing).
-			SetOrder("status desc, start_time asc")
+			SetOrder("start_time asc")
 	} else if mode == constvar.TaskModeLastFinished {
 		search.SetStatus(model.ProcedureStatusFinished).SetOrder("updated_at desc")
 		if len(channels) > 0 {
@@ -101,13 +101,14 @@
 		workers = append(workers, procedure.ProceduresInfo.Workers...)
 	}
 	taskResp = &response.TaskResponse{
-		Tasks:   taskList,
-		Workers: slf.WorkerDistinct(workers),
+		Tasks:     taskList,
+		TaskCount: count,
+		Workers:   slf.WorkerDistinct(workers),
 	}
 	return taskResp, ecode.OK
 }
-func (slf TaskService) NewTaskCount() (count int64) {
-	count, _ = model.NewProceduresSearch(nil).SetDeviceId(conf.Conf.System.DeviceId).SetStatus(model.ProcedureStatusWaitProcess).Count()
+func (slf TaskService) NewTaskCount(deviceId string) (count int64) {
+	count, _ = model.NewProceduresSearch(nil).SetDeviceId(deviceId).SetStatus(model.ProcedureStatusWaitProcess).Count()
 	return count
 }
 
@@ -123,7 +124,7 @@
 	return NoRepeatedWorkers
 }
 
-func (slf TaskService) GetProcedureById(id int) (*model.Procedures, int) {
+func (slf TaskService) GetProcedureById(id uint) (*model.Procedures, int) {
 	procedure, err := model.NewProceduresSearch(nil).SetId(id).First()
 	if err != nil {
 		return nil, ecode.DBErr
@@ -131,23 +132,39 @@
 	return procedure, ecode.OK
 }
 
-func (slf TaskService) UpdateProcedureStatusAndChannel(db *gorm.DB, id int, status model.ProcedureStatus, channel int32) error {
-	if status == model.ProcedureStatusFinished {
-		ProgressCacheUnset(channel)
+func (slf TaskService) GetProcedureAndWorkOrder(procedureId uint) (procedure *model.Procedures, workOrder *model.Order, err error) {
+	procedure, err = model.NewProceduresSearch(nil).SetId(procedureId).First()
+	if err != nil {
+		return nil, nil, err
 	}
-	return model.NewProceduresSearch(db).SetId(id).UpdateByMap(map[string]interface{}{
-		"status":  status,
-		"channel": channel,
-	})
+	workOrder, err = model.NewOrderSearch(nil).SetWorkOrderId(procedure.WorkOrderID).First()
+	if err != nil {
+		return nil, nil, err
+	}
+	return procedure, workOrder, nil
 }
 
-func (slf TaskService) UpdateProcedureStatus(db *gorm.DB, id int, status model.ProcedureStatus, channel int32) error {
-	if status == model.ProcedureStatusFinished {
-		ProgressCacheUnset(channel)
+func (slf TaskService) UpdateProcedureStatusAndChannel(db *gorm.DB, id uint, status model.ProcedureStatus, channel int32, processModelNumber string) error {
+	if status == model.ProcedureStatusFinished || status == model.ProcedureStatusWaitProcess {
+		ProgressCacheUnset(conf.Conf.CurrentDeviceID, channel)
 	}
-	return model.NewProceduresSearch(db).SetId(id).UpdateByMap(map[string]interface{}{
-		"status": status,
-	})
+
+	upMap := map[string]interface{}{"status": status}
+	if status == model.ProcedureStatusProcessing {
+		upMap["process_model_number"] = processModelNumber
+		upMap["real_start_time"] = time.Now().Unix()
+	} else if status == model.ProcedureStatusWaitProcess {
+		upMap["process_model_number"] = ""
+	} else if status == model.ProcedureStatusFinished {
+		upMap["real_end_time"] = time.Now().Unix()
+	}
+
+	return model.NewProceduresSearch(db).SetId(id).UpdateByMap(upMap)
+}
+
+func (slf TaskService) UpdateProcessModelNumber(db *gorm.DB, id uint, processModelNumber string) error {
+	upMap := map[string]interface{}{"process_model_number": processModelNumber}
+	return model.NewProceduresSearch(db).SetId(id).UpdateByMap(upMap)
 }
 
 func (slf TaskService) UpdateOrderStatus(db *gorm.DB, id uint, status model.OrderStatus) error {
@@ -169,32 +186,11 @@
 		logx.Errorf("TaskStart GetProcessModel  err:%v", err)
 		return nil, err
 	}
+	if err == gorm.ErrRecordNotFound {
+		return nil, err
+	}
 	if err == nil {
 		return data, nil
-	}
-
-	if err == gorm.ErrRecordNotFound { //濡傛灉鏁版嵁搴撴病鏈変粠浜戠鑾峰彇
-		caller := nsq.NewCaller(fmt.Sprintf(constvar.NsqTopicProcessParamsRequest, conf.Conf.NsqConf.NodeId), fmt.Sprintf(constvar.NsqTopicProcessParamsResponse, conf.Conf.NsqConf.NodeId))
-		var result common.ResponseProcessParams
-		err = caller.Call(common.RequestProcessParams{
-			WorkOrder: procedure.WorkOrderID,
-			OrderId:   procedure.OrderID,
-			Product:   order.ProductName,
-			Procedure: procedure.ProceduresInfo.ProcedureName,
-			Device:    procedure.ProceduresInfo.DeviceName,
-			DeviceId:  conf.Conf.System.DeviceId,
-		}, &result, time.Second*3)
-		if err != nil {
-			logx.Errorf("TaskStart GetProcessModel error:%v", err.Error())
-			return
-		}
-		if result.ParamsMap == nil {
-			logx.Errorf("TaskStart GetProcessModel response miss process params:%v", result)
-			return
-		}
-		processModel = new(model.ProcessModel)
-		err = structx.AssignTo(result, &processModel)
-		return processModel, err
 	}
 	return
 }
@@ -207,3 +203,107 @@
 		SetStartTimeMin(nowTs).
 		SetStatus(model.OrderStatusWaitProcess).First()
 }
+
+// SaveTaskStatusSync 淇濆瓨浠诲姟鍚屾淇℃伅
+func (slf TaskService) SaveTaskStatusSync(db *gorm.DB, record *model.TaskStatusSync) (err error) {
+	if record.IsProcessing {
+		record.StartTs = time.Now().Unix()
+	}
+	if record.IsFinish {
+		record.FinishTs = time.Now().Unix()
+		startedRecord, _ := model.NewTaskStatusSyncSearch(db).SetProductProcedureId(record.ProductProcedureID).SetWorkOrderId(record.WorkOrderId).First()
+		record.StartTs = startedRecord.StartTs
+	}
+	return model.NewTaskStatusSyncSearch(db).Create(record)
+}
+
+// GetTaskStatusSync 鑾峰彇寰呭悓姝ヤ换鍔′俊鎭�
+func (slf TaskService) GetTaskStatusSync(limit int) ([]*model.TaskStatusSync, error) {
+	return model.NewTaskStatusSyncSearch(nil).SetOrder("id asc").SetPage(1, limit).FindNotTotal()
+}
+
+// RemoveTaskStatusSync 鍒犻櫎鍚屾瀹岀殑璁板綍
+func (slf TaskService) RemoveTaskStatusSync(ids []uint) (err error) {
+	return model.NewTaskStatusSyncSearch(nil).SetIDs(ids).Delete()
+}
+
+// GetTask2 鑾峰彇浠诲姟鍒楄〃2
+func (slf TaskService) GetTask2(deviceID string, offset, limit int, channels []int32, queryType request.QueryType) (taskResp *response.TaskResponse, err error) {
+	var taskList []*response.TaskData
+	var count int64
+	var workers []*common.ProcedureWorker
+
+	taskResp = &response.TaskResponse{
+		Tasks:     taskList,
+		TaskCount: count,
+		Workers:   workers,
+	}
+
+	var (
+		orders       []*model.Order
+		procedures   []*model.Procedures
+		workOrderIds []string
+	)
+	search := model.NewProceduresSearch(nil).
+		SetDeviceId(deviceID).
+		SetOffset(offset, limit).SetChannels(channels)
+	todayStartTs := timex.TodayStartTs()
+	switch queryType {
+	case request.QueryTypeUnFinish:
+		search.SetStatusNot(model.ProcedureStatusFinished).SetOrder("status desc, start_time asc")
+	case request.QueryTypeToday:
+		search.SetStatusNot(model.ProcedureStatusFinished).SetStartTimeMin(todayStartTs).SetEndTimeMax(todayStartTs + 86400).SetOrder("status desc, start_time asc")
+	case request.QueryTypeFinished:
+		search.SetStatus(model.ProcedureStatusFinished).SetOrder("start_time desc")
+	}
+
+	procedures, total, err := search.Find()
+	if err != nil {
+		return taskResp, err
+	}
+
+	if err != nil {
+		return taskResp, err
+	}
+	if len(procedures) == 0 {
+		return taskResp, nil
+	}
+
+	for _, procedure := range procedures {
+		workOrderIds = append(workOrderIds, procedure.WorkOrderID)
+	}
+	orders, err = model.NewOrderSearch(nil).SetWorkOrderIds(workOrderIds).FindNotTotal() //鐢卞伐搴忓弽鎺ㄥ伐鍗�
+	if err != nil {
+		return taskResp, err
+	}
+
+	orderMap := make(map[string]*model.Order, len(workOrderIds))
+	for _, order := range orders {
+		orderMap[order.WorkOrderID] = order
+	}
+	for _, procedure := range procedures {
+		taskData := response.TaskData{
+			Order:                 orderMap[procedure.WorkOrderID],
+			Procedure:             procedure,
+			AllProcedures:         procedure.ProceduresInfo.AllProcedureNames,
+			CurrentProcedureIndex: 0,
+			Channel:               procedure.Channel,
+		}
+		index := 0
+		for _, name := range procedure.ProceduresInfo.AllProcedureNames {
+			if name == procedure.ProceduresInfo.ProcedureName {
+				break
+			}
+			index++
+		}
+		taskData.CurrentProcedureIndex = index
+		taskList = append(taskList, &taskData)
+		workers = append(workers, procedure.ProceduresInfo.Workers...)
+	}
+	taskResp = &response.TaskResponse{
+		Tasks:     taskList,
+		TaskCount: total,
+		Workers:   slf.WorkerDistinct(workers),
+	}
+	return taskResp, err
+}

--
Gitblit v1.8.0