handle update in run

cmd/pbm-agent/main.go

@@ -22,8 +22,6 @@ import (
 
 const mongoConnFlag = "mongodb-uri"
 
-var logOpts *log.Opts
-
 func main() {
 	rootCmd := rootCommand()
 	rootCmd.AddCommand(versionCommand())
@@ -35,12 +33,6 @@ func main() {
 	viper.AddConfigPath(".")
 	viper.WatchConfig()
 
-	viper.OnConfigChange(func(e fsnotify.Event) {
-		fmt.Printf("Config file changed: %s\n", e.Name)
-		*logOpts = *buildLogOpts()
-		fmt.Printf("Updated log options: %+v\n", logOpts)
-	})
-
 	if err := viper.ReadInConfig(); err == nil {
 		fmt.Println("Using config file:", viper.ConfigFileUsed())
 	}
@@ -83,7 +75,7 @@ func rootCommand() *cobra.Command {
 
 			hidecreds()
 
-			logOpts = buildLogOpts()
+			logOpts := buildLogOpts()
 
 			l := log.NewWithOpts(nil, "", "", logOpts).NewDefaultEvent()
 
@@ -196,6 +188,11 @@ func runAgent(
 		logOpts)
 	defer logger.Close()
 
+	viper.OnConfigChange(func(e fsnotify.Event) {
+		fmt.Printf("Config file changed: %s\n", e.Name)
+		logger.SetOpts(buildLogOpts())
+	})
+
 	ctx = log.SetLoggerToContext(ctx, logger)
 
 	mtLog.SetDateFormat(log.LogTimeFormat)