内存优化

Author: csznet

forward/forward.go

@@ -31,7 +31,7 @@ type LargeConnectionStats struct {
 // 复用缓冲区
 var bufPool = sync.Pool{
 	New: func() interface{} {
-		return make([]byte, 8192)
+		return make([]byte, 4096)
 	},
 }
 
@@ -183,8 +183,10 @@ func (cs *ConnectionStats) handleUDPConnection(localConn *net.UDPConn, remoteAdd
 			cs.TotalBytesLock.Unlock()
 
 			// 处理消息的边界和错误情况
-			go cs.forwardUDPMessage(localConn, remoteAddr, buf[:n])
-			bufPool.Put(buf[:n])
+			go func() {
+				cs.forwardUDPMessage(localConn, remoteAddr, buf[:n])
+				bufPool.Put(&buf)
+			}()
 		}
 	}
 }
@@ -199,11 +201,12 @@ func (cs *ConnectionStats) forwardUDPMessage(localConn *net.UDPConn, remoteAddr
 	if err != nil {
 		fmt.Println("写入目标时发生错误:", err)
 	}
+
 }
 
 func (cs *ConnectionStats) copyBytes(dst, src net.Conn) {
 	buf := bufPool.Get().([]byte)
-	defer bufPool.Put(buf)
+	defer bufPool.Put(&buf)
 	for {
 		n, err := src.Read(buf)
 		if n > 0 {
@@ -231,7 +234,7 @@ func (cs *ConnectionStats) copyBytes(dst, src net.Conn) {
 
 // 定时打印和处理流量变化
 func (cs *ConnectionStats) printStats(ctx context.Context) {
-	ticker := time.NewTicker(5 * time.Second)
+	ticker := time.NewTicker(10 * time.Second)
 	defer ticker.Stop() // 在函数结束时停止定时器
 	for {
 		select {
@@ -286,8 +289,9 @@ func (cs *ConnectionStats) printStats(ctx context.Context) {
 func closeTCPConnections(stats *ConnectionStats) {
 	stats.TotalBytesLock.Lock()
 	defer stats.TotalBytesLock.Unlock()
-	for _, conn := range stats.TCPConnections {
+	for i, conn := range stats.TCPConnections {
 		conn.Close()
+		stats.TCPConnections[i] = nil
 	}
 	stats.TCPConnections = nil // 清空切片
 }

main.go

@@ -63,6 +63,6 @@ func init() {
 	flag.StringVar(&conf.WebPort, "port", "8889", "Web Port")
 	flag.StringVar(&conf.WebIP, "ip", "0.0.0.0", "Web IP")
 	flag.StringVar(&conf.WebPass, "pass", "", "Web Password")
-	flag.IntVar(&conf.TcpTimeout, "tt", 3600, "Tcp Timeout")
+	flag.IntVar(&conf.TcpTimeout, "tt", 60, "Tcp Timeout")
 	flag.Parse()
 }

readme.md

@@ -1,57 +1,59 @@
-使用golang实现的tcp udp端口转发
+使用 golang 实现的 tcp udp 端口转发
 
 目前已实现：
 
- - 规则热加载
- - web管理面板
- - 流量统计
+- 规则热加载
+- web 管理面板
+- 流量统计
 
-支持：Linux、Windows、MacOS（MacOS需要自行编译）
+支持：Linux、Windows、MacOS（MacOS 需要自行编译）
 
 **截图**
 
 ![image](https://github.com/csznet/goForward/assets/127601663/2f7840ff-9b34-4f69-a7c1-41feb35e726b)
 
-
 **使用**
 
-Linux下载
+Linux 下载
+
 ```
 sudo bash -c "$(curl -fsSL https://raw.githubusercontent.com/csznet/goForward/main/get.sh)"
 ```
+
 运行
+
 ```
 ./goForward
 ```
 
-**参数**  
+**参数**
+
+TCP 无传输超时关闭
+默认 60，单位秒
 
-TCP无传输超时关闭
-默认3600，单位秒
 ```
 ./goForward -tt 18000
 ```
 
-
-自定义web管理端口
+自定义 web 管理端口
 
 ```
 ./goForward -port 8899
 ```
 
-指定IP绑定  
+指定 IP 绑定
 
 ```
 ./goForward -ip 1.1.1.1
 ```
 
-设置web管理访问密码
+设置 web 管理访问密码
 
 ```
 ./goForward -pass 666
 ```
 
-当24H内同一IP密码试错超过3次将会ban掉
+当 24H 内同一 IP 密码试错超过 3 次将会 ban 掉
 
 ## 开机自启
 
@@ -74,22 +76,28 @@ ExecStart=/full/path/to/your/goForward -pass 666
 WantedBy=default.target
 ```
 
-其中的```/full/path/to/your/goForward```改为二进制文件地址，后面可接参数
+其中的`/full/path/to/your/goForward`改为二进制文件地址，后面可接参数
 
 **重新加载 Systemd 配置**
+
 ```
 sudo systemctl daemon-reload
 ```
 
 **启用服务**
+
 ```
 sudo systemctl enable goForward
 ```
+
 **启动服务**
+
 ```
 sudo systemctl start goForward
 ```
+
 **检查状态**
+
 ```
 sudo systemctl status goForward.service
 ```