fix

zhangqian

2023-12-01 8324f872ef3a4d0c978a9b1d062800c6a1701c12

 service/plc.go

@@ -1,115 +1,248 @@
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) {
// 串口不支持并行读写，所以需要加个锁，防止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
   )

   if plcConfig.CurrentTryTimes > plcConfig.MaxTryTimes {
      return plcConfig.CurrentErr
   }
   plcConfig.CurrentTryTimes++

   for _, pc := range plcConfig.Details {
      if pc.FieldName == fieldType {
      if pc.FieldName == fieldType && pc.Channel == channel {
         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:
      return int(binary.BigEndian.Uint16(rawData)), nil
   }
   return nil, errors.New("undefined value type")
   return PlcWriteDirect(plcConfig, startAddress, value)
}

func PlcWrite(plcConfig *model.DevicePlc, fieldType constvar.PlcStartAddressType, value interface{}) (err error) {
func PlcWriteDirect(plcConfig *model.DevicePlc, address int, value interface{}) (err error) {
   lock.Lock()
   defer lock.Unlock()
   var (
      startAddress int
      ipAddr       string
      ipAddr string
   )
   defer func() {
      dealErr(err)
   }()
   if plcConfig.Method == constvar.PlcMethodModbusTCP {
      ipAddr = fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port)
      err = WriteByModbusTCP(ipAddr, address, value)

   for _, pc := range plcConfig.Details {
      if pc.FieldName == fieldType {
         startAddress = pc.StartAddress
      if err != nil {
         logx.Errorf("plc write failed, address: %v, value: %v, err: %v", address, value, err.Error())
         return err
      }
   }
   ipAddr = fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port)
      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)

   conn, err := plc.GetModbusConnection(ipAddr)
   if err != nil {
      logx.Errorf("PlcWrite 连接plc失败: %v", err.Error())
      return
      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.PlcMethodSerial {
      ipAddr = conf.Conf.Services.Serial
      if ipAddr == "" {
         return errors.New("conf.Conf.Services.Serial config not set yet")
      }
      label := fmt.Sprintf("D%d", address)
      return plccom.WritePLC(plccom.DeviceTypeMitsubishi, ipAddr, label, cast.ToInt(value))
   }

   result, err := plc.WriteHoldingRegister(conn, startAddress, value)
   if err != nil {
      logx.Infof("plc write failed, address: %v, value: %v, err: %v", startAddress, value, err.Error())
      return
   }
   logx.Infof("plc write ok, address: %v, value: %v, result: %v", startAddress, value, result)
   return
}

type CacheStore struct {
   cache map[string]interface{}
   mu    sync.Mutex
func PlcReadDirect(plcConfig *model.DevicePlc, address int, dataLength int, valueType constvar.PlcStartAddressValueType) (val interface{}, err error) {
   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)
         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.PlcStartAddressValueTypeInt32:
         if len(value) == 2 {
            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, 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, dataLength)
   }
   return
}

var defaultCacheStore *CacheStore

func init() {
   defaultCacheStore = newCacheManager()
}
func newCacheManager() *CacheStore {
   return &CacheStore{
      cache: make(map[string]interface{}),
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 (cm *CacheStore) Get(key string) (interface{}, bool) {
   cm.mu.Lock()
   defer cm.mu.Unlock()

   conn, ok := cm.cache[key]
   return conn, ok
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 (cm *CacheStore) Add(key string, value interface{}) {
   cm.mu.Lock()
   defer cm.mu.Unlock()
   cm.cache[key] = 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 PlcCacheGet(key string) (interface{}, bool) {
   return defaultCacheStore.Get(key)
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 PlcCacheSet(key string, value interface{}) {
   defaultCacheStore.Add(key, 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
}