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 | 519 +++++++++++++++++++++++++++++++++++++++++++++++---------- 1 files changed, 426 insertions(+), 93 deletions(-) diff --git a/api/v1/task.go b/api/v1/task.go index 1278e5b..6729282 100644 --- a/api/v1/task.go +++ b/api/v1/task.go @@ -1,49 +1,137 @@ 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" + "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, ¶ms) + if !ok { + return + } + + channelAmount, err := service.NewDevicePlcService().GetDeviceChannelAmount() + if err != nil { + ctx.FailWithMsg(ecode.NeedConfirmedErr, err.Error()) + return + } + taskCount := service.NewTaskService().NewTaskCount(conf.Conf.CurrentDeviceID) + 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(conf.Conf.CurrentDeviceID, 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(conf.Conf.CurrentDeviceID, 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 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{}{} + } + } + + 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 +141,53 @@ 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) + } else { + processModel, err = service.NewTaskService().GetProcessParams(procedure, order) } - resp, err := service.ProcessModel{}.GetProcessModel(params) - - if err != nil { - logx.Errorf("TaskStart Notice GetProcessModel error: %v", err.Error()) - ctx.FailWithMsg(ecode.UnknownErr, "鏈幏鍙栧埌宸ヨ壓鍙傛暟") - return + 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 + } } - 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 +207,309 @@ 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().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, + ProductProcedureID: procedure.ProductProcedureID, + IsProcessing: false, + IsFinish: true, + FinishedQuantity: procedure.FinishedQuantity, + } + 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, - } - - 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("----------------涓嬪彂宸ヨ壓鍙傛暟瀹屾瘯-----------------") + service.TaskFlagUnset(conf.Conf.CurrentDeviceID, procedure.Channel) + service.TaskEndTimeSet(conf.Conf.CurrentDeviceID, procedure.Channel, time.Now().Unix()) //璁剧疆宸ュ簭杩愯缁撴潫鏃堕棿 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, ¶ms) + if !ok { + return + } + taskService := service.NewTaskService() + procedure, code := taskService.GetProcedureById(params.ProcedureId) + if code != ecode.OK { + 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) + return + } + + if procedure.Status == model.ProcedureStatusProcessing { + ctx.Ok() + return + } + + var processModelNumber string + plcConfig, code := service.NewDevicePlcService().GetDevicePlc() + 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 + } + } + + err = model.WithTransaction(func(db *gorm.DB) error { + err = taskService.UpdateProcedureStatusAndChannel(db, params.ProcedureId, model.ProcedureStatusProcessing, procedure.Channel, processModelNumber) + if err != nil { + return err + } + err = taskService.UpdateOrderStatus(db, order.ID, model.OrderStatusProcessing) + if err != nil { + return err + } + record := model.TaskStatusSync{ + WorkOrderId: procedure.WorkOrderID, + ProcedureID: procedure.ProcedureID, + DeviceId: procedure.DeviceID, + ProductProcedureID: procedure.ProductProcedureID, + 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(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() +} + +// 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, ¶ms) + 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.UpdateProcessModelNumber(nil, params.ProcedureId, 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 + processModelAddress, err := service.GetProcessModelAddress(conf.Conf.CurrentDeviceID) + if err != nil { + return err + } + for k, v := range paramsMap { + address, ok := processModelAddress[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 +} + +// 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