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 {
@@ -200,7 +203,6 @@
}

func dealErr(err error) {
   logx.Infof("deal err: %v", err)
   if err != nil {
      FailureRemainingOpportunitiesDecr() //减少失败剩余机会
   } else {
@@ -208,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
}