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

anish531213 · web-flow · commit c11be59af075 · 2020-11-05T12:54:10.000-08:00
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)...)
 	}

Original file line number	Diff line number	Diff line change
`@@ -3076,6 +3076,10 @@ func (wh *WorkflowHandler) QueryWorkflow(`
`3076`	`3076`	`return nil, wh.error(errQueryDisallowedForDomain, scope, getWfIDRunIDTags(wfExecution)...)`
`3077`	`3077`	`}`
`3078`	`3078`
	`3079`	`+ if ok := wh.allow(queryRequest); !ok {`
	`3080`	`+ return nil, wh.error(createServiceBusyError(), scope)`
	`3081`	`+ }`
	`3082`	`+`
`3079`	`3083`	`if err := wh.versionChecker.ClientSupported(ctx, wh.config.EnableClientVersionCheck()); err != nil {`
`3080`	`3084`	`return nil, wh.error(err, scope, getWfIDRunIDTags(wfExecution)...)`
`3081`	`3085`	`}`