增加hook,实现上下线广播

liuligames · Aug 9, 2021 · c975f2c · c975f2c
1 parent 2f6e296
commit c975f2c
Show file tree

Hide file tree

Showing 4 changed files with 53 additions and 1 deletion.
diff --git a/demo/Server.go b/demo/Server.go
@@ -26,7 +26,6 @@ type HelloRouter struct {
 	net.BaseRouter
 }
 
-
 func (hr *HelloRouter) Handle(request iface.IRequest) {
 	fmt.Println("Call HelloRouter Handle ....")
 
@@ -39,10 +38,25 @@ func (hr *HelloRouter) Handle(request iface.IRequest) {
 	}
 }
 
+func DoConnectionBegin(conn iface.IConnection) {
+	fmt.Println("DoConnectionBegin is Called ...")
+	if err := conn.SendMsg(202, []byte("DoConnection Begin")); err != nil {
+		fmt.Println(err)
+	}
+}
+
+func DoConnectionLost(conn iface.IConnection) {
+	fmt.Println("DoConnectionLast is Called ...")
+	fmt.Println("conn id = ", conn.GetConnID(), "is lost")
+}
+
 func main() {
 
 	s := net.NewServer()
 
+	s.SetOnConnStart(DoConnectionBegin)
+	s.SetOnConnStop(DoConnectionLost)
+
 	s.AddRouter(0, &PingRouter{})
 	s.AddRouter(1, &HelloRouter{})
 

diff --git a/iface/IServer.go b/iface/IServer.go
@@ -7,4 +7,12 @@ type IServer interface {
 
 	AddRouter(msgId uint32, router IRouter)
 	GetConnManager() IConnManager
+
+	SetOnConnStart(hookFunc func(connection IConnection))
+
+	SetOnConnStop(hookFunc func(connection IConnection))
+
+	CallOnConnStart(connection IConnection)
+
+	CallOnConnStop(connection IConnection)
 }
diff --git a/net/Connection.go b/net/Connection.go
@@ -108,6 +108,7 @@ func (c *Connection) Start() {
 
 	go c.StartWriter()
 
+	c.TcpServer.CallOnConnStart(c)
 }
 
 func (c *Connection) Stop() {
@@ -118,6 +119,8 @@ func (c *Connection) Stop() {
 	}
 	c.isClosed = true
 
+	c.TcpServer.CallOnConnStop(c)
+
 	_ = c.Conn.Close()
 
 	c.ExitChan <- true

diff --git a/net/Server.go b/net/Server.go
@@ -14,6 +14,8 @@ type Server struct {
 	Port        int
 	MsgHandler  iface.IMsgHandler
 	ConnManager iface.IConnManager
+	OnConnStart func(conn iface.IConnection)
+	OnConnStop  func(conn iface.IConnection)
 }
 
 func (s *Server) Start() {
@@ -98,3 +100,28 @@ func NewServer() iface.IServer {
 		ConnManager: NewConnManager(),
 	}
 }
+
+func (s *Server) SetOnConnStart(hookFunc func(connection iface.IConnection))  {
+	s.OnConnStart = hookFunc
+}
+
+func (s *Server) SetOnConnStop(hookFunc func(connection iface.IConnection)) {
+	s.OnConnStop = hookFunc
+}
+
+func (s *Server) CallOnConnStart(connection iface.IConnection) {
+	if s.OnConnStart != nil{
+		fmt.Println("Call OnConnStart()")
+		s.OnConnStart(connection)
+	}
+}
+
+func (s *Server) CallOnConnStop(connection iface.IConnection) {
+	if s.OnConnStop != nil{
+		fmt.Println("Call OnConnStart()")
+		s.OnConnStop(connection)
+	}
+}
+
+
+