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().SendMsg(0, []byte("ping...ping...ping"))
    21.     if err != nil {
    22.         fmt.Println(err)
    23.     }
    24. }
    26. //HelloZinxRouter Handle
    27. type HelloZinxRouter struct {
    28.     znet.BaseRouter
    29. }
    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().SendMsg(1, []byte("Hello Zinx Router V0.6"))
    37.     if err != nil {
    38.         fmt.Println(err)
    39.     }
    40. }
    42. func main() {
    43.     //创建一个server句柄
    44.     s := znet.NewServer()
    46.     //配置路由
    47.     s.AddRouter(0, &PingRouter{})
    48.     s.AddRouter(1, &HelloZinxRouter{})
    50.     //开启服务
    51.     s.Serve()
    52. }

    Server端设置了2个路由，一个是MsgId为0的消息会执行PingRouter{}重写的Handle()方法，一个是MsgId为1的消息会执行HelloZinxRouter{}重写的Handle()方法。

    我们现在写两个客户端，分别发送0消息和1消息来进行测试Zinx是否能够处理2个不同的消息业务。

    Client0.go

    1. package main
    3. import (
    4.     "fmt"
    5.     "io"
    6.     "net"
    7.     "time"
    8.     "zinx/znet"
    9. )
    11. /*
    12.     模拟客户端
    13.  */
    14. func main() {
    16.     fmt.Println("Client Test ... start")
    17.     //3秒之后发起测试请求，给服务端开启服务的机会
    18.     time.Sleep(3 * time.Second)
    20.     conn,err := net.Dial("tcp", "127.0.0.1:7777")
    21.     if err != nil {
    22.         fmt.Println("client start err, exit!")
    23.         return
    24.     }
    26.     for {
    27.         //发封包message消息
    28.         dp := znet.NewDataPack()
    29.         msg, _ := dp.Pack(znet.NewMsgPackage(0,[]byte("Zinx V0.6 Client0 Test Message")))
    30.         _, err := conn.Write(msg)
    31.         if err !=nil {
    32.             fmt.Println("write error err ", err)
    33.             return
    34.         }
    36.         //先读出流中的head部分
    37.         headData := make([]byte, dp.GetHeadLen())
    38.         _, err = io.ReadFull(conn, headData) //ReadFull 会把msg填充满为止
    39.         if err != nil {
    40.             fmt.Println("read head error")
    41.             break
    42.         }
    43.         //将headData字节流 拆包到msg中
    44.         msgHead, err := dp.Unpack(headData)
    45.         if err != nil {
    46.             fmt.Println("server unpack err:", err)
    47.             return
    48.         }
    50.         if msgHead.GetDataLen() > 0 {
    51.             //msg 是有data数据的，需要再次读取data数据
    52.             msg := msgHead.(*znet.Message)
    53.             msg.Data = make([]byte, msg.GetDataLen())
    55.             //根据dataLen从io中读取字节流
    56.             _, err := io.ReadFull(conn, msg.Data)
    57.             if err != nil {
    58.                 fmt.Println("server unpack data err:", err)
    59.                 return
    60.             }
    62.             fmt.Println("==> Recv Msg: ID=", msg.Id, ", len=", msg.DataLen, ", data=", string(msg.Data))
    63.         }
    65.         time.Sleep(1*time.Second)
    66.     }
    67. }

    Client1.go

    1. package main
    3. import (
    4.     "fmt"
    5.     "io"
    6.     "net"
    7.     "time"
    8.     "zinx/znet"
    9. )
    11. /*
    12.     模拟客户端
    13.  */
    14. func main() {
    16.     fmt.Println("Client Test ... start")
    17.     //3秒之后发起测试请求，给服务端开启服务的机会
    18.     time.Sleep(3 * time.Second)
    20.     conn,err := net.Dial("tcp", "127.0.0.1:7777")
    21.     if err != nil {
    22.         fmt.Println("client start err, exit!")
    23.         return
    24.     }
    26.     for {
    27.         //发封包message消息
    28.         dp := znet.NewDataPack()
    29.         msg, _ := dp.Pack(znet.NewMsgPackage(1,[]byte("Zinx V0.6 Client1 Test Message")))
    30.         _, err := conn.Write(msg)
    31.         if err !=nil {
    32.             fmt.Println("write error err ", err)
    33.             return
    34.         }
    36.         //先读出流中的head部分
    37.         headData := make([]byte, dp.GetHeadLen())
    38.         _, err = io.ReadFull(conn, headData) //ReadFull 会把msg填充满为止
    39.         if err != nil {
    40.             fmt.Println("read head error")
    41.             break
    42.         }
    43.         //将headData字节流 拆包到msg中
    44.         msgHead, err := dp.Unpack(headData)
    45.         if err != nil {
    46.             fmt.Println("server unpack err:", err)
    47.             return
    48.         }
    50.         if msgHead.GetDataLen() > 0 {
    51.             //msg 是有data数据的，需要再次读取data数据
    52.             msg := msgHead.(*znet.Message)
    53.             msg.Data = make([]byte, msg.GetDataLen())
    55.             //根据dataLen从io中读取字节流
    56.             _, err := io.ReadFull(conn, msg.Data)
    57.             if err != nil {
    58.                 fmt.Println("server unpack data err:", err)
    59.                 return
    60.             }
    62.             fmt.Println("==> Recv Msg: ID=", msg.Id, ", len=", msg.DataLen, ", data=", string(msg.Data))
    63.         }
    65.         time.Sleep(1*time.Second)
    66.     }
    67. }

    分别执行服务端和两个客户端

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

    服务端显示结果

    1. $ go run Server.go 
    2. Add api msgId =  0
    3. Add api msgId =  1
    4. [START] Server name: zinx v-0.6 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.6 demoApp  succ, now listenning...
    7. Reader Goroutine is  running
    8. Call PingRouter Handle
    9. recv from client : msgId= 0 , data= Zinx V0.6 Client0 Test Message
    10. Reader Goroutine is  running
    11. Call HelloZinxRouter Handle
    12. recv from client : msgId= 1 , data= Zinx V0.6 Client1 Test Message
    13. Call PingRouter Handle
    14. recv from client : msgId= 0 , data= Zinx V0.6 Client0 Test Message
    15. Call HelloZinxRouter Handle
    16. recv from client : msgId= 1 , data= Zinx V0.6 Client1 Test Message
    17. Call PingRouter Handle
    18. recv from client : msgId= 0 , data= Zinx V0.6 Client0 Test Message
    19. Call HelloZinxRouter Handle
    20. recv from client : msgId= 1 , data= Zinx V0.6 Client1 Test Message

    客户端0显示结果

    1. $ go run Client0.go 
    2. Client Test ... start
    3. ==> Recv Msg: ID= 0 , len= 18 , data= ping...ping...ping
    4. ==> Recv Msg: ID= 0 , len= 18 , data= ping...ping...ping
    5. ==> Recv Msg: ID= 0 , len= 18 , data= ping...ping...ping

    客户端1显示结果

    1. $ go run Client1.go 
    2. Client Test ... start
    3. ==> Recv Msg: ID= 1 , len= 22 , data= Hello Zinx Router V0.6
    4. ==> Recv Msg: ID= 1 , len= 22 , data= Hello Zinx Router V0.6
    5. ==> Recv Msg: ID= 1 , len= 22 , data= Hello Zinx Router V0.6