Throttle log messages (#134)

Author: pglass

config/config.go

@@ -16,7 +16,8 @@ const (
 	ConfigPathFlag = "config"
 	LogLevelFlag   = "level"
 
-	DefaultPProfAddress = "localhost:6060"
+	DefaultPProfAddress          = "localhost:6060"
+	DefaultLoggingThrottleMaxRPS = 10.0
 )
 
 type TransportType string
@@ -115,6 +116,7 @@ type (
 		SearchAttributeTranslation SATranslationConfig   `yaml:"searchAttributeTranslation"`
 		Metrics                    *MetricsConfig        `yaml:"metrics"`
 		ProfilingConfig            ProfilingConfig       `yaml:"profiling"`
+		Logging                    LoggingConfig         `yaml:"logging"`
 	}
 
 	SATranslationConfig struct {
@@ -167,6 +169,10 @@ type (
 	MetricsConfig struct {
 		Prometheus PrometheusConfig `yaml:"prometheus"`
 	}
+
+	LoggingConfig struct {
+		ThrottleMaxRPS float64 `yaml:"throttleMaxRPS"`
+	}
 )
 
 func (c ProxyClientConfig) IsMux() bool {
@@ -369,3 +375,10 @@ func (s SATranslationConfig) ToMaps(inBound bool) (map[string]map[string]string,
 	}
 	return reqMap, respMap
 }
+
+func (l LoggingConfig) GetThrottleMaxRPS() float64 {
+	if l.ThrottleMaxRPS > 0 {
+		return l.ThrottleMaxRPS
+	}
+	return DefaultLoggingThrottleMaxRPS
+}

proxy/proxy.go

@@ -255,7 +255,12 @@ func NewProxy(
 	proxy := &Proxy{
 		config:       s2sConfig,
 		transManager: transManager,
-		logger:       logger,
+		logger: log.NewThrottledLogger(
+			logger,
+			func() float64 {
+				return s2sConfig.Logging.GetThrottleMaxRPS()
+			},
+		),
 	}
 
 	// Proxy consists of two grpc servers: inbound and outbound. The flow looks like the following:
@@ -272,7 +277,7 @@ func NewProxy(
 				Config:    s2sConfig,
 			},
 			transManager,
-			logger,
+			proxy.logger,
 		)
 	}
 
@@ -284,7 +289,7 @@ func NewProxy(
 				Config:    s2sConfig,
 			},
 			transManager,
-			logger,
+			proxy.logger,
 		)
 	}