From 8324f872ef3a4d0c978a9b1d062800c6a1701c12 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 01 十二月 2023 09:58:17 +0800
Subject: [PATCH] fix

---
 service/plc.go |  256 ++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 213 insertions(+), 43 deletions(-)

diff --git a/service/plc.go b/service/plc.go
index 1cef995..c577998 100644
--- a/service/plc.go
+++ b/service/plc.go
@@ -1,78 +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) {
+// 涓插彛涓嶆敮鎸佸苟琛岃鍐欙紝鎵�浠ラ渶瑕佸姞涓攣锛岄槻姝imeout鍜岃祫婧愪笉鍙敤鎶ラ敊
+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.NewModbusConnection(ipAddr)
-	if err != nil {
-		logx.Errorf("GetProductProgress 杩炴帴plc澶辫触: %v", err.Error())
-		return
-	}
-	defer conn.Close()
-
-	rawData, err := plc.ReadHoldingRegister(conn, startAddress, dataLength)
-	if err != nil {
-		logx.Errorf("GetProductProgress 鑾峰彇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.NewModbusConnection(ipAddr)
+		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))
+	}
+	return
+}
+
+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
+}
+
+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 {
-		logx.Errorf("GetProductProgress 杩炴帴plc澶辫触: %v", err.Error())
+		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
 	}
-	defer conn.Close()
+	failureRemainingOpportunities := val.(int)
+	if failureRemainingOpportunities > 0 {
+		failureRemainingOpportunities--
+	}
+	connectionStatus.Store(conf.Conf.CurrentDeviceID, failureRemainingOpportunities)
+	return
+}
 
-	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())
+func FailureRemainingOpportunitiesReset() {
+	val, ok := connectionStatus.Load(conf.Conf.CurrentDeviceID)
+	if !ok || val.(int) < defaultFailureRemainingOpportunities {
+		connectionStatus.Store(conf.Conf.CurrentDeviceID, defaultFailureRemainingOpportunities)
 		return
 	}
-	logx.Infof("plc write ok, address: %v, value: %v, result: %v", startAddress, value, result)
 	return
 }

--
Gitblit v1.8.0