fix

zhangqian

2023-12-01 8324f872ef3a4d0c978a9b1d062800c6a1701c12

 service/plc.go

@@ -4,73 +4,25 @@
   "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, position int) (val interface{}, err error) {
// 串口不支持并行读写，所以需要加个锁，防止timeout和资源不可用报错
var lock sync.Mutex

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

   for _, pc := range plcConfig.Details {
      if pc.FieldName == fieldType && pc.Position == position {
         startAddress = pc.StartAddress
         valueType = pc.Type
         dataLength = pc.Length
      }
   }

   if plcConfig.Method == constvar.PlcMethodModbusTCP {
      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 nil, err
      }

      rawData, err := plc.ReadHoldingRegister(conn, startAddress, dataLength)
      if err != nil {
         logx.Errorf("PlcRead 获取plc数据失败: %v", err.Error())
         return nil, err
      }
      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")
   } 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", startAddress)
      return plccom.ReadPLC(plccom.DeviceTypeMitsubishi, ipAddr, label)
   }
   return nil, errors.New("interface type not support")
}

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

   if plcConfig.CurrentTryTimes > plcConfig.MaxTryTimes {
@@ -79,54 +31,35 @@
   plcConfig.CurrentTryTimes++

   for _, pc := range plcConfig.Details {
      if pc.FieldName == fieldType && pc.Position == position {
      if pc.FieldName == fieldType && pc.Channel == channel {
         startAddress = pc.StartAddress
      }
   }

   if plcConfig.Method == constvar.PlcMethodModbusTCP {
      ipAddr = fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port)
      //conn, err := plc.GetModbusConnection(ipAddr)
      //if err != nil {
      //   logx.Errorf("plc write failed, 连接plc失败: %v", err.Error())
      //   plcConfig.CurrentErr = err
      //   return PlcWrite(plcConfig, fieldType, position, value)
      //}
      //
      //result, err := plc.WriteHoldingRegister(conn, startAddress, value)
      err = modbusx.Write(ipAddr, uint16(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, position, value)
      }
      logx.Infof("plc write ok, address: %v, value: %v", 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", startAddress)
      return plccom.WritePLC(plccom.DeviceTypeMitsubishi, ipAddr, label, cast.ToInt(value))
   }
   return
   return PlcWriteDirect(plcConfig, startAddress, value)
}

func PlcWriteDirect(plcConfig *model.DevicePlc, address int, value interface{}) (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)
      //conn, err := plc.GetModbusConnection(ipAddr)
      //if err != nil {
      //   logx.Errorf("plc write failed, 连接plc失败: %v", err.Error())
      //   plcConfig.CurrentErr = err
      //   return PlcWriteDirect(plcConfig, address, value)
      //}
      //result, err := plc.WriteHoldingRegister(conn, address, value)
      err = WriteByModbusTCP(ipAddr, address, value)

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

      if err != nil {
         logx.Errorf("plc write failed, address: %v, value: %v, err: %v", address, value, err.Error())
         return err
@@ -144,40 +77,172 @@
}

func PlcReadDirect(plcConfig *model.DevicePlc, address int, dataLength int, valueType constvar.PlcStartAddressValueType) (val interface{}, err error) {
   lock.Lock()
   defer lock.Unlock()
   var (
      ipAddr string
   )
   if plcConfig.Method == constvar.PlcMethodModbusTCP {
      ipAddr = fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port)
      //conn, err := plc.GetModbusConnection(ipAddr)
      //if err != nil {
      //   plcConfig.CurrentErr = err
      //   return PlcReadDirect(plcConfig, address, dataLength, valueType)
      //}
      //value, err := plc.ReadHoldingRegister(conn, address, dataLength)
      value, err := modbusx.Read(ipAddr, uint16(address), uint16(dataLength))
      if err != nil {
         return nil, err
   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)
         if err != nil {
            return nil, err
         }
      } else {
         value, err = ReadByModbusRTU(plcConfig, address, dataLength)
         if err != nil {
            return nil, err
         }
      }

      switch valueType {
      case constvar.PlcStartAddressValueTypeString:
         return string(value), nil
      case constvar.PlcStartAddressValueTypeInt:
      case constvar.PlcStartAddressValueTypeInt32:
         if len(value) == 2 {
            return int(binary.BigEndian.Uint16(value)), nil
            val = binary.BigEndian.Uint16(value)
         } else if len(value) == 4 {
            val = binary.BigEndian.Uint32(value)
         } else {
            logx.Errorf("plc read get an unknown int value: %v, address:%v", value, address)
            return nil, errors.New(fmt.Sprintf("unknown int value：%v", value))
         }
      default:
         if len(value) == 2 {
            val = binary.BigEndian.Uint16(value)
         } else if len(value) == 4 {
            low16Val := uint32(binary.BigEndian.Uint16(value[:2]))
            high16 := uint32(binary.BigEndian.Uint16(value[2:])) << 16
            val = low16Val + high16
         } else {
            logx.Errorf("plc read get an unknown int value: %v, address:%v", value, address)
            return nil, errors.New(fmt.Sprintf("unknown int value：%v", value))
         }
      }
      logx.Infof("plc read ok, address: %v, result: %v", address, value)
      logx.Infof("plc read ok, address: %v, result: %v, dataLength: %v, value: %v", address, val, dataLength, value)
      return val, nil
   } 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)
      return plccom.ReadPLC(plccom.DeviceTypeMitsubishi, ipAddr, label, dataLength)
   }
   return
}

func ReadByModbusTCP(ipAddr string, address, length int) ([]byte, error) {
   if conf.Conf.PLC.Package == constvar.PlcPackageApache {
      return apacheplc4x.ReadHoldingRegister(ipAddr, address, length)
   } 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) (err error) {
   if conf.Conf.PLC.Package == constvar.PlcPackageApache {
      _, err = apacheplc4x.WriteHoldingRegister(ipAddr, address, value)
      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))
   }
}

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,
   }
   if conf.Conf.PLC.Package == constvar.PlcPackageApache {
      return apacheplc4x.ReadHoldingRegisterByRTU(rtuConfig, address, length)
   } else {
      return modbusx.ReadByRTU(rtuConfig, uint16(address), uint16(length))
   }
}

func WriteByModbusRTU(plcConfig *model.DevicePlc, address int, value any) (err error) {
   rtuConfig := &common.RTUConfig{
      BaudRate:   plcConfig.BaudRate,
      SerialName: plcConfig.SerialName,
      DataBit:    plcConfig.DataBit,
      StopBit:    plcConfig.StopBit,
      Parity:     plcConfig.Parity,
   }
   if conf.Conf.PLC.Package == constvar.PlcPackageApache {
      _, err = apacheplc4x.WriteHoldingRegisterByRTU(rtuConfig, address, value)
      return err
   } else {
      return modbusx.WriteByRTU(rtuConfig, uint16(address), cast.ToInt(value))
   }
}

func PlcIsConnect() bool {
   return IsConnect()
}

func dealErr(err error) {
   if err != nil {
      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
}