优化连接处理逻辑,增加超时设置;重构命令读取与错误处理;新增公共工具函数以简化错误处理
This commit is contained in:
@@ -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("用户连接意外关闭")
|
||||
|
||||
Reference in New Issue
Block a user