加入网络层

Server/Gateway/handler/ws_handler/manager.go

@@ -0,0 +1,42 @@
+package ws_handler
+
+import (
+	"sync"
+)
+
+var UserMgr *userManager
+
+type userManager struct {
+	userMap map[int32]*Client
+	sync.RWMutex
+}
+
+func init() {
+	UserMgr = &userManager{
+		userMap: make(map[int32]*Client),
+	}
+}
+
+func (m *userManager) Add(cid int32, client *Client) {
+	m.Lock()
+	defer m.Unlock()
+	m.userMap[cid] = client
+}
+
+func (m *userManager) Delete(cid int32) {
+	m.Lock()
+	defer m.Unlock()
+	delete(m.userMap, cid)
+}
+
+func (m *userManager) GetAll() map[int32]*Client {
+	m.RLock()
+	defer m.RUnlock()
+	return m.userMap
+}
+
+func (m *userManager) GetByUID(uid int32) *Client {
+	m.RLock()
+	defer m.RUnlock()
+	return m.userMap[uid]
+}