修改节点心跳发送失败逻辑,现在会直接退出;完善数据通道连接超时的错误处理
This commit is contained in:
@@ -60,10 +60,7 @@ func ListenData(ctx context.Context) error {
|
||||
app.DataConnWg.Add(1)
|
||||
go func() {
|
||||
defer app.DataConnWg.Done()
|
||||
defer func() {
|
||||
utils.Close(conn)
|
||||
slog.Debug("关闭数据通道连接")
|
||||
}()
|
||||
defer utils.Close(conn)
|
||||
err := processDataConn(ctx, conn)
|
||||
if err != nil {
|
||||
slog.Error("处理数据通道连接失败", "err", err)
|
||||
@@ -89,12 +86,13 @@ func processDataConn(ctx context.Context, edge net.Conn) error {
|
||||
// 加载用户连接
|
||||
user, ok := app.UserConnMap.LoadAndDelete(tag)
|
||||
if !ok {
|
||||
return fmt.Errorf("用户连接已关闭,tag:%s", tag)
|
||||
if status == 1 {
|
||||
return fmt.Errorf("用户连接已关闭,tag:%s", tag)
|
||||
} else {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
defer func() {
|
||||
utils.Close(user)
|
||||
slog.Debug("关闭用户连接")
|
||||
}()
|
||||
defer utils.Close(user)
|
||||
|
||||
// 检查状态
|
||||
if status != 1 {
|
||||
@@ -119,13 +117,8 @@ func processDataConn(ctx context.Context, edge net.Conn) error {
|
||||
var waitEdge = make(chan error)
|
||||
go func() {
|
||||
_, err := io.Copy(user, reader)
|
||||
switch {
|
||||
case errors.Is(err, net.ErrClosed):
|
||||
slog.Warn("节点连接意外关闭")
|
||||
case err != nil:
|
||||
if ok, err := utils.WarpConnErr(err); !ok {
|
||||
slog.Error("读取节点数据失败", "err", err)
|
||||
default:
|
||||
slog.Debug("节点数据读取完成")
|
||||
}
|
||||
waitEdge <- err
|
||||
}()
|
||||
@@ -134,13 +127,8 @@ func processDataConn(ctx context.Context, edge net.Conn) error {
|
||||
var waitUser = make(chan error)
|
||||
go func() {
|
||||
_, err := io.Copy(edge, teeUser)
|
||||
switch {
|
||||
case errors.Is(err, net.ErrClosed):
|
||||
slog.Warn("用户连接意外关闭")
|
||||
case err != nil:
|
||||
if ok, err := utils.WarpConnErr(err); !ok {
|
||||
slog.Error("读取用户数据失败", "err", err)
|
||||
default:
|
||||
slog.Debug("用户数据读取完成")
|
||||
}
|
||||
waitUser <- err
|
||||
}()
|
||||
|
||||
Reference in New Issue
Block a user