From cab6eea89a0d82710369604ecb51f1cdb122e433 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期四, 19 十月 2023 17:26:34 +0800
Subject: [PATCH] fix

---
 api/v1/task.go |  420 ++++++++++++++++++++++++++++++++++++++++++++++-------------
 1 files changed, 327 insertions(+), 93 deletions(-)

diff --git a/api/v1/task.go b/api/v1/task.go
index 1278e5b..c2b864c 100644
--- a/api/v1/task.go
+++ b/api/v1/task.go
@@ -1,49 +1,136 @@
 package v1
 
 import (
+	"apsClient/conf"
+	"apsClient/constvar"
 	"apsClient/model"
+	"apsClient/model/request"
 	"apsClient/model/response"
-	_ "apsClient/model/response"
 	"apsClient/pkg/contextx"
-	"apsClient/pkg/convertx"
 	"apsClient/pkg/ecode"
 	"apsClient/pkg/logx"
 	"apsClient/service"
-	"fmt"
+	"apsClient/service/plc_address"
+	"errors"
 	"github.com/gin-gonic/gin"
+	"github.com/jinzhu/gorm"
+	"github.com/spf13/cast"
+	"sort"
+	"sync"
 	"time"
 )
 
 type TaskApi struct{}
 
-// TaskGet
+// TaskCountdown
 // @Tags      Task
-// @Summary   鑾峰彇浠诲姟
+// @Summary   鏂颁换鍔″�掕鏃�
 // @Produce   application/json
-// @Success   200   {object}  contextx.Response{data=response.TaskData}  "鎴愬姛"
-// @Router    /v1/task/get [get]
-func (slf *TaskApi) TaskGet(c *gin.Context) {
+// @Success   200   {object}  contextx.Response{data=response.TaskCountdown}  "鎴愬姛"
+// @Router    /v1/task/countdown [get]
+func (slf *TaskApi) TaskCountdown(c *gin.Context) {
 	ctx, ok := contextx.NewContext(c, nil)
 	if !ok {
 		return
 	}
+	var resp response.TaskCountdown
+	workOrder, err := service.NewTaskService().GetNextTask()
+	if err == nil {
+		seconds := workOrder.StartTime - time.Now().Unix()
+		resp.CountDownHour = seconds / 3600
+		resp.CountDownMinute = seconds % 3600 / 60
+		resp.ShowCountDown = true
+	}
+	ctx.OkWithDetailed(resp)
+}
 
-	taskData, code := service.NewTaskService().GetTask()
+// TaskGet
+// @Tags      Task
+// @Summary   鑾峰彇浠诲姟
+// @Produce   application/json
+// @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) {
+	var params request.TaskList
+	ctx, ok := contextx.NewContext(c, &params)
+	if !ok {
+		return
+	}
+
+	channelAmount, err := service.NewDevicePlcService().GetDeviceChannelAmount()
+	if err != nil {
+		ctx.FailWithMsg(ecode.NeedConfirmedErr, err.Error())
+		return
+	}
+
+	taskCount := service.NewTaskService().NewTaskCount()
+	params.Page = 1
+	if params.PageSize <= 0 {
+		if params.TaskMode == constvar.TaskModeUnStarted {
+			params.PageSize = int(taskCount)
+		} else {
+			params.PageSize = channelAmount
+		}
+	}
+
+	taskMode := constvar.TaskModeCurrent
+	if params.TaskMode == constvar.TaskModeUnStarted {
+		taskMode = params.TaskMode
+	}
+
+	taskResponse, code := service.NewTaskService().GetTask(params.Page, params.PageSize, taskMode, nil) //鍙栬繘琛屼腑鐨勬垨鏈紑濮嬬殑
 	if code != ecode.OK {
 		ctx.Fail(code)
 		return
 	}
-	ctx.OkWithDetailed(taskData)
+
+	if len(taskResponse.Tasks) < channelAmount && taskMode == constvar.TaskModeCurrent {
+		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, constvar.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
+	})
+
+	nowTs := time.Now().Unix()
+	flagMap := make(map[int32]struct{}, 0)
+	for _, task := range taskResponse.Tasks {
+		if _, ok := flagMap[task.Channel]; !ok && !service.TaskFlagGet(task.Channel) && task.Procedure.StartTime <= nowTs {
+			task.CanStarted = true
+			flagMap[task.Channel] = struct{}{}
+		}
+	}
+
+	taskResponse.Prompt = conf.Conf.Prompt
+	taskResponse.ChannelAmount = channelAmount
+	taskResponse.TaskCount = taskCount
+
+	ctx.OkWithDetailed(taskResponse)
 }
 
