Better func

Author: Fangliding

transport/internet/httpupgrade/hub.go

@@ -31,7 +31,17 @@ func (s *server) Addr() net.Addr {
 	return nil
 }
 
-func (s *server) Handle(conn net.Conn) (stat.Connection, error) {
+func (s *server) Handle(conn net.Conn) {
+	conn, err := s.upgrade(conn)
+	if err != nil {
+		errors.LogInfoInner(context.Background(), err, "failed to handle request")
+		return
+	}
+	s.addConn(conn)
+}
+
+// upgrade execute a fake websocket upgrade process and return the available connection
+func (s *server) upgrade(conn net.Conn) (stat.Connection, error) {
 	connReader := bufio.NewReader(conn)
 	req, err := http.ReadRequest(connReader)
 	if err != nil {
@@ -99,14 +109,7 @@ func (s *server) keepAccepting() {
 		if err != nil {
 			return
 		}
-		go func() {
-			handledConn, err := s.Handle(conn)
-			if err != nil {
-				errors.LogInfoInner(context.Background(), err, "failed to handle request")
-				return
-			}
-			s.addConn(handledConn)
-		}()
+		go s.Handle(conn)
 	}
 }