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

    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