当我们在使用链接处理的时候，希望和链接绑定一些用户的数据，或者参数。那么我们现在可以把当前链接设定一些传递参数的接口或者方法。

10.1 给链接添加链接配置接口

zinx/ziface/iconnection.go

  1. //定义连接接口
  2. type IConnection interface {
  3.     //启动连接，让当前连接开始工作
  4.     Start()
  5.     //停止连接，结束当前连接状态M
  6.     Stop()
  8.     //从当前连接获取原始的socket TCPConn
  9.     GetTCPConnection() *net.TCPConn
  10.     //获取当前连接ID
  11.     GetConnID() uint32
  12.     //获取远程客户端地址信息
  13.     RemoteAddr() net.Addr
  15.     //直接将Message数据发送数据给远程的TCP客户端(无缓冲)
  16.     SendMsg(msgId uint32, data []byte) error
  17.     //直接将Message数据发送给远程的TCP客户端(有缓冲)
  18.     SendBuffMsg(msgId uint32, data []byte) error
  20.     //设置链接属性
  21.     SetProperty(key string, value interface{})
  22.     //获取链接属性
  23.     GetProperty(key string)(interface{}, error)
  24.     //移除链接属性
  25.     RemoveProperty(key string)
  26. }

这里增添了3个方法SetProperty(),GetProperty(),RemoveProperty().那么property是什么类型的呢，我么接下来看看Connection的定义。

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

10.3 链接属性Zinx-V0.10单元测试

那么，接下来，我们简单测试一下链接属性的设置与提取是否可用。

Server.go

  1. package main
  3. import (
  4.     "fmt"
  5.     "zinx/ziface"
  6.     "zinx/znet"
  7. )
  9. //ping test 自定义路由
  10. type PingRouter struct {
  11.     znet.BaseRouter
  12. }
  14. //Ping Handle
  15. func (this *PingRouter) Handle(request ziface.IRequest) {
  16.     fmt.Println("Call PingRouter Handle")
  17.     //先读取客户端的数据，再回写ping...ping...ping
  18.     fmt.Println("recv from client : msgId=", request.GetMsgID(), ", data=", string(request.GetData()))
  20.     err := request.GetConnection().SendBuffMsg(0, []byte("ping...ping...ping"))
  21.     if err != nil {
  22.         fmt.Println(err)
  23.     }
  24. }
  26. type HelloZinxRouter struct {
  27.     znet.BaseRouter
  28. }
  30. //HelloZinxRouter Handle
  31. func (this *HelloZinxRouter) Handle(request ziface.IRequest) {
  32.     fmt.Println("Call HelloZinxRouter Handle")
  33.     //先读取客户端的数据，再回写ping...ping...ping
  34.     fmt.Println("recv from client : msgId=", request.GetMsgID(), ", data=", string(request.GetData()))
  36.     err := request.GetConnection().SendBuffMsg(1, []byte("Hello Zinx Router V0.10"))
  37.     if err != nil {
  38.         fmt.Println(err)
  39.     }
  40. }
  42. //创建连接的时候执行
  43. func DoConnectionBegin(conn ziface.IConnection) {
  44.     fmt.Println("DoConnecionBegin is Called ... ")
  46.     //=============设置两个链接属性，在连接创建之后===========
  47.     fmt.Println("Set conn Name, Home done!")
  48.     conn.SetProperty("Name", "Aceld")
  49.     conn.SetProperty("Home", "https://www.jianshu.com/u/35261429b7f1")
  50.     //===================================================
  52.     err := conn.SendMsg(2, []byte("DoConnection BEGIN..."))
  53.     if err != nil {
  54.         fmt.Println(err)
  55.     }
  56. }
  58. //连接断开的时候执行
  59. func DoConnectionLost(conn ziface.IConnection) {
  60.     //============在连接销毁之前，查询conn的Name，Home属性=====
  61.     if name, err:= conn.GetProperty("Name"); err == nil {
  62.         fmt.Println("Conn Property Name = ", name)
  63.     }
  65.     if home, err := conn.GetProperty("Home"); err == nil {
  66.         fmt.Println("Conn Property Home = ", home)
  67.     }
  68.     //===================================================
  70.     fmt.Println("DoConneciotnLost is Called ... ")
  71. }
  73. func main() {
  74.     //创建一个server句柄
  75.     s := znet.NewServer()
  77.     //注册链接hook回调函数
  78.     s.SetOnConnStart(DoConnectionBegin)
  79.     s.SetOnConnStop(DoConnectionLost)
  81.     //配置路由
  82.     s.AddRouter(0, &PingRouter{})
  83.     s.AddRouter(1, &HelloZinxRouter{})
  85.     //开启服务
  86.     s.Serve()
  87. }

