Enable dynamic logging level setting

uber · Mar 8, 2018 · 443457a · 443457a
1 parent a3272d1
commit 443457a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/runtime/gateway.go b/runtime/gateway.go
@@ -80,6 +80,7 @@ type Gateway struct {
 	TChannelRouter   *TChannelRouter
 	Tracer           opentracing.Tracer
 
+	atomLevel       *zap.AtomicLevel
 	loggerFile      *os.File
 	scopeCloser     io.Closer
 	metricsBackend  tally.CachedStatsReporter
@@ -231,6 +232,8 @@ func (gateway *Gateway) registerPredefined() {
 	gateway.HTTPRouter.RegisterRaw(
 		"GET", "/debug/pprof/block", pprof.Handler("block").ServeHTTP,
 	)
+	gateway.HTTPRouter.RegisterRaw("GET", "/debug/loglevel", gateway.atomLevel.ServeHTTP)
+	gateway.HTTPRouter.RegisterRaw("PUT", "/debug/loglevel", gateway.atomLevel.ServeHTTP)
 
 	gateway.HTTPRouter.Register("GET", "/health", NewRouterEndpoint(
 		gateway.Logger, gateway.AllHostScope,
@@ -420,17 +423,19 @@ func (gateway *Gateway) setupLogger(config *StaticConfig) error {
 		}
 	}
 
+	atomLevel := zap.NewAtomicLevelAt(logLevel)
 	prodCore := zapcore.NewCore(
 		logEncoder,
 		output,
-		logLevel,
+		atomLevel,
 	)
 	zapLogger := zap.New(
 		NewInstrumentedZapCore(
 			prodCore, gateway.AllHostScope,
 		),
 	)
 
+	gateway.atomLevel = &atomLevel
 	gateway.logEncoder = logEncoder
 	gateway.logWriteSyncer = output