From bf2b61519fd0d79ddb19f0469749fbbe1d6c4ad8 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期四, 19 十月 2023 17:23:50 +0800
Subject: [PATCH] 为保证正确获取集群状态, 添加程序启动时读取system-service运行状态

---
 api/v1/task.go |  115 +++++++++++++++++++++++----------------------------------
 1 files changed, 46 insertions(+), 69 deletions(-)

diff --git a/api/v1/task.go b/api/v1/task.go
index ce5b49d..c2b864c 100644
--- a/api/v1/task.go
+++ b/api/v1/task.go
@@ -4,18 +4,14 @@
 	"apsClient/conf"
 	"apsClient/constvar"
 	"apsClient/model"
-	"apsClient/model/common"
 	"apsClient/model/request"
 	"apsClient/model/response"
-	"apsClient/nsq"
 	"apsClient/pkg/contextx"
 	"apsClient/pkg/ecode"
 	"apsClient/pkg/logx"
-	"apsClient/pkg/safe"
 	"apsClient/service"
 	"apsClient/service/plc_address"
 	"errors"
-	"fmt"
 	"github.com/gin-gonic/gin"
 	"github.com/jinzhu/gorm"
 	"github.com/spf13/cast"
@@ -179,15 +175,6 @@
 		})
 	}
 
-	safe.Go(func() {
-		caller := nsq.NewCaller(fmt.Sprintf(constvar.NsqTopicGetPlcAddress, conf.Conf.NsqConf.NodeId), fmt.Sprintf(constvar.NsqTopicSendPlcAddress, conf.Conf.NsqConf.NodeId))
-		var addressResult common.ResponsePlcAddress
-		err := caller.Call(common.RequestPlcAddress{DeviceId: conf.Conf.System.DeviceId}, &addressResult, time.Second*3)
-		if err != nil {
-			logx.Infof("get plc address err: %v", err.Error())
-		}
-	})
-
 	resp := response.ProcessParamsResponse{
 		Number: processModel.Number,
 		Params: processParamsArr,
@@ -226,25 +213,26 @@
 		ctx.Fail(ecode.ParamsErr)
 		return
 	}
-	err := service.NewTaskService().UpdateProcedureStatus(nil, id, model.ProcedureStatusFinished, procedure.Channel)
+
+	err := model.WithTransaction(func(db *gorm.DB) error {
+		err := service.NewTaskService().UpdateProcedureStatus(db, id, model.ProcedureStatusFinished, procedure.Channel)
+		if err != nil {
+			return err
+		}
+		record := model.TaskStatusSync{
+			WorkOrderId:  procedure.WorkOrderID,
+			ProcedureID:  procedure.ProcedureID,
+			DeviceId:     procedure.DeviceID,
+			IsProcessing: false,
+			IsFinish:     true,
+		}
+		return service.NewTaskService().SaveTaskStatusSync(db, &record)
+	})
+
 	if err != nil {
-		logx.Errorf("UpdateProcedureStatus err: %v", err.Error())
+		logx.Errorf("TaskFinish UpdateProcedureStatus err: %v", err.Error())
 		ctx.Fail(ecode.UnknownErr)
 		return
-	}
-
-	msg := &common.MsgTaskStatusUpdate{
-		WorkOrderId:  procedure.WorkOrderID,
-		ProcedureID:  procedure.ProceduresInfo.ProcedureID,
-		DeviceId:     procedure.ProceduresInfo.DeviceID,
-		IsProcessing: false,
-		IsFinish:     true,
-	}
-
-	caller := nsq.NewCaller(fmt.Sprintf(constvar.NsqTopicTaskProcedureStatusUpdate, conf.Conf.NsqConf.NodeId), "")
-	err = caller.Send(msg)
-	if err != nil {
-		logx.Errorf("send task status update msg error:%v", err.Error())
 	}
 
 	service.TaskFlagUnset(procedure.Channel)
@@ -290,7 +278,23 @@
 		ctx.FailWithMsg(ecode.ParamsErr, "鏈幏鍙栧埌宸ヨ壓鍙傛暟锛岃鍦ㄥ伐鑹烘ā鍨嬪簱涓笂浼狅紒")
 		return
 	}
