You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

183 lines
5.3 KiB

  1. package znet
  2. import (
  3. "fmt"
  4. "net"
  5. "zinx/utils"
  6. "zinx/ziface"
  7. )
  8. var zinx_logo = `
  9. `
  10. var top_line = `┌───────────────────────────────────────────────────┐`
  11. var border_line = ``
  12. var bottom_line = `└───────────────────────────────────────────────────┘`
  13. //iServer 接口实现,定义一个Server服务类
  14. type Server struct {
  15. //服务器的名称
  16. Name string
  17. //tcp4 or other
  18. IPVersion string
  19. //服务绑定的IP地址
  20. IP string
  21. //服务绑定的端口
  22. Port int
  23. //当前Server的消息管理模块,用来绑定MsgId和对应的处理方法
  24. msgHandler ziface.IMsgHandle
  25. //当前Server的链接管理器
  26. ConnMgr ziface.IConnManager
  27. //该Server的连接创建时Hook函数
  28. OnConnStart func(conn ziface.IConnection)
  29. //该Server的连接断开时的Hook函数
  30. OnConnStop func(conn ziface.IConnection)
  31. }
  32. /*
  33. 创建一个服务器句柄
  34. */
  35. func NewServer () ziface.IServer {
  36. s:= &Server {
  37. Name :utils.GlobalObject.Name,
  38. IPVersion:"tcp4",
  39. IP:utils.GlobalObject.Host,
  40. Port:utils.GlobalObject.TcpPort,
  41. msgHandler: NewMsgHandle(),
  42. ConnMgr:NewConnManager(),
  43. }
  44. return s
  45. }
  46. //============== 实现 ziface.IServer 里的全部接口方法 ========
  47. //开启网络服务
  48. func (s *Server) Start() {
  49. fmt.Printf("[START] Server name: %s,listenner at IP: %s, Port %d is starting\n", s.Name, s.IP, s.Port)
  50. //开启一个go去做服务端Linster业务
  51. go func() {
  52. //0 启动worker工作池机制
  53. s.msgHandler.StartWorkerPool()
  54. //1 获取一个TCP的Addr
  55. addr, err := net.ResolveTCPAddr(s.IPVersion, fmt.Sprintf("%s:%d", s.IP, s.Port))
  56. if err != nil {
  57. fmt.Println("resolve tcp addr err: ", err)
  58. return
  59. }
  60. //2 监听服务器地址
  61. listenner, err:= net.ListenTCP(s.IPVersion, addr)
  62. if err != nil {
  63. fmt.Println("listen", s.IPVersion, "err", err)
  64. return
  65. }
  66. //已经监听成功
  67. fmt.Println("start Zinx server ", s.Name, " succ, now listenning...")
  68. //TODO server.go 应该有一个自动生成ID的方法
  69. var cid uint32
  70. cid = 0
  71. //3 启动server网络连接业务
  72. for {
  73. //3.1 阻塞等待客户端建立连接请求
  74. conn, err := listenner.AcceptTCP()
  75. if err != nil {
  76. fmt.Println("Accept err ", err)
  77. continue
  78. }
  79. fmt.Println("Get conn remote addr = ", conn.RemoteAddr().String())
  80. //3.2 设置服务器最大连接控制,如果超过最大连接,那么则关闭此新的连接
  81. if s.ConnMgr.Len() >= utils.GlobalObject.MaxConn {
  82. conn.Close()
  83. continue
  84. }
  85. //3.3 处理该新连接请求的 业务 方法, 此时应该有 handler 和 conn是绑定的
  86. dealConn := NewConntion(s, conn, cid, s.msgHandler)
  87. cid ++
  88. //3.4 启动当前链接的处理业务
  89. go dealConn.Start()
  90. }
  91. }()
  92. }
  93. //停止服务
  94. func (s *Server) Stop() {
  95. fmt.Println("[STOP] Zinx server , name " , s.Name)
  96. //将其他需要清理的连接信息或者其他信息 也要一并停止或者清理
  97. s.ConnMgr.ClearConn()
  98. }
  99. //运行服务
  100. func (s *Server) Serve() {
  101. s.Start()
  102. //TODO Server.Serve() 是否在启动服务的时候 还要处理其他的事情呢 可以在这里添加
  103. //阻塞,否则主Go退出, listenner的go将会退出
  104. select{}
  105. }
  106. //路由功能:给当前服务注册一个路由业务方法,供客户端链接处理使用
  107. func (s *Server)AddRouter(msgId uint32, router ziface.IRouter) {
  108. s.msgHandler.AddRouter(msgId, router)
  109. }
  110. //得到链接管理
  111. func (s *Server) GetConnMgr() ziface.IConnManager {
  112. return s.ConnMgr
  113. }
  114. //设置该Server的连接创建时Hook函数
  115. func (s *Server) SetOnConnStart(hookFunc func (ziface.IConnection)) {
  116. s.OnConnStart = hookFunc
  117. }
  118. //设置该Server的连接断开时的Hook函数
  119. func (s *Server) SetOnConnStop(hookFunc func (ziface.IConnection)) {
  120. s.OnConnStop = hookFunc
  121. }
  122. //调用连接OnConnStart Hook函数
  123. func (s *Server) CallOnConnStart(conn ziface.IConnection) {
  124. if s.OnConnStart != nil {
  125. fmt.Println("---> CallOnConnStart....")
  126. s.OnConnStart(conn)
  127. }
  128. }
  129. //调用连接OnConnStop Hook函数
  130. func (s *Server) CallOnConnStop(conn ziface.IConnection) {
  131. if s.OnConnStop != nil {
  132. fmt.Println("---> CallOnConnStop....")
  133. s.OnConnStop(conn)
  134. }
  135. }
  136. func init() {
  137. fmt.Println(zinx_logo)
  138. fmt.Println(top_line)
  139. fmt.Println(fmt.Sprintf("%s [Github] https://github.com/aceld %s", border_line, border_line))
  140. fmt.Println(fmt.Sprintf("%s [tutorial] https://www.jianshu.com/p/23d07c0a28e5 %s", border_line, border_line))
  141. fmt.Println(bottom_line)
  142. fmt.Printf("[Zinx] Version: %s, MaxConn: %d, MaxPacketSize: %d\n",
  143. utils.GlobalObject.Version,
  144. utils.GlobalObject.MaxConn,
  145. utils.GlobalObject.MaxPacketSize)
  146. }