Add ratelimiter to QueryWorkflow API (cadence-workflow#3735)

zshni · Nov 5, 2020 · c11be59 · c11be59
1 parent 995c4dc
commit c11be59
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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)...)
 	}