zhangqian
2024-04-16 8b6a262a56942a1a5764526f315029f1cd410ede
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
@@ -107,10 +105,12 @@
      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 {
      if task.Procedure.Status != model.ProcedureStatusWaitProcess {
         continue
      }
      if _, ok := flagMap[task.Channel]; !ok && !service.TaskFlagGet(conf.Conf.CurrentDeviceID, task.Channel) {
         task.CanStarted = true
         flagMap[task.Channel] = struct{}{}
      }
@@ -155,14 +155,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
      }
   }
@@ -220,11 +227,13 @@
         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 +244,8 @@
      return
   }
   service.TaskFlagUnset(procedure.Channel)
   service.TaskFlagUnset(procedure.DeviceID, procedure.Channel)
   service.TaskEndTimeSet(procedure.DeviceID, procedure.Channel, time.Now().Unix()) //设置工序运行结束时间
   ctx.Ok()
}
@@ -263,6 +272,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 +286,46 @@
      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请求失败,请检查PLC配置!")
            return
         }
         processModelNumber = processModel.Number
      }
      fieldType := constvar.PlcStartAddressTypeTotalNumber
      for _, pc := range plcConfig.Details { //如果有写入地址,优先往写入地址写入,没有的话往读取地址写入
         if pc.FieldName == constvar.PlcStartAddressTypeTotalNumberWrite && pc.Channel == procedure.Channel {
            fieldType = constvar.PlcStartAddressTypeTotalNumberWrite
         }
      }
      plcConfig.CurrentTryTimes = 0
      err = service.PlcWrite(plcConfig, fieldType, procedure.Channel, order.Amount.IntPart())
      if err != nil {
         ctx.FailWithMsg(ecode.NeedConfirmedErr, "PLC请求失败,请检查PLC配置!")
         return
      }
   }
   plcConfig.MaxTryTimes = 2
   err = SendParams(processModel.ParamsMap, plcConfig)
   if err != nil {
      ctx.FailWithMsg(ecode.NeedConfirmedErr, "PLC请求失败,请检查PLC配置!")
      return
   }
   plcConfig.CurrentTryTimes = 0
   err = service.PlcWrite(plcConfig, constvar.PlcStartAddressTypeTotalNumber, procedure.Channel, order.Amount.IntPart())
   if err != nil {
      ctx.FailWithMsg(ecode.NeedConfirmedErr, "PLC请求失败,请检查PLC配置!")
      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 +333,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 +349,9 @@
      return
   }
   service.TaskFlagSet(procedure.Channel)
   service.TaskFlagSet(procedure.DeviceID, procedure.Channel, int(procedure.ID))
   service.TaskStartTimeSet(procedure.DeviceID, procedure.Channel, time.Now().Unix()) //设置工序运行开始时间
   service.TaskEndTimeSet(procedure.DeviceID, procedure.Channel, 0)                   //设置工序运行结束时间
   ctx.Ok()
}
@@ -391,13 +419,19 @@
      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
      }
      err := service.PlcWriteDirect(plcConfig, address, v)
      err := service.PlcWriteDirect(plcConfig, v, &model.DevicePlcAddress{
         StartAddress: address,
      })
      if err != nil {
         plcConfig.CurrentErr = err
         failedNumbers++
@@ -414,3 +448,74 @@
   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, &params)
   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取所有channel的
   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
   }
   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.CanStarted = true
         flagMap[task.Channel] = struct{}{}
      }
   }
   taskResponse.Prompt = conf.Conf.Prompt
   return
}