From cf96f6bc1fb484fde16bd5ed6d49534c96958022 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期一, 25 九月 2023 15:43:12 +0800
Subject: [PATCH] fix

---
 api/v1/task.go |  161 ++++++++++++++++++++++++++++++++++++++++++-----------
 1 files changed, 126 insertions(+), 35 deletions(-)

diff --git a/api/v1/task.go b/api/v1/task.go
index 74287d5..480a58f 100644
--- a/api/v1/task.go
+++ b/api/v1/task.go
@@ -5,19 +5,21 @@
 	"apsClient/constvar"
 	"apsClient/model"
 	"apsClient/model/common"
+	"apsClient/model/request"
 	"apsClient/model/response"
 	"apsClient/nsq"
 	"apsClient/pkg/contextx"
 	"apsClient/pkg/convertx"
 	"apsClient/pkg/ecode"
 	"apsClient/pkg/logx"
-	"apsClient/pkg/plc"
+	"apsClient/pkg/safe"
 	"apsClient/service"
 	"apsClient/service/plc_address"
 	"errors"
 	"fmt"
 	"github.com/gin-gonic/gin"
 	"gorm.io/gorm"
+	"sort"
 	"time"
 )
 
@@ -49,20 +51,94 @@
 // @Tags      Task
 // @Summary   鑾峰彇浠诲姟
 // @Produce   application/json
-// @Success   200   {object}  contextx.Response{data=response.TaskData}  "鎴愬姛"
+// @Param     object  query    request.TaskList true  "鏌ヨ鍙傛暟"
+// @Success   200   {object}  contextx.Response{data=response.TaskResponse}  "鎴愬姛"
 // @Router    /v1/task/get [get]
 func (slf *TaskApi) TaskGet(c *gin.Context) {
-	ctx, ok := contextx.NewContext(c, nil)
+	var params request.TaskList
+	ctx, ok := contextx.NewContext(c, &params)
 	if !ok {
 		return
 	}
 
-	taskData, code := service.NewTaskService().GetTask()
+	channelAmount, err := service.NewDevicePlcService().GetDeviceChannelAmount()
+	if err != nil {
+		ctx.FailWithMsg(ecode.NeedConfirmedErr, err.Error())
+		return
+	}
+
+	params.Page = 1
+
+	if params.PageSize <= 0 {
+		params.PageSize = channelAmount
+	}
+
+	taskResponse, code := service.NewTaskService().GetTask(params.Page, params.PageSize, service.TaskModeCurrent, nil) //鍙栬繘琛屼腑鐨勬垨鏈紑濮嬬殑
 	if code != ecode.OK {
 		ctx.Fail(code)
 		return
 	}
-	ctx.OkWithDetailed(taskData)
+
+	if len(taskResponse.Tasks) < channelAmount {
+		var existsChannel = make(map[int32]bool, channelAmount)
+		for _, task := range taskResponse.Tasks {
+			existsChannel[task.Channel] = true
+		}
+		for i := 0; i < channelAmount; i++ {
+			if existsChannel[int32(i)] {
+				continue
+			}
+			taskResponseTemp, code := service.NewTaskService().GetTask(params.Page, 1, service.TaskModeLastFinished, []int32{int32(i)}) //鍙栦笂涓�涓畬鎴愮殑
+			if code != ecode.OK {
+				ctx.Fail(code)
+				return
+			}
+			taskResponse.Tasks = append(taskResponse.Tasks, taskResponseTemp.Tasks...)
+		}
+	}
+
+	sort.Slice(taskResponse.Tasks, func(i, j int) bool {
+		return taskResponse.Tasks[i].Channel < taskResponse.Tasks[i].Channel
+	})
+
+	for _, task := range taskResponse.Tasks {
+		if task.Procedure.Status == model.ProcedureStatusWaitProcess {
+			task.CanStarted = true
+		}
+	}
+
+	taskResponse.Prompt = conf.Conf.Prompt
+	taskResponse.ChannelAmount = channelAmount
+
+	ctx.OkWithDetailed(taskResponse)
+}
+
+// TaskGetUnStarted
+// @Tags      Task
+// @Summary   鑾峰彇鏈紑濮嬬殑浠诲姟
+// @Produce   application/json
+// @Param     object  query    request.TaskList true  "鏌ヨ鍙傛暟"
+// @Success   200   {object}  contextx.Response{data=response.TaskData}  "鎴愬姛"
+// @Router    /v1/task/get/unStarted [get]
+func (slf *TaskApi) TaskGetUnStarted(c *gin.Context) {
+	var params request.TaskList
+	ctx, ok := contextx.NewContext(c, &params)
+	if !ok {
+		return
+	}
+	if params.Page <= 0 {
+		params.Page = 1
+	}
+	if params.PageSize <= 0 {
+		params.PageSize = 100
+	}
+
+	taskResponse, code := service.NewTaskService().GetTask(params.Page, params.PageSize, service.TaskModeUnStarted, nil) //鏃堕棿鍒颁簡鏈紑濮嬬殑
+	if code != ecode.OK {
+		ctx.Fail(code)
+		return
+	}
+	ctx.OkWithDetailed(taskResponse)
 }
 
 // GetProcessParams
@@ -97,7 +173,7 @@
 
 	processModel, err := service.NewTaskService().GetProcessParams(procedure, order)
 	if err != nil || processModel == nil || processModel.ParamsMap == nil {
-		ctx.Fail(ecode.UnknownErr)
+		ctx.FailWithMsg(ecode.ParamsErr, "璇峰厛閰嶇疆宸ヨ壓鍙傛暟")
 		return
 	}
 
