Add ManagementNode config option to disable DRL

Fixes TykTechnologies#613.
giddily · Mar 13, 2017 · 07d1a62 · 07d1a62
1 parent 931ebcb
commit 07d1a62
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/config.go b/config.go
@@ -194,6 +194,7 @@ type Config struct {
 	EnableNonTransactionalRateLimiter bool                   `json:"enable_non_transactional_rate_limiter"`
 	EnableSentinelRateLImiter         bool                   `json:"enable_sentinel_rate_limiter"`
 	EnableRedisRollingLimiter         bool                   `json:"enable_redis_rolling_limiter"`
+	ManagementNode                    bool                   `json:"management_node"`
 	Monitor                           MonitorConfig
 	OauthRefreshExpire                int64                                 `json:"oauth_refresh_token_expire"`
 	OauthTokenExpire                  int32                                 `json:"oauth_token_expire"`

diff --git a/main.go b/main.go
@@ -1241,13 +1241,17 @@ func startHeartBeat() {
 }
 
 func startDRL() {
-	if !config.EnableSentinelRateLImiter && !config.EnableRedisRollingLimiter {
-		log.WithFields(logrus.Fields{
-			"prefix": "main",
-		}).Info("Initialising distributed rate limiter")
-		setupDRL()
-		startRateLimitNotifications()
+	switch {
+	case config.ManagementNode,
+		config.EnableSentinelRateLImiter,
+		config.EnableRedisRollingLimiter:
+		return
 	}
+	log.WithFields(logrus.Fields{
+		"prefix": "main",
+	}).Info("Initialising distributed rate limiter")
+	setupDRL()
+	startRateLimitNotifications()
 }
 
 func listen(l, controlListener net.Listener, err error) {