init: 通用库抽离——httpx/logger/wsc/wscclient/jwtx(合并分叉副本); 顶层包布局; module path git.zeroonesoft.cn/golib/zogo

wsc/server.go

@@ -0,0 +1,123 @@
+package wsc
+
+import (
+	"net/http"
+	"sync"
+	"sync/atomic"
+
+	"github.com/gin-gonic/gin"
+	"github.com/gorilla/websocket"
+)
+
+// ============================================================
+// 默认配置
+// ============================================================
+
+var defaultUpgrader = websocket.Upgrader{
+	ReadBufferSize:  4096,
+	WriteBufferSize: 4096,
+	CheckOrigin: func(r *http.Request) bool {
+		return true // 生产环境请限制 Origin
+	},
+}
+
+// ============================================================
+// Server — 连接管理器
+// ============================================================
+
+// Server 管理 WebSocket 连接的全局状态。
+type Server struct {
+	upgrader websocket.Upgrader
+
+	// 连接生命周期回调
+	OnConnect    func(ctx *Context)
+	OnDisconnect func(ctx *Context)
+
+	// OnUpgrade 升级前钩子：可用于鉴权（如校验 URL token），返回 error 即拒绝连接（401）。
+	// 钩子中通过 c.Set(k, v) 写入的值会自动透传到 *Context（见 newContext）。
+	OnUpgrade func(c *gin.Context) error
+
+	// 房间管理
+	mu    sync.RWMutex
+	rooms map[string]*Room
+
+	// 连接统计
+	connCount atomic.Int64
+}
+
+// NewServer 创建 Server。
+func NewServer() *Server {
+	return &Server{
+		upgrader: defaultUpgrader,
+		rooms:    make(map[string]*Room),
+	}
+}
+
+// SetUpgrader 自定义升级器。
+func (s *Server) SetUpgrader(u websocket.Upgrader) {
+	s.upgrader = u
+}
+
+// ============================================================
+// 连接统计
+// ============================================================
+
+// ActiveCount 返回当前活跃连接数。
+func (s *Server) ActiveCount() int64 {
+	return s.connCount.Load()
+}
+
+func (s *Server) onSessionOpen() {
+	s.connCount.Add(1)
+}
+
+func (s *Server) onSessionClosed() {
+	s.connCount.Add(-1)
+}
+
+// ============================================================
+// 房间管理
+// ============================================================
+
+// GetOrCreateRoom 获取或创建房间。
+func (s *Server) GetOrCreateRoom(name string) *Room {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+
+	if room, ok := s.rooms[name]; ok {
+		return room
+	}
+
+	room := newRoom(name)
+	room.onEmpty = func(name string) {
+		s.mu.Lock()
+		defer s.mu.Unlock()
+		// 二次判空：防止 leave→onEmpty 之间新成员加入
+		if r, exists := s.rooms[name]; exists && r.Len() == 0 {
+			delete(s.rooms, name)
+		}
+	}
+	s.rooms[name] = room
+	return room
+}
+
+// GetRoom 获取房间（不存在返回 nil）。
+func (s *Server) GetRoom(name string) *Room {
+	s.mu.RLock()
+	defer s.mu.RUnlock()
+	return s.rooms[name]
+}
+
+// RemoveRoom 移除房间。
+func (s *Server) RemoveRoom(name string) {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+	delete(s.rooms, name)
+}
+
+// RoomCount 返回当前房间数。
+func (s *Server) RoomCount() int {
+	s.mu.RLock()
+	defer s.mu.RUnlock()
+	return len(s.rooms)
+}