main: make sure the Welcome TiProxy info is always printed. (#507)

pingcap · Apr 13, 2024 · f427940 · f427940
1 parent c51070d
commit f427940
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/pkg/manager/logger/manager.go b/pkg/manager/logger/manager.go
@@ -52,6 +52,10 @@ func (lm *LoggerManager) Init(cfgch <-chan *config.Config) {
 	}, nil, lm.logger)
 }
 
+func (lm *LoggerManager) SetLoggerLevel(l zapcore.Level) {
+	lm.level.SetLevel(l)
+}
+
 func (lm *LoggerManager) watchCfg(ctx context.Context, cfgch <-chan *config.Config) {
 	for {
 		select {

diff --git a/pkg/server/server.go b/pkg/server/server.go
@@ -75,7 +75,11 @@ func NewServer(ctx context.Context, sctx *sctx.Context) (srv *Server, err error)
 	//
 	// TODO: there is a race condition that printInfo and logmgr may concurrently execute:
 	// logmgr may havenot been initialized with logfile yet
+	// Make sure the TiProxy info is always printed.
+	level := lg.Level()
+	srv.LoggerManager.SetLoggerLevel(zap.InfoLevel)
 	printInfo(lg)
+	srv.LoggerManager.SetLoggerLevel(level)
 
 	// setup metrics
 	srv.MetricsManager.Init(ctx, lg.Named("metrics"))