fix(utils): wait for goroutine locks in Stop()

pkg/remote/trans/netpollmux/server_handler.go

@@ -27,7 +27,6 @@ import (
 
 	"github.com/cloudwego/netpoll"
 
-	goroutinelock "github.com/cloudwego/kitex/internal/utils/goroutine_lock"
 	"github.com/cloudwego/kitex/pkg/endpoint"
 	"github.com/cloudwego/kitex/pkg/gofunc"
 	"github.com/cloudwego/kitex/pkg/kerrors"
@@ -366,9 +365,7 @@ func (t *svrTransHandler) GracefulShutdown(ctx context.Context) error {
 			}
 			return true
 		})
-		// 4. waiting for goroutine locks to be released
-		goroutinelock.GoroutineWg.Wait()
-		// 5. waiting all crrst packets received by client
+		// 4. waiting all crrst packets received by client
 		deadline := time.Now().Add(defaultExitWaitGracefulShutdownTime)
 		ticker := time.NewTicker(defaultExitWaitGracefulShutdownTime / 10)
 		defer ticker.Stop()

pkg/remote/trans/nphttp2/server_handler.go

@@ -31,7 +31,6 @@ import (
 
 	"github.com/cloudwego/netpoll"
 
-	goroutinelock "github.com/cloudwego/kitex/internal/utils/goroutine_lock"
 	"github.com/cloudwego/kitex/pkg/endpoint"
 	"github.com/cloudwego/kitex/pkg/gofunc"
 	"github.com/cloudwego/kitex/pkg/kerrors"
@@ -408,7 +407,6 @@ func (t *svrTransHandler) GracefulShutdown(ctx context.Context) error {
 			}
 			t.mu.Unlock()
 			if activeNums == 0 {
-				goroutinelock.GoroutineWg.Wait()
 				return nil
 			}
 		case <-graceTimer.C:
@@ -418,9 +416,6 @@ func (t *svrTransHandler) GracefulShutdown(ctx context.Context) error {
 				svrTrans := elem.Value.(*SvrTrans)
 				svrTrans.tr.Close()
 			}
-			// Goroutine Locks must wait after all connections are closed, otherwise new connections might be created
-			// while waiting for goroutine locks.
-			goroutinelock.GoroutineWg.Wait()
 			t.mu.Unlock()
 			return nil
 		}

server/server.go

@@ -30,6 +30,7 @@ import (
 	"github.com/cloudwego/localsession/backup"
 
 	internal_server "github.com/cloudwego/kitex/internal/server"
+	goroutinelock "github.com/cloudwego/kitex/internal/utils/goroutine_lock"
 	"github.com/cloudwego/kitex/pkg/acl"
 	"github.com/cloudwego/kitex/pkg/diagnosis"
 	"github.com/cloudwego/kitex/pkg/discovery"
@@ -319,6 +320,9 @@ func (s *server) Stop() (err error) {
 			}
 			s.svr = nil
 		}
+		// Goroutine Locks must wait after all connections are closed, otherwise new connections might be created while
+		// waiting for goroutine locks.
+		goroutinelock.GoroutineWg.Wait()
 	})
 	return
 }