更改定时任务时区设置

zhangqian

2024-04-25 2d6875c93b25d0b7336c7fa11e066d213259fe2e

 service/plc.go

@@ -1,87 +1,269 @@
package service

import (
   "apsClient/conf"
   "apsClient/constvar"
   "apsClient/model"
   "apsClient/model/common"
   "apsClient/pkg/logx"
   "apsClient/pkg/plc"
   "apsClient/pkg/plc/apacheplc4x"
   "apsClient/pkg/plc/modbusx"
   "apsClient/pkg/plccom"
   "encoding/binary"
   "errors"
   "fmt"
   "github.com/spf13/cast"
   "sync"
)

func PlcRead(plcConfig *model.DevicePlc, fieldType constvar.PlcStartAddressType) (val interface{}, err error) {
   var (
      startAddress int
      valueType    constvar.PlcStartAddressValueType
      dataLength   int
      ipAddr       string
   )
// 串口不支持并行读写，所以需要加个锁，防止timeout和资源不可用报错
var lock sync.Mutex

   for _, pc := range plcConfig.Details {
      if pc.FieldName == fieldType {
         startAddress = pc.StartAddress
         valueType = pc.Type
         dataLength = pc.Length
      }
   }
   ipAddr = fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port)

   conn, err := plc.GetModbusConnection(ipAddr)
   if err != nil {
      logx.Errorf("PlcRead 连接plc失败: %v", err.Error())
      return
   }

   rawData, err := plc.ReadHoldingRegister(conn, startAddress, dataLength)
   if err != nil {
      logx.Errorf("PlcRead 获取plc数据失败: %v", err.Error())
      return
   }
   switch valueType {
   case constvar.PlcStartAddressValueTypeString:
      return string(rawData), nil
   case constvar.PlcStartAddressValueTypeInt:
      if len(rawData) == 2 {
         return int(binary.BigEndian.Uint16(rawData)), nil
      } else {
         logx.Errorf("plc read get an unknown int value: %v, address:%v", rawData, startAddress)
         return nil, errors.New(fmt.Sprintf("unknown int value：%v", rawData))
      }
   }
   return nil, errors.New("undefined value type")
}

func PlcWrite(plcConfig *model.DevicePlc, fieldType constvar.PlcStartAddressType, value interface{}) (err error) {
   var (
      startAddress int
      ipAddr       string
   )

func PlcWrite(plcConfig *model.DevicePlc, fieldType constvar.PlcStartAddressType, channel int32, value interface{}) (err error) {
   if plcConfig.CurrentTryTimes > plcConfig.MaxTryTimes {
      return plcConfig.CurrentErr
   }
   plcConfig.CurrentTryTimes++

   var targetConfig *model.DevicePlcAddress
   for _, pc := range plcConfig.Details {
      if pc.FieldName == fieldType {
         startAddress = pc.StartAddress
      if pc.FieldName == fieldType && pc.Channel == channel {
         targetConfig = pc
      }
   }
   ipAddr = fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port)
   conn, err := plc.GetModbusConnection(ipAddr)
   return PlcWriteDirect(plcConfig, value, targetConfig)
}

func PlcWriteDirect(plcConfig *model.DevicePlc, value interface{}, c *model.DevicePlcAddress) (err error) {
   lock.Lock()
   defer lock.Unlock()
   var (
      ipAddr string
   )
   defer func() {
      dealErr(err)
   }()
   if plcConfig.Method == constvar.PlcMethodModbusTCP {
      ipAddr = fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port)
      err = WriteByModbusTCP(ipAddr, c.StartAddress, value, c.Endian, c.Length)

      if err != nil {
         logx.Errorf("plc write failed, address: %v, value: %v, err: %v", c.StartAddress, value, err.Error())
         return err
      }
      logx.Infof("plc write ok, address: %v, value: %v", c.StartAddress, value)
   } else if plcConfig.Method == constvar.PlcMethodModbusRTU {
      ipAddr = fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port)
      err = WriteByModbusRTU(plcConfig, c.StartAddress, value, c.Endian, c.Length)

      if err != nil {
         logx.Errorf("plc write failed, address: %v, value: %v, err: %v", c.StartAddress, value, err.Error())
         return err
      }
      logx.Infof("plc write ok, address: %v, value: %v", c.StartAddress, value)
   } else if plcConfig.Method == constvar.PlcMethodSerial {
      ipAddr = conf.Conf.Services.Serial
      if ipAddr == "" {
         return errors.New("conf.Conf.Services.Serial config not set yet")
      }
      label := fmt.Sprintf("D%d", c.StartAddress)
      return plccom.WritePLC(plccom.DeviceTypeMitsubishi, ipAddr, label, cast.ToInt(value))
   }
   return
}

