From 907a822ec333e4caee8dbf60654a1b8a1e40bf8c Mon Sep 17 00:00:00 2001 From: Gurudatta Patil Date: Thu, 9 Apr 2026 11:50:26 -0500 Subject: [PATCH] add defer --- service/history/handler.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/service/history/handler.go b/service/history/handler.go index d3850db8332..ff41a64a04f 100644 --- a/service/history/handler.go +++ b/service/history/handler.go @@ -1968,7 +1968,7 @@ func (h *Handler) StreamWorkflowReplicationMessages( server historyservice.HistoryService_StreamWorkflowReplicationMessagesServer, ) (retErr error) { // Note that since this is not a unary RPC, we cannot use the interceptor to capture panics. - metrics.CapturePanic(h.logger, h.metricsHandler, &retErr) + defer metrics.CapturePanic(h.logger, h.metricsHandler, &retErr) getter := headers.NewGRPCHeaderGetter(server.Context()) clientClusterShardID, serverClusterShardID, err := history.DecodeClusterShardMD(getter) if err != nil {