merge: broker b01-b12

cmd/nixmsg/uplink.go

@@ -5,12 +5,14 @@ import (
 	"encoding/json"
 	"errors"
 	"log/slog"
+	"sync"
 
 	"git.asio.asia/nixevol/NixMsg/internal/app/group"
 	"git.asio.asia/nixevol/NixMsg/internal/app/identity"
 	"git.asio.asia/nixevol/NixMsg/internal/app/message"
 	"git.asio.asia/nixevol/NixMsg/internal/app/port"
 	"git.asio.asia/nixevol/NixMsg/internal/app/presence"
+	"git.asio.asia/nixevol/NixMsg/internal/broker"
 	"git.asio.asia/nixevol/NixMsg/internal/metrics"
 	"git.asio.asia/nixevol/NixMsg/internal/protocol"
 )
@@ -25,9 +27,31 @@ type appUplink struct {
 	down     port.Downlink
 	log      *slog.Logger
 	metrics  *metrics.Registry
+
+	lifeMu    sync.Mutex
+	lifeLocks map[string]*sync.Mutex
+	hsMu      sync.Mutex
+	handshake map[port.ConnID]string // 已 hello 的连接代号 → 端编号
+}
+
+func (u *appUplink) epLife(endpointID string) *sync.Mutex {
+	u.lifeMu.Lock()
+	defer u.lifeMu.Unlock()
+	if u.lifeLocks == nil {
+		u.lifeLocks = make(map[string]*sync.Mutex)
+	}
+	m := u.lifeLocks[endpointID]
+	if m == nil {
+		m = &sync.Mutex{}
+		u.lifeLocks[endpointID] = m
+	}
+	return m
 }
 
 func (u *appUplink) OnSessionEstablished(ctx context.Context, conn port.ConnInfo) error {
+	lk := u.epLife(conn.EndpointID)
+	lk.Lock()
+	defer lk.Unlock()
 	u.conns.Set(conn.EndpointID, message.LiveConn{
 		ConnID:        conn.ConnID,
 		MaxPacketSize: conn.MaxPacketSize,
@@ -36,6 +60,15 @@ func (u *appUplink) OnSessionEstablished(ctx context.Context, conn port.ConnInfo
 }
 
 func (u *appUplink) OnHandshakeComplete(ctx context.Context, hs port.HandshakeInfo) error {
+	lk := u.epLife(hs.EndpointID)
+	lk.Lock()
+	defer lk.Unlock()
+	u.hsMu.Lock()
+	if u.handshake == nil {
+		u.handshake = make(map[port.ConnID]string)
+	}
+	u.handshake[hs.ConnID] = hs.EndpointID
+	u.hsMu.Unlock()
 	live := message.LiveConn{
 		ConnID:          hs.ConnID,
 		MaxReceiveBytes: hs.MaxReceiveBytes,
@@ -50,11 +83,18 @@ func (u *appUplink) OnHandshakeComplete(ctx context.Context, hs port.HandshakeIn
 }
 
 func (u *appUplink) OnDisconnect(ctx context.Context, conn port.ConnInfo, reason port.DisconnectReason) {
+	lk := u.epLife(conn.EndpointID)
+	lk.Lock()
+	defer lk.Unlock()
 	if u.presence != nil {
 		u.presence.ClearWatch(conn.ConnID)
 	}
+	u.hsMu.Lock()
+	_, handshook := u.handshake[conn.ConnID]
+	delete(u.handshake, conn.ConnID)
+	u.hsMu.Unlock()
 	live, ok := u.conns.Current(conn.EndpointID)
-	isCurrent := ok && live.ConnID == conn.ConnID
+	isCurrent := ok && live.ConnID == conn.ConnID && handshook
 	if err := u.msg.OnDisconnect(ctx, conn.EndpointID, conn.ConnID, isCurrent); err != nil {
 		u.log.Error("message disconnect", "endpoint", conn.EndpointID, "err", err)
 	}
@@ -293,7 +333,7 @@ func (u *appUplink) publishResp(ctx context.Context, conn port.ConnInfo, resp pr
 		return
 	}
 	if live, ok := u.conns.Current(conn.EndpointID); ok && live.ConnID == conn.ConnID {
-		limit := respPayloadLimit(live.MaxPacketSize, live.MaxReceiveBytes)
+		limit := broker.EffectivePayloadLimit(live.MaxPacketSize, live.MaxReceiveBytes)
 		if limit > 0 && len(b) > limit {
 			tooLarge := protocol.Resp{
 				V:     protocol.Version,
@@ -313,20 +353,6 @@ func (u *appUplink) publishResp(ctx context.Context, conn port.ConnInfo, resp pr
 	}
 }
 
-func respPayloadLimit(maxPacketSize uint32, maxRecvBytes int) int {
-	limit := 0
-	if maxRecvBytes > 0 {
-		limit = maxRecvBytes
-	}
-	if maxPacketSize > 0 {
-		n := int(maxPacketSize)
-		if limit == 0 || n < limit {
-			limit = n
-		}
-	}
-	return limit
-}
-
 func peekRID(payload []byte) string {
 	var peek struct {
 		RID string `json:"rid"`