fix

zhangqian

2023-12-01 8324f872ef3a4d0c978a9b1d062800c6a1701c12

 service/plc.go

@@ -15,7 +15,6 @@
   "fmt"
   "github.com/spf13/cast"
   "sync"
   "sync/atomic"
)

// 串口不支持并行读写，所以需要加个锁，防止timeout和资源不可用报错
@@ -45,7 +44,9 @@
   var (
      ipAddr string
   )
   defer dealErr(err)
   defer func() {
      dealErr(err)
   }()
   if plcConfig.Method == constvar.PlcMethodModbusTCP {
      ipAddr = fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port)
      err = WriteByModbusTCP(ipAddr, address, value)
@@ -81,7 +82,9 @@
   var (
      ipAddr string
   )
   defer dealErr(err)
   defer func() {
      dealErr(err)
   }()
   if plcConfig.Method == constvar.PlcMethodModbusTCP || plcConfig.Method == constvar.PlcMethodModbusRTU {
      var value []byte
      if plcConfig.Method == constvar.PlcMethodModbusTCP {
@@ -100,7 +103,16 @@
      switch valueType {
      case constvar.PlcStartAddressValueTypeString:
         return string(value), nil
      case constvar.PlcStartAddressValueTypeInt:
      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 {
@@ -198,32 +210,39 @@
   }
}

var connectionStatus atomic.Bool

var failureRemainingOpportunities atomic.Int64
var connectionStatus sync.Map

const (
   defaultFailureRemainingOpportunities = 20
)

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

func FailureRemainingOpportunitiesDecr() {
   newValue := failureRemainingOpportunities.Add(-1)
   if newValue <= 0 {
      connectionStatus.Store(false)
   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() {
   if failureRemainingOpportunities.Load() < defaultFailureRemainingOpportunities {
      failureRemainingOpportunities.Store(defaultFailureRemainingOpportunities)
   }
   if connectionStatus.Load() == false {
      connectionStatus.Store(true)
   val, ok := connectionStatus.Load(conf.Conf.CurrentDeviceID)
   if !ok || val.(int) < defaultFailureRemainingOpportunities {
      connectionStatus.Store(conf.Conf.CurrentDeviceID, defaultFailureRemainingOpportunities)
      return
   }
   return
}