From c11be59af075947cade8e0e1cbf332afc1e7ce2e Mon Sep 17 00:00:00 2001 From: Anish Adhikari Date: Thu, 5 Nov 2020 12:54:10 -0800 Subject: [PATCH] Add ratelimiter to QueryWorkflow API (#3735) --- service/frontend/workflowHandler.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/service/frontend/workflowHandler.go b/service/frontend/workflowHandler.go index 1b34dd009c1..16f89698371 100644 --- a/service/frontend/workflowHandler.go +++ b/service/frontend/workflowHandler.go @@ -3076,6 +3076,10 @@ func (wh *WorkflowHandler) QueryWorkflow( return nil, wh.error(errQueryDisallowedForDomain, scope, getWfIDRunIDTags(wfExecution)...) } + if ok := wh.allow(queryRequest); !ok { + return nil, wh.error(createServiceBusyError(), scope) + } + if err := wh.versionChecker.ClientSupported(ctx, wh.config.EnableClientVersionCheck()); err != nil { return nil, wh.error(err, scope, getWfIDRunIDTags(wfExecution)...) }