fix

zhangqian

2023-10-30 10c65da3d2af7056f48d9301e83f53f102f76e18

 crontask/cron_task.go

@@ -1,44 +1,154 @@
package crontask

import (
   "apsClient/conf"
   "apsClient/constvar"
   "apsClient/model/common"
   "apsClient/nsq"
   "apsClient/pkg/ecode"
   "apsClient/pkg/logx"
   "apsClient/service"
   "fmt"
   "github.com/go-co-op/gocron"
   "github.com/spf13/cast"
   "time"
)

func InitTask() error {
   s := gocron.NewScheduler(time.UTC)
   _, err := s.Every(2).Seconds().StartImmediately().Do(func() {
var s *gocron.Scheduler

func StartTask(isMaster bool) error {
   finishNumberTimeInterval := conf.Conf.PLC.FinishNumberTimeInterval
   totalNumberTimeInterval := conf.Conf.PLC.TotalNumberTimeInterval
   if finishNumberTimeInterval == 0 {
      finishNumberTimeInterval = 6
   }
   if totalNumberTimeInterval == 0 {
      totalNumberTimeInterval = 60
   }
   s = gocron.NewScheduler(time.UTC)
   _, err := s.Every(finishNumberTimeInterval).Seconds().Do(func() {
      plcConfig, code := service.NewDevicePlcService().GetDevicePlc()
      if code != ecode.OK {
         return
      }
      finishNumber, err := service.PlcRead(plcConfig, constvar.PlcStartAddressTypeFinishNumber)
      if err == nil {
         service.PlcCacheSet(constvar.PlcCacheKeyFinishNumber, finishNumber)
      for _, addressItem := range plcConfig.Details {
         if addressItem.FieldName == constvar.PlcStartAddressTypeFinishNumber {
            value, err := service.PlcReadDirect(plcConfig, addressItem.StartAddress, addressItem.Length, addressItem.Type)
            if err != nil {
               logx.Infof("plc read finish number err: %v", err)
               continue
            }
            finishNumber := cast.ToInt64(value)
            if finishNumber != 0 {
               service.PlcCacheSet(addressItem.Channel, constvar.PlcCacheKeyFinishNumber, finishNumber)
               _ = service.NewProgressService().UpdateProgress(addressItem.Channel, cast.ToInt64(finishNumber))
            }
            logx.Infof("plc read finish number: %v", finishNumber)
         }
      }
      logx.Infof("plc read finish number:%v, err:%v", finishNumber, err)

   })
   if err != nil {
      return err
   }

   s.Every(1).Minutes().StartImmediately().Do(func() {
   s.Every(totalNumberTimeInterval).Seconds().Do(func() {
      plcConfig, code := service.NewDevicePlcService().GetDevicePlc()
      if code != ecode.OK {
         return
      }
      totalNumber, err := service.PlcRead(plcConfig, constvar.PlcStartAddressTypeTotalNumber)
      if err == nil {
         service.PlcCacheSet(constvar.PlcCacheKeyTotalNumber, totalNumber)
      for _, addressItem := range plcConfig.Details {
         if addressItem.FieldName == constvar.PlcStartAddressTypeTotalNumber {
            value, err := service.PlcReadDirect(plcConfig, addressItem.StartAddress, addressItem.Length, addressItem.Type)
            if err != nil {
               logx.Infof("plc read total number err: %v", err)
               continue
            }
            totalNumber := cast.ToInt64(value)
            if totalNumber != 0 {
               service.PlcCacheSet(addressItem.Channel, constvar.PlcCacheKeyTotalNumber, totalNumber)
               _ = service.NewProgressService().UpdateProgress(addressItem.Channel, cast.ToInt64(totalNumber))
            }
            logx.Infof("plc read total number: %v", totalNumber)
         }
      }
      logx.Infof("plc read total number:%v, err:%v", totalNumber, err)

   })

   if isMaster {
      s.Every(60).Seconds().Do(SyncProductionProgress) //同步生产数据
      s.Every(30).Seconds().Do(SyncTaskStatus)         //同步任务状态
   }
   s.StartAsync()
   return nil
}

func SyncProductionProgress() {
   progressList, err := service.NewProgressService().GetProgressList()
   if err != nil {
      logx.Errorf("SyncProductionProgress get records err:%v", err)
      return
   }
   for _, progress := range progressList {
      caller := nsq.NewCaller(fmt.Sprintf(constvar.NsqTopicSyncTaskProgress, conf.Conf.NsqConf.NodeId), "")
      err = caller.Send(progress)
      if err != nil {
         logx.Errorf("SyncProductionProgress error:%v", err.Error())
      }
   }
}

func SyncTaskStatus() {
   records, err := service.NewTaskService().GetTaskStatusSync(100)
   if err != nil {
      logx.Errorf("SyncTaskStatus get records err:%v", err)
   }
   syncOkIds := make([]uint, 0, len(records))
   for _, record := range records {
      msg := &common.MsgTaskStatusUpdate{
         WorkOrderId:  record.WorkOrderId,
         ProcedureID:  record.ProcedureID,
         DeviceId:     record.DeviceId,
         IsProcessing: record.IsProcessing,
         IsFinish:     record.IsFinish,
      }
      caller := nsq.NewCaller(fmt.Sprintf(constvar.NsqTopicTaskProcedureStatusUpdate, conf.Conf.NsqConf.NodeId), "")
      err = caller.Send(msg)
      if err != nil {
         logx.Errorf("sync task status send msg error:%v", err.Error())
      }
      syncOkIds = append(syncOkIds, record.ID)
   }
   if len(syncOkIds) > 0 {
      err = service.NewTaskService().RemoveTaskStatusSync(syncOkIds)
      if err != nil {
         logx.Errorf("sync task status delete sync ok records error:%v", err)
      }
   }
}

func RestartTask(isMaster bool) error {
   if s != nil {
      s.Stop()
   }
   return StartTask(isMaster)
}

// Once 一次性任务
func Once(isMaster bool) {
   if !isMaster {
      return
   }
   msg := &common.MsgPullDataRequest{DataType: common.PullDataTypeProcessModel}
   caller := nsq.NewCaller(fmt.Sprintf(constvar.NsqTopicPullDataRequest, conf.Conf.NsqConf.NodeId), constvar.NsqTopicPullDataResponse)
   err := caller.Send(msg)
   if err != nil {
      logx.Errorf("send pull data msg error:%v, msg:%+v", err.Error(), msg)
   }
   msg = &common.MsgPullDataRequest{DataType: common.PullDataTypeDevice}
   caller = nsq.NewCaller(fmt.Sprintf(constvar.NsqTopicPullDataRequest, conf.Conf.NsqConf.NodeId), constvar.NsqTopicPullDataResponse)
   err = caller.Send(msg)
   if err != nil {
      logx.Errorf("send pull data msg error:%v, msg:%+v", err.Error(), msg)
   }
}