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. }