func PlcReadDirect(plcConfig *model.DevicePlc, addressConfig *model.DevicePlcAddress) (val interface{}, err error) {
   address, dataLength, endian := addressConfig.StartAddress, addressConfig.Length, addressConfig.Endian
   lock.Lock()
   defer lock.Unlock()
   var (
      ipAddr string
   )
   defer func() {
      dealErr(err)
   }()
   if plcConfig.Method == constvar.PlcMethodModbusTCP || plcConfig.Method == constvar.PlcMethodModbusRTU {
      var value []byte
      if plcConfig.Method == constvar.PlcMethodModbusTCP {
         ipAddr = fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port)
         value, err = ReadByModbusTCP(ipAddr, address, dataLength, endian)
         if err != nil {
            return nil, err
         }
      } else {
         value, err = ReadByModbusRTU(plcConfig, address, dataLength)
         if err != nil {
            return nil, err
         }
      }
      return Convert(value, addressConfig)
   } else if plcConfig.Method == constvar.PlcMethodSerial {
      ipAddr = conf.Conf.Services.Serial
      if ipAddr == "" {
         return nil, errors.New("conf.Conf.Services.Serial config not set yet")
      }
      label := fmt.Sprintf("D%d", address)
      return plccom.ReadPLC(plccom.DeviceTypeMitsubishi, ipAddr, label, dataLength)
   }
   return
}

func ReadByModbusTCP(ipAddr string, address, length int, endian constvar.EndianType) ([]byte, error) {
   if conf.Conf.PLC.Package == constvar.PlcPackageApache {
      var intType string
      if endian == constvar.EndianTypeMix {
         intType = "UINT"
      } else {
         intType = "DINT"
      }
      return apacheplc4x.ReadHoldingRegister(ipAddr, address, length, intType)
   } else if conf.Conf.PLC.Package == constvar.PlcPackageApacheLongConnection {
      conn, err := plc.GetModbusConnection(ipAddr)
      if err != nil {
         return nil, err
      }
      return plc.ReadHoldingRegister(conn, address, length)
   } else {
      return modbusx.Read(ipAddr, uint16(address), uint16(length))
   }
}

func WriteByModbusTCP(ipAddr string, address int, value any, endian constvar.EndianType, length int) (err error) {
   if conf.Conf.PLC.Package == constvar.PlcPackageApache {
      var intType string
      if endian == constvar.EndianTypeMix {
         intType = "UINT"
         int32Value := cast.ToInt32(value)
         highVal := int32Value / (1 << 16)
         lowValue := int32Value % (1 << 16)
         _, err = apacheplc4x.WriteHoldingRegister(ipAddr, address, lowValue, intType)
         if err != nil {
            return err
         }
         _, err = apacheplc4x.WriteHoldingRegister(ipAddr, address+1, highVal, intType)
      } else {
         intType = "DINT"
         _, err = apacheplc4x.WriteHoldingRegister(ipAddr, address, value, intType)
      }
      return err
   } else if conf.Conf.PLC.Package == constvar.PlcPackageApacheLongConnection {
      conn, err := plc.GetModbusConnection(ipAddr)
      if err != nil {
         return err
      }
      _, err = plc.WriteHoldingRegister(conn, address, value)
      return err
   } else {
      return modbusx.Write(ipAddr, uint16(address), cast.ToInt(value), string(endian), length)
   }
}

func ReadByModbusRTU(plcConfig *model.DevicePlc, address, length int) ([]byte, error) {
   rtuConfig := &common.RTUConfig{
      BaudRate:   plcConfig.BaudRate,
      SerialName: plcConfig.SerialName,
      DataBit:    plcConfig.DataBit,
      StopBit:    plcConfig.StopBit,
      Parity:     plcConfig.Parity,
   }
   return modbusx.ReadByRTU(rtuConfig, uint16(address), uint16(length))
}

