现在已注册的节点会检查并更新 host；修复一个导致断开的连接没有正常退出的问题

gateway/app/app.go

@@ -47,6 +47,28 @@ func NewEdge(id int32, port uint16, addr *net.TCPAddr) {
 	EdgeUpdates <- edge
 }
 
+func TryUpdateEdge(id int32, addr *net.TCPAddr) error {
+	if addr == nil {
+		return fmt.Errorf("边缘节点 %d 地址无效", id)
+	}
+
+	edge, ok := Edges.Load(id)
+	if !ok {
+		return fmt.Errorf("边缘节点 %d 不存在", id)
+	}
+
+	host := addr.IP.String()
+	if edge.Host == nil || *edge.Host != host {
+		edge.Host = &host
+		EdgeUpdates <- &core.Edge{
+			Id:   edge.Id,
+			Host: &host,
+		}
+	}
+
+	return nil
+}
+
 func StoreEdge(edge *core.Edge) error {
 	if edge == nil || edge.Id == 0 || edge.Port == nil {
 		return fmt.Errorf("无效的边缘节点: %+v", edge)

gateway/fwd/ctrl.go

@@ -107,7 +107,7 @@ func processCtrlConn(_ctx context.Context, conn net.Conn) (err error) {
 			}
 
 			cmd, err := reader.ReadByte()
-			if err := utils.WarpConnErr(err); err != nil {
+			if ok, err := utils.WarpConnErr(err); !ok {
 				errCh <- err
 				return
 			}
@@ -169,12 +169,14 @@ func processCtrlConn(_ctx context.Context, conn net.Conn) (err error) {
 }
 
 func onOpen(ctx context.Context, writer io.Writer, edgeId int32, addr net.Addr) (err error) {
-	var port uint16
+	tcpAddr, ok := addr.(*net.TCPAddr)
+	if !ok {
+		return fmt.Errorf("无效的地址类型: %T", addr)
+	}
 
+	var port uint16
 	edge, ok := app.Edges.Load(edgeId)
-	if ok && edge.Port != nil {
-		port = *edge.Port
-	} else {
+	if !ok || edge.Port == nil {
 		// 分配端口
 		app.LockPortAssign.Lock()
 
@@ -191,13 +193,16 @@ func onOpen(ctx context.Context, writer io.Writer, edgeId int32, addr net.Addr)
 			return errors.New("没有可用的端口")
 		}
 
-		if tcpAddr, ok := addr.(*net.TCPAddr); ok {
-			app.NewEdge(edgeId, port, tcpAddr)
-		} else {
-			return fmt.Errorf("无效的地址类型: %T", addr)
-		}
+		app.NewEdge(edgeId, port, tcpAddr)
 
 		app.LockPortAssign.Unlock()
+	} else {
+		// 更新边缘节点地址
+		port = *edge.Port
+		err := app.TryUpdateEdge(edgeId, tcpAddr)
+		if err != nil {
+			return fmt.Errorf("尝试更新边缘节点地址失败: %w", err)
+		}
 	}
 
 	// 启动转发服务