3.4 Zinx-V0.3代码实现

    zinx/znet/conneciont.go

    3. import (
    4.     "fmt"
    5.     "net"
    6.     "zinx/ziface"
    7. )
    9. type Connection struct {
    10.     //当前连接的socket TCP套接字
    11.     Conn *net.TCPConn
    12.     //当前连接的ID 也可以称作为SessionID，ID全局唯一
    13.     ConnID uint32
    14.     //当前连接的关闭状态
    15.     isClosed bool
    17.     //该连接的处理方法router
    18.     Router  ziface.IRouter
    20.     //告知该链接已经退出/停止的channel
    21.     ExitBuffChan chan bool
    22. }
    25. //创建连接的方法
    26. func NewConntion(conn *net.TCPConn, connID uint32, router ziface.IRouter) *Connection{
    27.     c := &Connection{
    28.         Conn:     conn,
    29.         ConnID:   connID,
    30.         isClosed: false,
    31.         Router: router,
    32.         ExitBuffChan: make(chan bool, 1),
    33.     }
    35.     return c
    36. }
    39.     defer fmt.Println(c.RemoteAddr().String(), " conn reader exit!")
    40.     defer c.Stop()
    42.     for  {
    43.         //读取我们最大的数据到buf中
    44.         buf := make([]byte, 512)
    45.         _, err := c.Conn.Read(buf)
    46.         if err != nil {
    47.             fmt.Println("recv buf err ", err)
    48.             c.ExitBuffChan <- true
    49.             continue
    50.         }
    51.         //得到当前客户端请求的Request数据
    52.         req := Request{
    53.             conn:c,
    54.             data:buf,
    55.         }
    56.         //从路由Routers 中找到注册绑定Conn的对应Handle
    57.         go func (request ziface.IRequest) {
    58.             //执行注册的路由方法
    59.             c.Router.PreHandle(request)
    60.             c.Router.Handle(request)
    61.             c.Router.PostHandle(request)
    62.         }(&req)
    63.     }
    64. }
    66. //启动连接，让当前连接开始工作
    67. func (c *Connection) Start() {
    69.     //开启处理该链接读取到客户端数据之后的请求业务
    70.     go c.StartReader()
    72.     for {
    73.         select {
    74.         case <- c.ExitBuffChan:
    76.         }
    77.     }
    78. }
    80. //停止连接，结束当前连接状态M
    81. func (c *Connection) Stop() {
    82.     //1. 如果当前链接已经关闭
    83.     if c.isClosed == true {
    84.         return
    85.     }
    86.     c.isClosed = true
    88.     //TODO Connection Stop() 如果用户注册了该链接的关闭回调业务，那么在此刻应该显示调用
    90.     // 关闭socket链接
    91.     c.Conn.Close()
    93.     //通知从缓冲队列读数据的业务，该链接已经关闭
    94.     c.ExitBuffChan <- true
    96.     //关闭该链接全部管道
    97.     close(c.ExitBuffChan)
    98. }
    100. //从当前连接获取原始的socket TCPConn
    101. func (c *Connection) GetTCPConnection() *net.TCPConn {
    102.     return c.Conn
    103. }
    105. //获取当前连接ID
    106. func (c *Connection) GetConnID() uint32{
    107.     return c.ConnID
    108. }
    110. //获取远程客户端地址信息
    111. func (c *Connection) RemoteAddr() net.Addr {