@@ -108,6 +184,16 @@
 			Value: v,
 		})
 	}
+
+	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,
@@ -138,7 +224,7 @@
 		ctx.Fail(code)
 		return
 	}
-	err := service.NewTaskService().UpdateProcedureStatus(nil, id, model.ProcedureStatusFinished)
+	err := service.NewTaskService().UpdateProcedureStatus(nil, id, model.ProcedureStatusFinished, procedure.Channel)
 	if err != nil {
 		logx.Errorf("UpdateProcedureStatus err: %v", err.Error())
 		ctx.Fail(ecode.UnknownErr)
@@ -166,22 +252,17 @@
 // @Tags      Task
 // @Summary   涓嬪彂宸ヨ壓鍙傛暟锛堝紑濮嬩换鍔★級
 // @Produce   application/json
-// @Param     id  path    int true  "宸ュ簭id"
+// @Param     object  body    request.SendProcessParams true  "鏌ヨ鍙傛暟"
 // @Success   200   {object}  contextx.Response{service.GetProcessModel}  "鎴愬姛"
-// @Router    /v1/task/sendProcessParams/{id} [post]
+// @Router    /v1/task/sendProcessParams [post]
 func (slf *TaskApi) TaskStart(c *gin.Context) {
-	ctx, ok := contextx.NewContext(c, nil)
+	var params request.SendProcessParams
+	ctx, ok := contextx.NewContext(c, &params)
 	if !ok {
 		return
 	}
-	idx := c.Param("id")
-	if idx == "" {
-		ctx.Fail(ecode.ParamsErr)
-		return
-	}
-	id := convertx.Atoi(idx)
 	taskService := service.NewTaskService()
-	procedure, code := taskService.GetProcedureById(id)
+	procedure, code := taskService.GetProcedureById(params.ProcedureId)
 	if code != ecode.OK {
 		ctx.Fail(code)
 		return
@@ -204,15 +285,16 @@
 	}
 
 	err = model.WithTransaction(func(db *gorm.DB) error {
-		err = taskService.UpdateProcedureStatus(db, id, model.ProcedureStatusProcessing)
+		err = taskService.UpdateProcedureStatusAndChannel(db, params.ProcedureId, model.ProcedureStatusProcessing, params.Channel)
 		if err != nil {
 			return err
 		}
+		procedure.Channel = params.Channel
 		err = taskService.UpdateOrderStatus(db, order.ID, model.OrderStatusProcessing)
 		if err != nil {
 			return err
 		}
-		return service.NewProgressService().AddProgress(db, procedure, order)
+		return service.NewProgressService().Add(db, procedure, order)
 	})
 	if err != nil {
 		logx.Errorf("SendProcessParams update order and procedure status error:%v", err.Error())
@@ -228,6 +310,18 @@
 	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, params.Channel)
+			if err != nil {
+				return err
+			}
+			procedure.Channel = params.Channel
+			err = taskService.UpdateOrderStatus(db, order.ID, model.OrderStatusWaitProcess)
+			if err != nil {
+				return err
+			}
+			return nil
+		})
 		ctx.FailWithMsg(ecode.NeedConfirmedErr, "绯熺硶锛屽伐鑹轰笅鍙戝け璐ャ��")
 		return
 	}
@@ -236,7 +330,7 @@
 		return
 	}
 	plcConfig.CurrentTryTimes = 0
-	err = service.PlcWrite(plcConfig, constvar.PlcStartAddressTypeTotalNumber, order.Amount.IntPart())
+	err = service.PlcWrite(plcConfig, constvar.PlcStartAddressTypeTotalNumber, params.Channel, order.Amount.IntPart())
 	if err != nil {
 		ctx.FailWithMsg(ecode.NeedConfirmedErr, "绯熺硶锛屽伐鑹轰笅鍙戝け璐ャ��")
 		return
@@ -266,27 +360,24 @@
 	if plcConfig.CurrentTryTimes > plcConfig.MaxTryTimes {
 		return plcConfig.CurrentErr
 	}
-	conn, err := plc.GetModbusConnection(fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port))
-	if err != nil {
-		return errors.New(fmt.Sprintf("杩炴帴plc澶辫触: %v", err.Error()))
-	}
 	if plcConfig.CurrentTryTimes == 0 {
 		logx.Info("----------------寮�濮嬩笅鍙戝伐鑹哄弬鏁�-----------------")
 	}
 	var failedNumbers int
 	for k, v := range paramsMap {
-		if address, ok := plc_address.Get(k); ok {
-			result, err := plc.WriteHoldingRegister(conn, address, v)
-			if err != nil {
-				plcConfig.CurrentErr = err
-				failedNumbers++
-				logx.Errorf("plc write err:%v, address: %v, key: %v value: %v", err.Error(), address, k, v)
-			} else {
-				delete(paramsMap, k)
-				logx.Infof("plc write ok: key: %v, value: %v, result: %v", k, v, result)
-			}
-		} else {
+		address, ok := plc_address.Get(k)
+		if !ok {
 			logx.Errorf("miss param address, k:%v, v:%v", k, v)
+			continue
+		}
+		err := service.PlcWriteDirect(plcConfig, address, v)
+		if err != nil {
+			plcConfig.CurrentErr = err
+			failedNumbers++
+			logx.Errorf("plc write err:%v, address: %v, key: %v value: %v", err.Error(), address, k, v)
+		} else {
+			delete(paramsMap, k)
+			logx.Infof("plc write ok: key: %v, value: %v", k, v)
 		}
 	}
 	if failedNumbers >= 1 { //鍐欏叆plc澶辫触, 閲嶈瘯

--
Gitblit v1.8.0