优化接入网关，解决协程泄漏的问题

server/global/gateway/tunnel.go

@@ -1,59 +1,55 @@
 package gateway
 
 import (
-	"context"
 	"fmt"
 	"io"
 	"net"
 
 	"next-terminal/server/log"
+
+	"golang.org/x/crypto/ssh"
 )
 
 type Tunnel struct {
-	ID                string // 唯一标识
-	LocalHost         string // 本地监听地址
-	LocalPort         int    // 本地端口
-	RemoteHost        string // 远程连接地址
-	RemotePort        int    // 远程端口
-	Gateway           *Gateway
-	ctx               context.Context
-	cancel            context.CancelFunc
+	id                string // 唯一标识
+	localHost         string // 本地监听地址
+	localPort         int    // 本地端口
+	remoteHost        string // 远程连接地址
+	remotePort        int    // 远程端口
 	listener          net.Listener
 	localConnections  []net.Conn
 	remoteConnections []net.Conn
 }
 
-func (r *Tunnel) Open() {
-	localAddr := fmt.Sprintf("%s:%d", r.LocalHost, r.LocalPort)
+func (r *Tunnel) Open(sshClient *ssh.Client) {
+	localAddr := fmt.Sprintf("%s:%d", r.localHost, r.localPort)
 
-	go func() {
-		<-r.ctx.Done()
-		_ = r.listener.Close()
-		log.Debugf("SSH 隧道 %v 关闭", localAddr)
-	}()
 	for {
-		log.Debugf("等待客户端访问 %v", localAddr)
+		log.Debugf("隧道 %v 等待客户端访问 %v", r.id, localAddr)
 		localConn, err := r.listener.Accept()
 		if err != nil {
-			log.Debugf("接受连接失败 %v, 退出循环", err.Error())
+			log.Debugf("隧道 %v 接受连接失败 %v, 退出循环", r.id, err.Error())
+			log.Debug("-------------------------------------------------")
 			return
 		}
 		r.localConnections = append(r.localConnections, localConn)
 
-		log.Debugf("客户端 %v 连接至 %v", localConn.RemoteAddr().String(), localAddr)
-		remoteAddr := fmt.Sprintf("%s:%d", r.RemoteHost, r.RemotePort)
-		log.Debugf("连接远程主机 %v ...", remoteAddr)
-		remoteConn, err := r.Gateway.SshClient.Dial("tcp", remoteAddr)
+		log.Debugf("隧道 %v 新增本地连接 %v", r.id, localConn.RemoteAddr().String())
+		remoteAddr := fmt.Sprintf("%s:%d", r.remoteHost, r.remotePort)
+		log.Debugf("隧道 %v 连接远程地址 %v ...", r.id, remoteAddr)
+		remoteConn, err := sshClient.Dial("tcp", remoteAddr)
 		if err != nil {
-			log.Debugf("连接远程主机 %v 失败", remoteAddr)
+			log.Debugf("隧道 %v 连接远程地址 %v, 退出循环", r.id, err.Error())
+			log.Debug("-------------------------------------------------")
 			return
 		}
 		r.remoteConnections = append(r.remoteConnections, remoteConn)
 
-		log.Debugf("连接远程主机 %v 成功", remoteAddr)
+		log.Debugf("隧道 %v 连接远程主机成功", r.id)
 		go copyConn(localConn, remoteConn)
 		go copyConn(remoteConn, localConn)
-		log.Debugf("转发数据 [%v]->[%v]", localAddr, remoteAddr)
+		log.Debugf("隧道 %v 开始转发数据 [%v]->[%v]", r.id, localAddr, remoteAddr)
+		log.Debug("~~~~~~~~~~~~~~~~~~~~分割线~~~~~~~~~~~~~~~~~~~~~~~~")
 	}
 }
 
@@ -66,7 +62,8 @@ func (r *Tunnel) Close() {
 		_ = r.remoteConnections[i].Close()
 	}
 	r.remoteConnections = nil
-	r.cancel()
+	_ = r.listener.Close()
+	log.Debugf("隧道 %v 监听器关闭", r.id)
 }
 
 func copyConn(writer, reader net.Conn) {