From 620d904921c76b6b29d1b69787da7a82121d4c31 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期四, 26 十月 2023 21:15:35 +0800
Subject: [PATCH] 读写plc支持modbusRTU

---
 service/plc.go |   99 +++++++++++++++++++++++++++++++------------------
 1 files changed, 63 insertions(+), 36 deletions(-)

diff --git a/service/plc.go b/service/plc.go
index 183cb00..68258b8 100644
--- a/service/plc.go
+++ b/service/plc.go
@@ -4,6 +4,7 @@
 	"apsClient/conf"
 	"apsClient/constvar"
 	"apsClient/model"
+	"apsClient/model/common"
 	"apsClient/pkg/logx"
 	"apsClient/pkg/plc"
 	"apsClient/pkg/plc/apacheplc4x"
@@ -19,7 +20,6 @@
 func PlcWrite(plcConfig *model.DevicePlc, fieldType constvar.PlcStartAddressType, channel int32, value interface{}) (err error) {
 	var (
 		startAddress int
-		ipAddr       string
 	)
 
 	if plcConfig.CurrentTryTimes > plcConfig.MaxTryTimes {
@@ -32,26 +32,7 @@
 			startAddress = pc.StartAddress
 		}
 	}
-
-	if plcConfig.Method == constvar.PlcMethodModbusTCP {
-		ipAddr = fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port)
-		err = WriteHoldingRegister(ipAddr, 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, channel, 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) {
@@ -61,7 +42,16 @@
 	defer dealErr(err)
 	if plcConfig.Method == constvar.PlcMethodModbusTCP {
 		ipAddr = fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port)
-		err = WriteHoldingRegister(ipAddr, address, value)
+		err = WriteByModbusTCP(ipAddr, 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())
@@ -84,12 +74,21 @@
 		ipAddr string
 	)
 	defer dealErr(err)
-	if plcConfig.Method == constvar.PlcMethodModbusTCP {
-		ipAddr = fmt.Sprintf("%s:%v", plcConfig.Address, plcConfig.Port)
-		value, err := ReadHoldingRegister(ipAddr, address, dataLength)
-		if err != nil {
-			return nil, 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
@@ -118,7 +117,7 @@
 	return
 }
 
-func ReadHoldingRegister(ipAddr string, address, length int) ([]byte, error) {
+func ReadByModbusTCP(ipAddr string, address, length int) ([]byte, error) {
 	if conf.Conf.PLC.Package == constvar.PlcPackageApache {
 		newLength := length / 2
 		if newLength == 0 {
@@ -140,7 +139,7 @@
 	}
 }
 
-func WriteHoldingRegister(ipAddr string, address int, value any) (err error) {
+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
@@ -156,15 +155,43 @@
 	}
 }
 
+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 {
+		newLength := length / 2
+		if newLength == 0 {
+			newLength = 1
+		}
+		return apacheplc4x.ReadHoldingRegisterByRTU(rtuConfig, address, newLength)
+	} 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()
-	//if conf.Conf.PLC.Package == constvar.PlcPackageApache {
-	//	return apacheplc4x.IsConnect()
-	//} else if conf.Conf.PLC.Package == constvar.PlcPackageApacheLongConnection {
-	//	return false
-	//} else {
-	//	return modbusx.IsConnect()
-	//}
 }
 
 func dealErr(err error) {

--
Gitblit v1.8.0