fix(server): add goroutine locks in rpc handlers

cloudwego · Jan 10, 2025 · 7aa6fbd · 7aa6fbd
1 parent 6d967d1
commit 7aa6fbd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/server/server.go b/server/server.go
@@ -45,6 +45,7 @@ import (
 	"github.com/cloudwego/kitex/pkg/rpcinfo"
 	"github.com/cloudwego/kitex/pkg/serviceinfo"
 	"github.com/cloudwego/kitex/pkg/stats"
+	"github.com/cloudwego/kitex/pkg/utils"
 )
 
 // Server is an abstraction of an RPC server. It accepts connections and dispatches them to the service
@@ -377,6 +378,8 @@ func (s *server) buildCoreMiddleware() endpoint.Middleware {
 
 func (s *server) invokeHandleEndpoint() endpoint.Endpoint {
 	return func(ctx context.Context, args, resp interface{}) (err error) {
+		utils.GoroutineLock()
+		defer utils.GoroutineUnlock()
 		ri := rpcinfo.GetRPCInfo(ctx)
 		methodName := ri.Invocation().MethodName()
 		serviceName := ri.Invocation().ServiceName()