A) IServer增添路由添加功能

我们需要给IServer类，增加一个抽象方法AddRouter,目的也是让Zinx框架使用者，可以自定一个Router处理业务方法。

zinx/ziface/irouter.go

  1. package ziface
  3. //定义服务器接口
  4. type IServer interface{
  5.     //启动服务器方法
  6.     Start()
  7.     //停止服务器方法
  8.     Stop()
  9.     //开启业务服务方法
  10.     Serve()
  11.     //路由功能：给当前服务注册一个路由业务方法，供客户端链接处理使用
  12.     AddRouter(router IRouter)
  13. }

B) Server类增添Router成员

有了抽象的方法，自然Server就要实现，并且还要添加一个Router成员.

zinx/znet/server.go

  1. //iServer 接口实现，定义一个Server服务类
  2. type Server struct {
  3.     //服务器的名称
  4.     Name string
  5.     //tcp4 or other
  6.     IPVersion string
  7.     //服务绑定的IP地址
  8.     IP string
  9.     //服务绑定的端口
  10.     Port int
  11.     //当前Server由用户绑定的回调router,也就是Server注册的链接对应的处理业务
  12.     Router ziface.IRouter
  13. }

然后NewServer()方法， 初始化Server对象的方法也要加一个初始化成员

  1. /*
  2.   创建一个服务器句柄
  3.  */
  4. func NewServer (name string) ziface.IServer {
  5.     s:= &Server {
  6.         Name :name,
  7.         IPVersion:"tcp4",
  8.         IP:"0.0.0.0",
  9.         Port:7777,
  10.         Router: nil,
  11.     }
  13.     return s
  14. }

C) Connection类绑定一个Router成员

zinx/znet/connection.go

  1. type Connection struct {
  2.     //当前连接的socket TCP套接字
  3.     Conn *net.TCPConn
  4.     //当前连接的ID 也可以称作为SessionID，ID全局唯一
  5.     ConnID uint32
  6.     //当前连接的关闭状态
  7.     isClosed bool
  9.     //该连接的处理方法router
  10.     Router  ziface.IRouter
  12.     //告知该链接已经退出/停止的channel
  13.     ExitBuffChan chan bool
  14. }

D) 在Connection调用注册的Router处理业务

zinx/znet/connection.go

  1. func (c *Connection) StartReader() {
  2.     fmt.Println("Reader Goroutine is  running")
  3.     defer fmt.Println(c.RemoteAddr().String(), " conn reader exit!")
  4.     defer c.Stop()
  6.     for  {
  7.         //读取我们最大的数据到buf中
  8.         buf := make([]byte, 512)
  9.         _, err := c.Conn.Read(buf)
  10.         if err != nil {
  11.             fmt.Println("recv buf err ", err)
  12.             c.ExitBuffChan <- true
  13.             continue
  14.         }
  15.         //得到当前客户端请求的Request数据
  16.         req := Request{
  17.             conn:c,
  18.             data:buf,
  19.         }
  20.         //从路由Routers 中找到注册绑定Conn的对应Handle
  21.         go func (request ziface.IRequest) {
  22.             //执行注册的路由方法
  23.             c.Router.PreHandle(request)
  24.             c.Router.Handle(request)
  25.             c.Router.PostHandle(request)
  26.         }(&req)
  27.     }
  28. }

这里我们在conn读取完客户端数据之后，将数据和conn封装到一个Request中，作为Router的输入数据。

然后我们开启一个goroutine去调用给Zinx框架注册好的路由业务。