Support Graceful Shutdown for server framework

This commit is contained in:
Tnze
2022-11-29 14:37:56 +08:00
parent 458e8af5d0
commit 3750dff99d

View File

@ -60,11 +60,11 @@ func (s *Server) Listen(addr string) error {
if err != nil { if err != nil {
return err return err
} }
go s.acceptConn(&conn) go s.AcceptConn(&conn)
} }
} }
func (s *Server) acceptConn(conn *net.Conn) { func (s *Server) AcceptConn(conn *net.Conn) {
defer conn.Close() defer conn.Close()
protocol, intention, err := s.handshake(conn) protocol, intention, err := s.handshake(conn)
if err != nil { if err != nil {