func WriteByModbusRTU(plcConfig *model.DevicePlc, address int, value any, endian constvar.EndianType, length int) (err error) {
   rtuConfig := &common.RTUConfig{
      BaudRate:   plcConfig.BaudRate,
      SerialName: plcConfig.SerialName,
      DataBit:    plcConfig.DataBit,
      StopBit:    plcConfig.StopBit,
      Parity:     plcConfig.Parity,
   }
   return modbusx.WriteByRTU(rtuConfig, uint16(address), cast.ToInt(value), string(endian), length)
}

func PlcIsConnect() bool {
   return IsConnect()
}

func dealErr(err error) {
   if err != nil {
      logx.Errorf("plc write failed, 连接plc失败: %v", err.Error())
      plcConfig.CurrentErr = err
      return PlcWrite(plcConfig, fieldType, value)
      FailureRemainingOpportunitiesDecr() //减少失败剩余机会
   } else {
      FailureRemainingOpportunitiesReset() //重置失败剩余机会
   }
}

var connectionStatus sync.Map

const (
   defaultFailureRemainingOpportunities = 20
)

func IsConnect() bool {
   val, ok := connectionStatus.Load(conf.Conf.CurrentDeviceID)
   if !ok {
      return false
   }
   failureRemainingOpportunities := val.(int)
   return failureRemainingOpportunities > 0
}

func FailureRemainingOpportunitiesDecr() {
   val, ok := connectionStatus.Load(conf.Conf.CurrentDeviceID)
   if !ok {
      return
   }
   failureRemainingOpportunities := val.(int)
   if failureRemainingOpportunities > 0 {
      failureRemainingOpportunities--
   }
   connectionStatus.Store(conf.Conf.CurrentDeviceID, failureRemainingOpportunities)
   return
}

func FailureRemainingOpportunitiesReset() {
   val, ok := connectionStatus.Load(conf.Conf.CurrentDeviceID)
   if !ok || val.(int) < defaultFailureRemainingOpportunities {
      connectionStatus.Store(conf.Conf.CurrentDeviceID, defaultFailureRemainingOpportunities)
      return
   }
   return
}

func Convert(rawValue []byte, c *model.DevicePlcAddress) (value interface{}, err error) {
   switch c.Type {
   case constvar.PlcStartAddressValueTypeString:
      return string(rawValue), nil
   case constvar.PlcStartAddressValueTypeInt16:
      if c.Endian == constvar.EndianTypeLittle {
         value = binary.LittleEndian.Uint16(rawValue)
      } else {
         value = binary.BigEndian.Uint16(rawValue)
      }
   default:
      if len(rawValue) == 2 {
         if c.Endian == constvar.EndianTypeLittle {
            value = binary.LittleEndian.Uint16(rawValue)
         } else {
            value = binary.BigEndian.Uint16(rawValue)
         }
      } else if len(rawValue) == 4 {
         if c.Endian == constvar.EndianTypeBig {
            value = binary.BigEndian.Uint32(rawValue)
         } else if c.Endian == constvar.EndianTypeLittle {
            value = binary.LittleEndian.Uint32(rawValue)
         } else {
            highBts := rawValue[2:]
            lowBts := rawValue[:2]
            highVal := uint32(binary.BigEndian.Uint16(highBts)) << 16
            lowVal := uint32(binary.BigEndian.Uint16(lowBts))
            value = highVal + lowVal
         }
      } else {
         logx.Errorf("plc read get an unknown int value: %v, address:%v", rawValue, c.StartAddress)
         return nil, errors.New(fmt.Sprintf("unknown value：%v", rawValue))
      }
   }

   result, err := plc.WriteHoldingRegister(conn, startAddress, value)
   if err != nil {
      logx.Errorf("plc write failed, address: %v, value: %v, err: %v", startAddress, value, err.Error())
      plcConfig.CurrentErr = err
      return PlcWrite(plcConfig, fieldType, value)
   }
   logx.Infof("plc write ok, address: %v, value: %v, result: %v", startAddress, value, result)
   logx.Infof("plc read ok, address: %v, result: %v, value: %v, c:%+v", c.StartAddress, rawValue, value, c)
   return
}