/
connmanager.go
82 lines (66 loc) · 1.98 KB
/
connmanager.go
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
package tcpserver
import (
"errors"
"github.com/jqiris/kungfu/logger"
"sync"
tcpface "github.com/jqiris/kungfu/tcpface"
)
type ConnManager struct {
connections map[int]tcpface.IConnection //管理的连接信息
connLock sync.RWMutex //读写连接的读写锁
}
func NewConnManager() *ConnManager {
return &ConnManager{
connections: make(map[int]tcpface.IConnection),
}
}
// Add 添加链接
func (connMgr *ConnManager) Add(conn tcpface.IConnection) {
//保护共享资源Map 加写锁
connMgr.connLock.Lock()
defer connMgr.connLock.Unlock()
//将conn连接添加到ConnMananger中
connMgr.connections[conn.GetConnID()] = conn
logger.Info("connection add to ConnManager successfully: conn num = ", connMgr.Len())
}
// Remove 删除连接
func (connMgr *ConnManager) Remove(conn tcpface.IConnection) {
//保护共享资源Map 加写锁
connMgr.connLock.Lock()
defer connMgr.connLock.Unlock()
//删除连接信息
delete(connMgr.connections, conn.GetConnID())
logger.Info("connection Remove ConnID=", conn.GetConnID(), " successfully: conn num = ", connMgr.Len())
}
// Get 利用ConnID获取链接
func (connMgr *ConnManager) Get(connID int) (tcpface.IConnection, error) {
//保护共享资源Map 加读锁
connMgr.connLock.RLock()
defer connMgr.connLock.RUnlock()
if conn, ok := connMgr.connections[connID]; ok {
return conn, nil
} else {
return nil, errors.New("connection not found")
}
}
// Len 获取当前连接
func (connMgr *ConnManager) Len() int {
return len(connMgr.connections)
}
// ClearConn 清除并停止所有连接
func (connMgr *ConnManager) ClearConn() {
//保护共享资源Map 加写锁
connMgr.connLock.Lock()
defer connMgr.connLock.Unlock()
//停止并删除全部的连接信息
for connID, conn := range connMgr.connections {
//停止
err := conn.Close()
if err != nil {
logger.Error(err)
}
//删除
delete(connMgr.connections, connID)
}
logger.Info("Clear All Connections successfully: conn num = ", connMgr.Len())
}