connmanage.go
1.9 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package components
import "sync"
type ConnManage struct {
mu sync.RWMutex
conns map[uint32]IConnection
r2cRW sync.RWMutex
r2c map[string]uint32 // role to connID
u2cRW sync.RWMutex
u2c map[string]uint32 // uid to connID
}
func NewConnManage() *ConnManage {
return &ConnManage{
mu: sync.RWMutex{},
conns: make(map[uint32]IConnection),
r2cRW: sync.RWMutex{},
r2c: make(map[string]uint32),
u2cRW: sync.RWMutex{},
u2c: make(map[string]uint32),
}
}
func (c *ConnManage) AddConn(id uint32, connection IConnection) {
c.mu.Lock()
defer c.mu.Unlock()
c.conns[id] = connection
}
func (c *ConnManage) GetConn(id uint32) IConnection {
c.mu.RLock()
defer c.mu.RUnlock()
return c.conns[id]
}
func (c *ConnManage) DelConn(id uint32) IConnection {
c.mu.Lock()
defer c.mu.Unlock()
conn := c.conns[id]
delete(c.conns, id)
return conn
}
func (c *ConnManage) Range(f func(key interface{}, value interface{}) bool) {
c.mu.Lock()
defer c.mu.Unlock()
for k, v := range c.conns {
if ok := f(k, v); !ok {
return
}
}
}
func (c *ConnManage) StopAllConns() {
c.Range(func(key interface{}, value interface{}) bool {
conn := value.(IConnection)
conn.Stop()
return true
})
}
func (c *ConnManage) AddRID(rid string, id uint32) {
c.r2cRW.Lock()
defer c.r2cRW.Unlock()
c.r2c[rid] = id
}
func (c *ConnManage) DelRID(rid string) {
c.r2cRW.Lock()
defer c.r2cRW.Unlock()
delete(c.r2c, rid)
}
func (c *ConnManage) GetConnByRID(rid string) IConnection {
c.r2cRW.RLock()
defer c.r2cRW.RUnlock()
cid := c.r2c[rid]
return c.GetConn(cid)
}
func (c *ConnManage) AddUID(uid string, id uint32) {
c.u2cRW.Lock()
defer c.u2cRW.Unlock()
c.u2c[uid] = id
}
func (c *ConnManage) DelUID(uid string) {
c.r2cRW.Lock()
defer c.r2cRW.Unlock()
delete(c.r2c, uid)
}
func (c *ConnManage) GetConnByUID(uid string) IConnection {
c.u2cRW.RLock()
defer c.u2cRW.RUnlock()
cid := c.u2c[uid]
return c.GetConn(cid)
}