• 10.2 链接属性方法实现

    10.2 链接属性方法实现

    zinx/znet/connction.go

    1. type Connection struct {
    2.     //当前Conn属于哪个Server
    3.     TcpServer ziface.IServer
    4.     //当前连接的socket TCP套接字
    5.     Conn *net.TCPConn
    6.     //当前连接的ID 也可以称作为SessionID，ID全局唯一
    7.     ConnID uint32
    8.     //当前连接的关闭状态
    9.     isClosed bool
    10.     //消息管理MsgId和对应处理方法的消息管理模块
    11.     MsgHandler ziface.IMsgHandle
    12.     //告知该链接已经退出/停止的channel
    13.     ExitBuffChan chan bool
    14.     //无缓冲管道，用于读、写两个goroutine之间的消息通信
    15.     msgChan chan []byte
    16.     //有关冲管道，用于读、写两个goroutine之间的消息通信
    17.     msgBuffChan chan []byte
    19.     // ================================
    20.     //链接属性
    21.     property     map[string]interface{}
    22.     //保护链接属性修改的锁
    23.     propertyLock sync.RWMutex
    24.     // ================================
    25. }
    27. //创建连接的方法
    28. func NewConntion(server ziface.IServer, conn *net.TCPConn, connID uint32, msgHandler ziface.IMsgHandle) *Connection {
    29.     //初始化Conn属性
    30.     c := &Connection{
    31.         TcpServer:    server,
    32.         Conn:         conn,
    33.         ConnID:       connID,
    34.         isClosed:     false,
    35.         MsgHandler:   msgHandler,
    36.         ExitBuffChan: make(chan bool, 1),
    37.         msgChan:      make(chan []byte),
    38.         msgBuffChan:  make(chan []byte, utils.GlobalObject.MaxMsgChanLen),
    39.         property:     make(map[string]interface{}), //对链接属性map初始化
    40.     }
    42.     //将新创建的Conn添加到链接管理中
    43.     c.TcpServer.GetConnMgr().Add(c)
    44.     return c
    45. }
    47. // ... 
    49. //设置链接属性
    50. func (c *Connection) SetProperty(key string, value interface{}) {
    51.     c.propertyLock.Lock()
    52.     defer c.propertyLock.Unlock()
    54.     c.property[key] = value
    55. }
    57. //获取链接属性
    58. func (c *Connection) GetProperty(key string) (interface{}, error) {
    59.     c.propertyLock.RLock()
    60.     defer c.propertyLock.RUnlock()
    62.     if value, ok := c.property[key]; ok  {
    63.         return value, nil
    64.     } else {
    65.         return nil, errors.New("no property found")
    66.     }
    67. }
    69. //移除链接属性
    70. func (c *Connection) RemoveProperty(key string) {
    71.     c.propertyLock.Lock()
    72.     defer c.propertyLock.Unlock()
    74.     delete(c.property, key)
    75. }