-// TaskStart
+// GetProcessParams
 // @Tags      Task
-// @Summary   浠诲姟寮�濮�
+// @Summary   浠诲姟寮�濮嬶紙鑾峰彇宸ヨ壓鍙傛暟锛�
 // @Produce   application/json
 // @Param     id  path    int true  "宸ュ簭id"
-// @Success   200   {object}  contextx.Response{data=[]response.ProcessParams}  "鎴愬姛"
+// @Success   200   {object}  contextx.Response{data=response.ProcessParamsResponse}  "鎴愬姛"
 // @Router    /v1/task/start/{id} [get]
-func (slf *TaskApi) TaskStart(c *gin.Context) {
+func (slf *TaskApi) GetProcessParams(c *gin.Context) {
 	ctx, ok := contextx.NewContext(c, nil)
 	if !ok {
 		return
@@ -53,48 +140,46 @@
 		ctx.Fail(ecode.ParamsErr)
 		return
 	}
-	id := convertx.Atoi(idx)
+	id := cast.ToUint(idx)
 	procedure, code := service.NewTaskService().GetProcedureById(id)
 	if code != ecode.OK {
 		ctx.Fail(code)
 		return
 	}
 
-	if procedure.Status != model.ProcedureStatusUnFinished {
-		ctx.FailWithMsg(ecode.ParamsErr, "璇ュ伐搴忓凡缁撴潫")
-		return
-	}
-
 	order, err := service.NewTaskService().GetOrderByWorkOrderId(procedure.WorkOrderID)
 	if err != nil {
-		ctx.Fail(ecode.UnknownErr)
+		ctx.Fail(ecode.DBErr)
 		return
 	}
-
-	params := service.GetProcessModelParams{
-		WorkOrder: procedure.WorkOrderID,
-		OrderId:   procedure.OrderID,
-		Product:   order.ProductName,
-		Procedure: procedure.ProceduresInfo.ProcedureName,
-		Device:    procedure.ProceduresInfo.DeviceName,
+	var processModel *model.ProcessModel
+	if procedure.ProcessModelNumber != "" {
+		processModel, err = service.GetProcessModelByNumber(procedure.ProcessModelNumber)
+		if err != nil || processModel == nil || processModel.ParamsMap == nil {
+			ctx.FailWithMsg(ecode.ParamsErr, "鏈幏鍙栧埌宸ヨ壓鍙傛暟锛岃鍦ㄥ伐鑹烘ā鍨嬪簱涓笂浼狅紒")
+			return
+		}
+	} else {
+		processModel, err = service.NewTaskService().GetProcessParams(procedure, order)
+		if err != nil || processModel == nil || processModel.ParamsMap == nil {
+			ctx.FailWithMsg(ecode.ParamsErr, "鏈幏鍙栧埌宸ヨ壓鍙傛暟锛岃鍦ㄥ伐鑹烘ā鍨嬪簱涓笂浼狅紒")
+			return
+		}
 	}
 
-	resp, err := service.ProcessModel{}.GetProcessModel(params)
-
-	if err != nil {
-		logx.Errorf("TaskStart Notice GetProcessModel error: %v", err.Error())
-		ctx.FailWithMsg(ecode.UnknownErr, "鏈幏鍙栧埌宸ヨ壓鍙傛暟")
-		return
-	}
-	processParamsArr := make([]*response.ProcessParams, 0, len(resp.ParamsMap))
-	for k, v := range resp.ParamsMap {
-		processParamsArr = append(processParamsArr, &response.ProcessParams{
+	processParamsArr := make([]response.ProcessParams, 0, len(processModel.ParamsMap))
+	for k, v := range processModel.ParamsMap {
+		processParamsArr = append(processParamsArr, response.ProcessParams{
 			Key:   k,
 			Value: v,
 		})
 	}
-	logx.Infof("TaskStart Notice GetProcessModel: %+v", resp)
-	ctx.OkWithDetailed(processParamsArr)
+
+	resp := response.ProcessParamsResponse{
+		Number: processModel.Number,
+		Params: processParamsArr,
+	}
+	ctx.OkWithDetailed(resp)
 }
 
 // TaskFinish
@@ -114,69 +199,218 @@
 		ctx.Fail(ecode.ParamsErr)
 		return
 	}
-	id := convertx.Atoi(idx)
-	_, code := service.NewTaskService().GetProcedureById(id)
-	if code != ecode.OK {
-		ctx.Fail(code)
-		return
-	}
-	err := service.NewTaskService().UpdateProcedureStatus(id, model.ProcedureStatusFinished)
-	if err != nil {
-		logx.Errorf("UpdateProcedureStatus err: %v", err.Error())
-		ctx.Fail(ecode.UnknownErr)
-		return
-	}
-	ctx.Ok()
-}
-
-// SendProcessParams
-// @Tags      Task
-// @Summary   涓嬪彂宸ヨ壓鍙傛暟
-// @Produce   application/json
-// @Param     id  path    int true  "宸ュ簭id"
-// @Success   200   {object}  contextx.Response{service.GetProcessModel}  "鎴愬姛"
-// @Router    /v1/task/sendProcessParams/{id} [post]
-func (slf *TaskApi) SendProcessParams(c *gin.Context) {
-	ctx, ok := contextx.NewContext(c, nil)
-	if !ok {
-		return
-	}
-	idx := c.Param("id")
-	if idx == "" {
-		ctx.Fail(ecode.ParamsErr)
-		return
-	}
-	id := convertx.Atoi(idx)
+	id := cast.ToUint(idx)
 	procedure, code := service.NewTaskService().GetProcedureById(id)
 	if code != ecode.OK {
 		ctx.Fail(code)
 		return
 	}
-	order, err := service.NewTaskService().GetOrderByWorkOrderId(procedure.WorkOrderID)
+	if procedure.Status == model.ProcedureStatusFinished {
+		ctx.Ok()
+		return
+	}
+	if procedure.Status != model.ProcedureStatusProcessing {
+		ctx.Fail(ecode.ParamsErr)
+		return
+	}
+
+	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("TaskFinish UpdateProcedureStatus err: %v", err.Error())
 		ctx.Fail(ecode.UnknownErr)
 		return
 	}
 
-	params := service.GetProcessModelParams{
-		WorkOrder: "",
-		OrderId:   procedure.OrderID,
-		Product:   order.ProductName,
-		Procedure: procedure.ProceduresInfo.ProcedureName,
-		Device:    procedure.ProceduresInfo.DeviceID,
-	}
+	service.TaskFlagUnset(procedure.Channel)
 
-	resp, err := service.ProcessModel{}.GetProcessModel(params)
-	if err != nil {
-		logx.Errorf("SendProcessModel GetProcessModel err: %v", err.Error())
-		ctx.Fail(ecode.UnknownErr)
-		return
-	}
-	fmt.Println("----------------寮�濮嬩笅鍙戝伐鑹哄弬鏁�-----------------")
-	for k, v := range resp.ParamsMap {
-		fmt.Println(fmt.Sprintf("%v : %v", k, v))
-		time.Sleep(time.Millisecond * 300)
-	}
-	fmt.Println("----------------涓嬪彂宸ヨ壓鍙傛暟瀹屾瘯-----------------")
 	ctx.Ok()
 }
+
+var mutex sync.Mutex
+
+// TaskStart
+// @Tags      Task
+// @Summary   涓嬪彂宸ヨ壓鍙傛暟锛堝紑濮嬩换鍔★級
+// @Produce   application/json
+// @Param     object  body    request.SendProcessParams true  "鏌ヨ鍙傛暟"
+// @Success   200   {object}  contextx.Response{service.GetProcessModel}  "鎴愬姛"
+// @Router    /v1/task/sendProcessParams [post]
+func (slf *TaskApi) TaskStart(c *gin.Context) {
+	mutex.Lock()
+	defer mutex.Unlock()
+	var params request.SendProcessParams
+	ctx, ok := contextx.NewContext(c, &params)
+	if !ok {
+		return
+	}
+	taskService := service.NewTaskService()
+	procedure, code := taskService.GetProcedureById(params.ProcedureId)
+	if code != ecode.OK {
+		ctx.Fail(code)
+		return
+	}
+	order, err := taskService.GetOrderByWorkOrderId(procedure.WorkOrderID)
+	if err != nil {
+		ctx.Fail(ecode.UnknownErr)
+		return
+	}
+
+	if procedure.Status == model.ProcedureStatusProcessing {
+		ctx.Ok()
+		return
+	}
+	processModel, err := taskService.GetProcessParams(procedure, order)
+	if err != nil || processModel == nil || processModel.ParamsMap == nil {
+		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 {
+			return err
+		}
+		err = taskService.UpdateOrderStatus(db, order.ID, model.OrderStatusProcessing)
+		if err != nil {
+			return err
+		}
+		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
+	}
+
+	service.TaskFlagSet(procedure.Channel)
+	ctx.Ok()
+}
+
+// UpdateProcessParams
+// @Tags      Task
+// @Summary   鏇存柊宸ヨ壓鍙傛暟锛堣繘琛屼腑鐨勪换鍔★級
+// @Produce   application/json
+// @Param     object  body    request.SendProcessParams true  "鏌ヨ鍙傛暟"
+// @Success   200   {object}  contextx.Response{}  "鎴愬姛"
+// @Router    /v1/task/updateProcessParams [post]
+func (slf *TaskApi) UpdateProcessParams(c *gin.Context) {
+	mutex.Lock()
+	defer mutex.Unlock()
+	var params request.SendProcessParams
+	ctx, ok := contextx.NewContext(c, &params)
+	if !ok {
+		return
+	}
+	taskService := service.NewTaskService()
+	procedure, code := taskService.GetProcedureById(params.ProcedureId)
+	if code != ecode.OK {
+		ctx.Fail(code)
+		return
+	}
+	order, err := taskService.GetOrderByWorkOrderId(procedure.WorkOrderID)
+	if err != nil {
+		ctx.Fail(ecode.UnknownErr)
+		return
+	}
+
+	if procedure.Status != model.ProcedureStatusProcessing { //鍙兘杩涜涓殑鍙互鏇存柊
+		ctx.FailWithMsg(ecode.ParamsErr, "鍙兘杩涜涓殑宸ュ簭鍙互鏇存柊宸ヨ壓鍙傛暟")
+		return
+	}
+	processModel, err := taskService.GetProcessParams(procedure, order)
+	if err != nil || processModel == nil || processModel.ParamsMap == nil {
+		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 {
+		logx.Errorf("update process params err: %v", err.Error())
+		ctx.FailWithMsg(ecode.NeedConfirmedErr, "PLC璇锋眰澶辫触锛岃妫�鏌LC閰嶇疆锛�")
+		return
+	}
+	_ = taskService.UpdateProcedureStatusAndChannel(nil, params.ProcedureId, model.ProcedureStatusProcessing, procedure.Channel, processModel.Number)
+	ctx.Ok()
+}
+
+func SendParams(paramsMap map[string]interface{}, plcConfig *model.DevicePlc) error {
+	if len(paramsMap) == 0 {
+		return errors.New("empty params")
+	}
+	if plcConfig.CurrentTryTimes > plcConfig.MaxTryTimes {
+		return plcConfig.CurrentErr
+	}
+	if plcConfig.CurrentTryTimes == 0 {
+		logx.Info("----------------寮�濮嬩笅鍙戝伐鑹哄弬鏁�-----------------")
+	}
+	var failedNumbers int
+	for k, v := range paramsMap {
+		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澶辫触, 閲嶈瘯
+		plcConfig.CurrentTryTimes++
+		return SendParams(paramsMap, plcConfig)
+	}
+	logx.Info("----------------涓嬪彂宸ヨ壓鍙傛暟瀹屾瘯-----------------")
+	return nil
+}

--
Gitblit v1.8.0