优化连接处理逻辑,增加超时设置;重构命令读取与错误处理;新增公共工具函数以简化错误处理

This commit is contained in:
2025-05-26 16:37:54 +08:00
parent 8c928a8321
commit c2dcae7af5
7 changed files with 205 additions and 154 deletions

View File

@@ -73,8 +73,8 @@ func ListenData(ctx context.Context) error {
}
}
func processDataConn(ctx context.Context, client net.Conn) error {
var reader = bufio.NewReader(client)
func processDataConn(ctx context.Context, edge net.Conn) error {
var reader = bufio.NewReader(edge)
// 接收连接结果
var buf = make([]byte, 17)
@@ -133,7 +133,7 @@ func processDataConn(ctx context.Context, client net.Conn) error {
// 复制用户数据到节点
var waitUser = make(chan error)
go func() {
_, err := io.Copy(client, teeUser)
_, err := io.Copy(edge, teeUser)
switch {
case errors.Is(err, net.ErrClosed):
slog.Debug("用户连接意外关闭")