deployment query rate limit

command/agent/agent.go

@@ -3,6 +3,7 @@ package agent
 import (
 	"context"
 	"fmt"
+	"github.com/hashicorp/nomad/nomad/deploymentwatcher"
 	"io"
 	"io/ioutil"
 	golog "log"
@@ -420,6 +421,15 @@ func convertServerConfig(agentConfig *Config) (*nomad.Config, error) {
 		conf.RPCMaxConnsPerClient = limit
 	}
 
+	// Set deployment rate limit
+	if rate := agentConfig.Server.DeploymentRateLimit; rate == 0 {
+		conf.DeploymentRateLimit = deploymentwatcher.LimitStateQueriesPerSecond
+	} else if rate > 0 {
+		conf.DeploymentRateLimit = rate
+	} else {
+		return nil, fmt.Errorf("deployment-rate-limit must be greater than 0")
+	}
+
 	// Add Enterprise license configs
 	conf.LicenseEnv = agentConfig.Server.LicenseEnv
 	conf.LicensePath = agentConfig.Server.LicensePath

command/agent/command.go

@@ -98,6 +98,7 @@ func (c *Command) readConfig() *Config {
 		cmdConfig.Server.ServerJoin.RetryInterval = d
 		return nil
 	}), "retry-interval", "")
+	flags.Float64Var(&cmdConfig.Server.DeploymentRateLimit, "deployment-rate-limit", 0, "")
 
 	// Client-only options
 	flags.StringVar(&cmdConfig.Client.StateDir, "state-dir", "", "")

command/agent/config.go

@@ -516,6 +516,10 @@ type ServerConfig struct {
 	ExtraKeysHCL []string `hcl:",unusedKeys" json:"-"`
 
 	Search *Search `hcl:"search"`
+
+	// DeploymentRateLimit is in queries per second and is used by the
+	// DeploymentWatcher to throttle the amount of simultaneously deployments
+	DeploymentRateLimit float64
 }
 
 // Search is used in servers to configure search API options.

nomad/config.go

@@ -363,6 +363,10 @@ type Config struct {
 	// AgentShutdown is used to call agent.Shutdown from the context of a Server
 	// It is used primarily for licensing
 	AgentShutdown func() error
+
+	// DeploymentRateLimit is in queries per second and is used by the
+	// DeploymentWatcher to throttle the amount of simultaneously deployments
+	DeploymentRateLimit float64
 }
 
 // CheckVersion is used to check if the ProtocolVersion is valid

nomad/server.go

@@ -1021,7 +1021,7 @@ func (s *Server) setupDeploymentWatcher() error {
 		raftShim,
 		s.staticEndpoints.Deployment,
 		s.staticEndpoints.Job,
-		deploymentwatcher.LimitStateQueriesPerSecond,
+		s.config.DeploymentRateLimit,
 		deploymentwatcher.CrossDeploymentUpdateBatchDuration,
 	)