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.

195 lines
5.6 KiB

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