-
+	plcConfig, code := service.NewDevicePlcService().GetDevicePlc()
+	if code != ecode.OK || plcConfig.ID == 0 {
+		ctx.FailWithMsg(ecode.NeedConfirmedErr, "璇峰厛閰嶇疆PLC")
+		return
+	}
+	plcConfig.MaxTryTimes = 2
+	err = SendParams(processModel.ParamsMap, plcConfig)
+	if err != nil {
+		ctx.FailWithMsg(ecode.NeedConfirmedErr, "PLC璇锋眰澶辫触锛岃妫�鏌LC閰嶇疆锛�")
+		return
+	}
+	plcConfig.CurrentTryTimes = 0
+	err = service.PlcWrite(plcConfig, constvar.PlcStartAddressTypeTotalNumber, procedure.Channel, order.Amount.IntPart())
+	if err != nil {
+		ctx.FailWithMsg(ecode.NeedConfirmedErr, "PLC璇锋眰澶辫触锛岃妫�鏌LC閰嶇疆锛�")
+		return
+	}
 	err = model.WithTransaction(func(db *gorm.DB) error {
 		err = taskService.UpdateProcedureStatusAndChannel(db, params.ProcedureId, model.ProcedureStatusProcessing, procedure.Channel, processModel.Number)
 		if err != nil {
@@ -300,52 +304,25 @@
 		if err != nil {
 			return err
 		}
-		return service.NewProgressService().Add(db, procedure, order)
+		err = service.NewProgressService().Add(db, procedure, order)
+		if err != nil {
+			return err
+		}
+		record := model.TaskStatusSync{
+			WorkOrderId:  procedure.WorkOrderID,
+			ProcedureID:  procedure.ProcedureID,
+			DeviceId:     procedure.DeviceID,
+			IsProcessing: true,
+			IsFinish:     false,
+		}
+		return service.NewTaskService().SaveTaskStatusSync(db, &record)
 	})
 	if err != nil {
 		logx.Errorf("SendProcessParams update order and procedure status error:%v", err.Error())
 		ctx.FailWithMsg(ecode.DBErr, "鏇存敼宸ュ崟鐘舵�佸け璐�")
 		return
 	}
-	plcConfig, code := service.NewDevicePlcService().GetDevicePlc()
-	if code != ecode.OK || plcConfig.ID == 0 {
-		ctx.FailWithMsg(ecode.NeedConfirmedErr, "璇峰厛閰嶇疆PLC")
-		return
-	}
-	plcConfig.MaxTryTimes = 2
-	err = SendParams(processModel.ParamsMap, plcConfig)
-	if err != nil {
-		logx.Errorf("SendProcessParams: %v", err.Error())
-		err = model.WithTransaction(func(db *gorm.DB) error {
-			err = taskService.UpdateProcedureStatusAndChannel(db, params.ProcedureId, model.ProcedureStatusWaitProcess, procedure.Channel, "")
-			if err != nil {
-				return err
-			}
-			return nil
-		})
-		ctx.FailWithMsg(ecode.NeedConfirmedErr, "PLC璇锋眰澶辫触锛岃妫�鏌LC閰嶇疆锛�")
-		return
-	}
-	plcConfig.CurrentTryTimes = 0
-	err = service.PlcWrite(plcConfig, constvar.PlcStartAddressTypeTotalNumber, procedure.Channel, order.Amount.IntPart())
-	if err != nil {
-		ctx.FailWithMsg(ecode.NeedConfirmedErr, "PLC璇锋眰澶辫触锛岃妫�鏌LC閰嶇疆")
-		return
-	}
 
-	msg := &common.MsgTaskStatusUpdate{
-		WorkOrderId:  procedure.WorkOrderID,
-		ProcedureID:  procedure.ProceduresInfo.ProcedureID,
-		DeviceId:     procedure.ProceduresInfo.DeviceID,
-		IsProcessing: true,
-		IsFinish:     false,
-	}
-
-	caller := nsq.NewCaller(fmt.Sprintf(constvar.NsqTopicTaskProcedureStatusUpdate, conf.Conf.NsqConf.NodeId), "")
-	err = caller.Send(msg)
-	if err != nil {
-		logx.Errorf("send task status update msg error:%v", err.Error())
-	}
 	service.TaskFlagSet(procedure.Channel)
 	ctx.Ok()
 }

--
Gitblit v1.8.0