#### 9.1 創建鏈接管理模塊 這里面我們就需要對鏈接有一個管理的模塊. 我們在`ziface`和`znet`分別建立`iconnmanager.go`和`connmanager.go`文件 > zinx/ziface/iconmanager.go ```go package ziface /* 連接管理抽象層 */ type IConnManager interface { Add(conn IConnection) //添加鏈接 Remove(conn IConnection) //刪除連接 Get(connID uint32) (IConnection, error) //利用ConnID獲取鏈接 Len() int //獲取當前連接 ClearConn() //刪除并停止所有鏈接 } ``` 這里定義了一些接口方法，添加鏈接、刪除鏈接、根據ID獲取鏈接、鏈接數量、和清除鏈接等。 > zinx/znet/connmanager.go ```go package znet import ( "errors" "fmt" "sync" "zinx/ziface" ) /* 連接管理模塊 */ type ConnManager struct { connections map[uint32]ziface.IConnection //管理的連接信息 connLock sync.RWMutex //讀寫連接的讀寫鎖 } /* 創建一個鏈接管理 */ func NewConnManager() *ConnManager { return &ConnManager{ connections:make(map[uint32] ziface.IConnection), } } //添加鏈接 func (connMgr *ConnManager) Add(conn ziface.IConnection) { //保護共享資源Map 加寫鎖 connMgr.connLock.Lock() defer connMgr.connLock.Unlock() //將conn連接添加到ConnMananger中 connMgr.connections[conn.GetConnID()] = conn fmt.Println("connection add to ConnManager successfully: conn num = ", connMgr.Len()) } //刪除連接 func (connMgr *ConnManager) Remove(conn ziface.IConnection) { //保護共享資源Map 加寫鎖 connMgr.connLock.Lock() defer connMgr.connLock.Unlock() //刪除連接信息 delete(connMgr.connections, conn.GetConnID()) fmt.Println("connection Remove ConnID=",conn.GetConnID(), " successfully: conn num = ", connMgr.Len()) } //利用ConnID獲取鏈接 func (connMgr *ConnManager) Get(connID uint32) (ziface.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") } } //獲取當前連接 func (connMgr *ConnManager) Len() int { return len(connMgr.connections) } //清除并停止所有連接 func (connMgr *ConnManager) ClearConn() { //保護共享資源Map 加寫鎖 connMgr.connLock.Lock() defer connMgr.connLock.Unlock() //停止并刪除全部的連接信息 for connID, conn := range connMgr.connections { //停止 conn.Stop() //刪除 delete(connMgr.connections,connID) } fmt.Println("Clear All Connections successfully: conn num = ", connMgr.Len()) } ``` 這里面`ConnManager`中，其中用一個map來承載全部的連接信息，key是連接ID，value則是連接本身。其中有一個讀寫鎖`connLock`主要是針對map做多任務修改時的保護作用。 `Remove()`方法只是單純的將conn從map中摘掉，而`ClearConn()`方法則會先停止鏈接業務，`c.Stop()`，然后再從map中摘除。