fix: 停机先停接受并等待循环再断开 MQTT

cmd/nixmsg/serve_shutdown_test.go

@@ -0,0 +1,73 @@
+package main
+
+import (
+	"context"
+	"sync"
+	"testing"
+	"time"
+
+	"git.asio.asia/nixevol/NixMsg/internal/config"
+	"github.com/mochi-mqtt/server/v2/packets"
+)
+
+func TestServeShutdownWithLiveMQTT(t *testing.T) {
+	dataDir := t.TempDir()
+	cfgPath := writeTestConfig(t, dataDir)
+	initAdminForTest(t, dataDir)
+	enableRegistration(t, dataDir, "uplink-code")
+
+	cfg, err := config.Load(cfgPath)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if vErr := cfg.Validate(); vErr != nil {
+		t.Fatal(vErr)
+	}
+
+	ctx, cancel := context.WithCancel(context.Background())
+	errCh := make(chan error, 1)
+	go func() { errCh <- runServe(ctx, cfg) }()
+
+	addr := waitListenAddr(t, dataDir, 15*time.Second)
+	base := "http://" + addr
+	registerEP(t, base, "alice", "password12", "Alice")
+	registerEP(t, base, "bob", "password12", "Bob")
+
+	ws := mqttSessionLogin(t, base, "alice", "password12")
+	tcp := mqttSessionLoginTCP(t, addr, "bob", "password12")
+
+	cancel()
+
+	var gotWG sync.WaitGroup
+	var wsReason, tcpReason byte
+	var wsOK, tcpOK bool
+	gotWG.Add(2)
+	go func() {
+		defer gotWG.Done()
+		wsReason, wsOK = ws.disconnectReason(12 * time.Second)
+	}()
+	go func() {
+		defer gotWG.Done()
+		tcpReason, tcpOK = tcp.disconnectReason(12 * time.Second)
+	}()
+
+	select {
+	case err := <-errCh:
+		if err != nil {
+			t.Fatalf("runServe: %v", err)
+		}
+	case <-time.After(15 * time.Second):
+		t.Fatal("runServe did not return within 15s")
+	}
+	gotWG.Wait()
+
+	want := packets.ErrServerShuttingDown.Code
+	if !wsOK || wsReason != want {
+		t.Fatalf("ws disconnect ok=%v reason=%#x want %#x", wsOK, wsReason, want)
+	}
+	if !tcpOK || tcpReason != want {
+		t.Fatalf("tcp disconnect ok=%v reason=%#x want %#x", tcpOK, tcpReason, want)
+	}
+	ws.Close()
+	tcp.Close()
+}