From d91a802e7aa2ad4075ed803b8ddc7536a91a0ef0 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 27 十月 2023 20:33:06 +0800
Subject: [PATCH] update

---
 pkg/plc/modbusx/connection_manager.go |   46 ++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 44 insertions(+), 2 deletions(-)

diff --git a/pkg/plc/modbusx/connection_manager.go b/pkg/plc/modbusx/connection_manager.go
index ae0b377..4389fe5 100644
--- a/pkg/plc/modbusx/connection_manager.go
+++ b/pkg/plc/modbusx/connection_manager.go
@@ -2,6 +2,8 @@
 
 import (
 	"apsClient/conf"
+	"apsClient/model/common"
+	"apsClient/pkg/logx"
 	"github.com/goburrow/modbus"
 	"sync"
 	"time"
@@ -49,7 +51,7 @@
 	if conn, ok := connectionManager.GetConnection(ipAddr); ok {
 		return conn
 	}
-	conn := newGetModbusConnection(ipAddr)
+	conn := newModbusConnection(ipAddr)
 	connectionManager.AddConnection(ipAddr, conn)
 	return conn
 }
@@ -65,9 +67,49 @@
 	}
 }
 
-func newGetModbusConnection(ipAddr string) modbus.Client {
+func newModbusConnection(ipAddr string) modbus.Client {
 	handler = modbus.NewTCPClientHandler(ipAddr)
 	handler.Timeout = 10 * time.Second
 	handler.SlaveId = byte(conf.Conf.PLC.SlaveId)
 	return modbus.NewClient(handler)
 }
+
+var rtuHandler *modbus.RTUClientHandler
+
+func getModbusRTUConnection(c *common.RTUConfig) modbus.Client {
+	if conn, ok := connectionManager.GetConnection(c.SerialName); ok {
+		return conn
+	}
+	conn := newModbusRTUConnection(c)
+	connectionManager.AddConnection(c.SerialName, conn)
+	return conn
+}
+
+func unsetModbusRTUConnection(serialName string) {
+	_, ok := connectionManager.GetConnection(serialName)
+	if !ok {
+		return
+	}
+	connectionManager.RemoveConnection(serialName)
+	if handler != nil {
+		handler.Close()
+	}
+}
+
+func newModbusRTUConnection(c *common.RTUConfig) modbus.Client {
+	// Modbus RTU/ASCII
+	rtuHandler = modbus.NewRTUClientHandler(c.SerialName)
+	rtuHandler.BaudRate = c.BaudRate
+	rtuHandler.DataBits = c.DataBit
+	rtuHandler.Parity = c.Parity.String()
+	rtuHandler.StopBits = c.StopBit
+	rtuHandler.SlaveId = 1
+	rtuHandler.Timeout = 5 * time.Second
+
+	err := rtuHandler.Connect()
+	if err != nil {
+		logx.Errorf("getModbusRTUConnection  Connect err:%v, config: %+v", err, c)
+		return nil
+	}
+	return modbus.NewClient(rtuHandler)
+}

--
Gitblit v1.8.0