这里主要看DoConnectionBegin()和DoConnectionLost()两个函数的实现， 利用在两个Hook函数中，设置链接属性和提取链接属性。链接创建之后给当前链接绑定两个属性”Name”,”Home”, 那么我们在随时可以通过conn.GetProperty()方法得到链接已经设置的属性。

  1. $go run Server.go
  1. $go run Client0.go

服务端：

  1. $ go run Server.go 
  2. Add api msgId =  0
  3. Add api msgId =  1
  4. [START] Server name: zinx v-0.10 demoApp,listenner at IP: 127.0.0.1, Port 7777 is starting
  5. [Zinx] Version: V0.4, MaxConn: 3, MaxPacketSize: 4096
  6. start Zinx server   zinx v-0.10 demoApp  succ, now listenning...
  7. Worker ID =  9  is started.
  8. Worker ID =  5  is started.
  9. Worker ID =  6  is started.
  10. Worker ID =  7  is started.
  11. Worker ID =  8  is started.
  12. Worker ID =  1  is started.
  13. Worker ID =  0  is started.
  14. Worker ID =  2  is started.
  15. Worker ID =  3  is started.
  16. Worker ID =  4  is started.
  17. connection add to ConnManager successfully: conn num =  1
  18. ---> CallOnConnStart....
  19. DoConnecionBegin is Called ... 
  20. Set conn Name, Home done!
  21. [Writer Goroutine is running]
  22. [Reader Goroutine is running]
  23. Add ConnID= 0  request msgID= 0 to workerID= 0
  24. Call PingRouter Handle
  25. recv from client : msgId= 0 , data= Zinx V0.8 Client0 Test Message
  26. Add ConnID= 0  request msgID= 0 to workerID= 0
  27. Call PingRouter Handle
  28. recv from client : msgId= 0 , data= Zinx V0.8 Client0 Test Message
  29. Add ConnID= 0  request msgID= 0 to workerID= 0
  30. Call PingRouter Handle
  31. recv from client : msgId= 0 , data= Zinx V0.8 Client0 Test Message
  32. read msg head error  read tcp4 127.0.0.1:7777->127.0.0.1:55208: read: connection reset by peer
  33. Conn Stop()...ConnID =  0
  34. ---> CallOnConnStop....
  35. Conn Property Name =  Aceld
  36. Conn Property Home =  https://www.jianshu.com/u/35261429b7f1
  37. DoConneciotnLost is Called ... 
  38. connection Remove ConnID= 0  successfully: conn num =  0
  39. 127.0.0.1:55208 [conn Reader exit!]
  40. 127.0.0.1:55208 [conn Writer exit!]

客户端：

  1. $ go run Client0.go 
  2. Client Test ... start
  3. ==> Recv Msg: ID= 2 , len= 21 , data= DoConnection BEGIN...
  4. ==> Recv Msg: ID= 0 , len= 18 , data= ping...ping...ping
  5. ==> Recv Msg: ID= 0 , len= 18 , data= ping...ping...ping
  6. ^Csignal: interrupt

当我们终止客户端链接，那么服务端在断开链接之前，已经读取到了conn的两个属性Name和Home