feat 排队

internal/handler/ws_handler/client/manager.go

@@ -0,0 +1,64 @@
+package client
+
+import (
+	"sync"
+)
+
+var UserMgr *userManager
+
+type userManager struct {
+	userMap map[int64]*Client
+	sync.RWMutex
+}
+
+func init() {
+	UserMgr = &userManager{
+		userMap: make(map[int64]*Client),
+	}
+}
+
+func (m *userManager) Add(usn int64, client *Client) {
+	m.Lock()
+	defer m.Unlock()
+	m.userMap[usn] = client
+}
+
+func (m *userManager) Delete(usn int64) {
+	m.Lock()
+	defer m.Unlock()
+	delete(m.userMap, usn)
+}
+
+func (m *userManager) GetAll() map[int64]*Client {
+	m.RLock()
+	defer m.RUnlock()
+
+	copyMap := make(map[int64]*Client, len(m.userMap))
+	for k, v := range m.userMap {
+		copyMap[k] = v
+	}
+	return copyMap
+}
+
+func (m *userManager) GetAllInterface() []interface{} {
+	m.RLock()
+	defer m.RUnlock()
+
+	r := make([]interface{}, 0)
+	for _, v := range m.userMap {
+		r = append(r, v)
+	}
+	return r
+}
+
+func (m *userManager) GetByUSN(usn int64) *Client {
+	m.RLock()
+	defer m.RUnlock()
+	return m.userMap[usn]
+}
+
+func (m *userManager) GetSize() int32 {
+	m.RLock()
+	defer m.RUnlock()
+	return int32(len(m.userMap))
+}