review feedback

grpc · Aug 29, 2024 · 6deaf1e · 6deaf1e
1 parent 0941401
commit 6deaf1e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/server.go b/server.go
@@ -1359,6 +1359,7 @@ func (s *Server) processUnaryRPC(ctx context.Context, t transport.ServerTranspor
 		}
 		return err
 	}
+	defer d.Free()
 	if channelz.IsOn() {
 		t.IncrMsgRecv()
 	}
@@ -1391,8 +1392,6 @@ func (s *Server) processUnaryRPC(ctx context.Context, t transport.ServerTranspor
 	}
 	ctx = NewContextWithServerTransportStream(ctx, stream)
 	reply, appErr := md.Handler(info.serviceImpl, ctx, df, s.opts.unaryInt)
-	defer d.Free()
-
 	if appErr != nil {
 		appStatus, ok := status.FromError(appErr)
 		if !ok {