From 8324f872ef3a4d0c978a9b1d062800c6a1701c12 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期五, 01 十二月 2023 09:58:17 +0800 Subject: [PATCH] fix --- api/v1/task.go | 201 +++++++++++++++++++++++++++++++++++++------------ 1 files changed, 150 insertions(+), 51 deletions(-) diff --git a/api/v1/task.go b/api/v1/task.go index 8cdb4e5..6729282 100644 --- a/api/v1/task.go +++ b/api/v1/task.go @@ -10,7 +10,6 @@ "apsClient/pkg/ecode" "apsClient/pkg/logx" "apsClient/service" - "apsClient/service/plc_address" "errors" "github.com/gin-gonic/gin" "github.com/jinzhu/gorm" @@ -63,8 +62,7 @@ ctx.FailWithMsg(ecode.NeedConfirmedErr, err.Error()) return } - - taskCount := service.NewTaskService().NewTaskCount() + taskCount := service.NewTaskService().NewTaskCount(conf.Conf.CurrentDeviceID) params.Page = 1 if params.PageSize <= 0 { if params.TaskMode == constvar.TaskModeUnStarted { @@ -79,7 +77,7 @@ taskMode = params.TaskMode } - taskResponse, code := service.NewTaskService().GetTask(params.Page, params.PageSize, taskMode, nil) //鍙栬繘琛屼腑鐨勬垨鏈紑濮嬬殑 + taskResponse, code := service.NewTaskService().GetTask(conf.Conf.CurrentDeviceID, params.Page, params.PageSize, taskMode, nil) //鍙栬繘琛屼腑鐨勬垨鏈紑濮嬬殑 if code != ecode.OK { ctx.Fail(code) return @@ -94,7 +92,7 @@ if existsChannel[int32(i)] { continue } - taskResponseTemp, code := service.NewTaskService().GetTask(params.Page, 1, constvar.TaskModeLastFinished, []int32{int32(i)}) //鍙栦笂涓�涓畬鎴愮殑 + taskResponseTemp, code := service.NewTaskService().GetTask(conf.Conf.CurrentDeviceID, params.Page, 1, constvar.TaskModeLastFinished, []int32{int32(i)}) //鍙栦笂涓�涓畬鎴愮殑 if code != ecode.OK { ctx.Fail(code) return @@ -110,7 +108,10 @@ 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 { + if task.Procedure.Status != model.ProcedureStatusWaitProcess { + continue + } + if _, ok := flagMap[task.Channel]; !ok && !service.TaskFlagGet(conf.Conf.CurrentDeviceID, task.Channel) && task.Procedure.StartTime <= nowTs { task.CanStarted = true flagMap[task.Channel] = struct{}{} } @@ -155,14 +156,21 @@ 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 { + } + + if err != nil || processModel == nil || processModel.ParamsMap == nil { + device, err := service.GetCurrentDevice() + if err != nil { + ctx.FailWithMsg(ecode.UnknownErr, err.Error()) + return + } + if device.NeedSetProcessParams { ctx.FailWithMsg(ecode.ParamsErr, "鏈幏鍙栧埌宸ヨ壓鍙傛暟锛岃鍦ㄥ伐鑹烘ā鍨嬪簱涓笂浼狅紒") + return + } else { + ctx.OkWithDetailed(response.ProcessParamsResponse{}) return } } @@ -215,16 +223,18 @@ } err := model.WithTransaction(func(db *gorm.DB) error { - err := service.NewTaskService().UpdateProcedureStatus(db, id, model.ProcedureStatusFinished, procedure.Channel) + err := service.NewTaskService().UpdateProcedureStatusAndChannel(db, id, model.ProcedureStatusFinished, procedure.Channel, procedure.ProcessModelNumber) if err != nil { return err } record := model.TaskStatusSync{ - WorkOrderId: procedure.WorkOrderID, - ProcedureID: procedure.ProcedureID, - DeviceId: procedure.DeviceID, - IsProcessing: false, - IsFinish: true, + WorkOrderId: procedure.WorkOrderID, + ProcedureID: procedure.ProcedureID, + DeviceId: procedure.DeviceID, + ProductProcedureID: procedure.ProductProcedureID, + IsProcessing: false, + IsFinish: true, + FinishedQuantity: procedure.FinishedQuantity, } return service.NewTaskService().SaveTaskStatusSync(db, &record) }) @@ -235,8 +245,8 @@ return } - service.TaskFlagUnset(procedure.Channel) - + service.TaskFlagUnset(conf.Conf.CurrentDeviceID, procedure.Channel) + service.TaskEndTimeSet(conf.Conf.CurrentDeviceID, procedure.Channel, time.Now().Unix()) //璁剧疆宸ュ簭杩愯缁撴潫鏃堕棿 ctx.Ok() } @@ -263,6 +273,10 @@ ctx.Fail(code) return } + if procedure.DeviceID != conf.Conf.CurrentDeviceID { + ctx.FailWithMsg(ecode.ParamsErr, "璇ヤ换鍔′笉灞炰簬褰撳墠璁惧") + return + } order, err := taskService.GetOrderByWorkOrderId(procedure.WorkOrderID) if err != nil { ctx.Fail(ecode.UnknownErr) @@ -273,30 +287,40 @@ ctx.Ok() return } - processModel, err := taskService.GetProcessParams(procedure, order) - if err != nil || processModel == nil || processModel.ParamsMap == nil { - ctx.FailWithMsg(ecode.ParamsErr, "鏈幏鍙栧埌宸ヨ壓鍙傛暟锛岃鍦ㄥ伐鑹烘ā鍨嬪簱涓笂浼狅紒") - return - } + + var processModelNumber string plcConfig, code := service.NewDevicePlcService().GetDevicePlc() - if code != ecode.OK || plcConfig.ID == 0 { - ctx.FailWithMsg(ecode.NeedConfirmedErr, "璇峰厛閰嶇疆PLC") - return + if code == ecode.OK && plcConfig.ID != 0 && len(plcConfig.Details) > 0 { + plcConfig.MaxTryTimes = 2 + device, err := service.GetCurrentDevice() + if err != nil { + ctx.FailWithMsg(ecode.DBErr, err.Error()) + return + } + if device.NeedSetProcessParams { + processModel, err := taskService.GetProcessParams(procedure, order) + if err != nil || processModel == nil || processModel.ParamsMap == nil { + ctx.FailWithMsg(ecode.ParamsErr, "鏈幏鍙栧埌宸ヨ壓鍙傛暟锛岃鍦ㄥ伐鑹烘ā鍨嬪簱涓笂浼狅紒") + return + } + err = SendParams(processModel.ParamsMap, plcConfig) + if err != nil { + ctx.FailWithMsg(ecode.NeedConfirmedErr, "PLC璇锋眰澶辫触锛岃妫�鏌LC閰嶇疆锛�") + return + } + processModelNumber = processModel.Number + } + + plcConfig.CurrentTryTimes = 0 + err = service.PlcWrite(plcConfig, constvar.PlcStartAddressTypeTotalNumber, procedure.Channel, order.Amount.IntPart()) + if err != nil { + ctx.FailWithMsg(ecode.NeedConfirmedErr, "PLC璇锋眰澶辫触锛岃妫�鏌LC閰嶇疆锛�") + 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) + err = taskService.UpdateProcedureStatusAndChannel(db, params.ProcedureId, model.ProcedureStatusProcessing, procedure.Channel, processModelNumber) if err != nil { return err } @@ -304,16 +328,13 @@ 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, + WorkOrderId: procedure.WorkOrderID, + ProcedureID: procedure.ProcedureID, + DeviceId: procedure.DeviceID, + ProductProcedureID: procedure.ProductProcedureID, + IsProcessing: true, + IsFinish: false, } return service.NewTaskService().SaveTaskStatusSync(db, &record) }) @@ -323,7 +344,9 @@ return } - service.TaskFlagSet(procedure.Channel) + service.TaskFlagSet(conf.Conf.CurrentDeviceID, procedure.Channel, int(procedure.ID)) + service.TaskStartTimeSet(conf.Conf.CurrentDeviceID, procedure.Channel, time.Now().Unix()) //璁剧疆宸ュ簭杩愯寮�濮嬫椂闂� + service.TaskEndTimeSet(conf.Conf.CurrentDeviceID, procedure.Channel, 0) //璁剧疆宸ュ簭杩愯缁撴潫鏃堕棿 ctx.Ok() } @@ -391,8 +414,12 @@ logx.Info("----------------寮�濮嬩笅鍙戝伐鑹哄弬鏁�-----------------") } var failedNumbers int + processModelAddress, err := service.GetProcessModelAddress(conf.Conf.CurrentDeviceID) + if err != nil { + return err + } for k, v := range paramsMap { - address, ok := plc_address.Get(k) + address, ok := processModelAddress[k] if !ok { logx.Errorf("miss param address, k:%v, v:%v", k, v) continue @@ -414,3 +441,75 @@ logx.Info("----------------涓嬪彂宸ヨ壓鍙傛暟瀹屾瘯-----------------") return nil } + +// TaskListByChannel +// @Tags Task +// @Summary 鑾峰彇浠诲姟鍒楄〃2 +// @Produce application/json +// @Param object query request.TaskListByChannel true "鏌ヨ鍙傛暟" +// @Success 200 {object} contextx.Response{data=map[int32]response.TaskResponse} "鎴愬姛" +// @Router /v1/task/list [get] +func (slf *TaskApi) TaskListByChannel(c *gin.Context) { + var params request.TaskListByChannel + ctx, ok := contextx.NewContext(c, ¶ms) + if !ok { + return + } + + channelAmount, err := service.NewDevicePlcService().GetDeviceChannelAmount() + if err != nil { + ctx.FailWithMsg(ecode.NeedConfirmedErr, err.Error()) + return + } + params.DeviceID = conf.Conf.CurrentDeviceID + + dataMap := make(map[int32]*response.TaskResponse, channelAmount) + if params.Channel != nil { + taskResponse, err := getTaskResponseByChannel(params, *params.Channel) + if err != nil { + ctx.FailWithMsg(ecode.DBErr, err.Error()) + return + } + dataMap[*params.Channel] = taskResponse + ctx.OkWithDetailed(dataMap) + return + } + + //涓嶄紶channel鍙栨墍鏈塩hannel鐨� + var wg sync.WaitGroup + var mu sync.Mutex + for i := 0; i < channelAmount; i++ { + wg.Add(1) + go func(channel int32) { + defer wg.Done() + taskResponse, err := getTaskResponseByChannel(params, channel) + if err != nil { + ctx.FailWithMsg(ecode.DBErr, err.Error()) + return + } + mu.Lock() + defer mu.Unlock() + dataMap[channel] = taskResponse + }(int32(i)) + } + wg.Wait() + ctx.OkWithDetailed(dataMap) +} + +func getTaskResponseByChannel(params request.TaskListByChannel, channel int32) (taskResponse *response.TaskResponse, err error) { + taskResponse, err = service.NewTaskService().GetTask2(params.DeviceID, params.Offset, params.Limit, []int32{channel}, params.Type) //鍙栬繘琛屼腑鐨勬垨鏈紑濮嬬殑 + if err != nil { + return + } + + nowTs := time.Now().Unix() + flagMap := make(map[int32]struct{}, 0) + for _, task := range taskResponse.Tasks { + if _, ok := flagMap[task.Channel]; !ok && !service.TaskFlagGet(conf.Conf.CurrentDeviceID, task.Channel) && task.Procedure.StartTime <= nowTs { + task.CanStarted = true + flagMap[task.Channel] = struct{}{} + } + } + taskResponse.Prompt = conf.Conf.Prompt + return +} -- Gitblit v1.8.0