99 "strconv"
1010 "strings"
1111
12+ "github.com/go-logr/zapr"
1213 "go.uber.org/zap"
1314 "k8s.io/apimachinery/pkg/runtime"
1415 "k8s.io/apimachinery/pkg/types"
@@ -25,6 +26,7 @@ import (
2526 localruntime "runtime"
2627 ctrl "sigs.k8s.io/controller-runtime"
2728 runtime_cluster "sigs.k8s.io/controller-runtime/pkg/cluster"
29+ kubelog "sigs.k8s.io/controller-runtime/pkg/log"
2830 metricsServer "sigs.k8s.io/controller-runtime/pkg/metrics/server"
2931 crWebhook "sigs.k8s.io/controller-runtime/pkg/webhook"
3032
@@ -275,13 +277,7 @@ func setupWebhook(ctx context.Context, cfg *rest.Config, log *zap.SugaredLogger,
275277}
276278
277279func initializeEnvironment () {
278- omOperatorEnv := os .Getenv (util .OmOperatorEnv )
279- configuredEnv := omOperatorEnv
280- if ! validateEnv (omOperatorEnv ) {
281- omOperatorEnv = operatorEnvironments [0 ]
282- }
283-
284- initLogger (omOperatorEnv )
280+ omOperatorEnv , configuredEnv := getOperatorEnvs ()
285281
286282 if configuredEnv != omOperatorEnv {
287283 log .Infof ("Configured environment %s, not recognized. Must be one of %v" , configuredEnv , operatorEnvironments )
@@ -320,6 +316,15 @@ func initializeEnvironment() {
320316 env .PrintWithPrefix (printableEnvPrefixes )
321317}
322318
319+ func getOperatorEnvs () (string , string ) {
320+ omOperatorEnv := os .Getenv (util .OmOperatorEnv )
321+ configuredEnv := omOperatorEnv
322+ if ! validateEnv (omOperatorEnv ) {
323+ omOperatorEnv = operatorEnvironments [0 ]
324+ }
325+ return omOperatorEnv , configuredEnv
326+ }
327+
323328// quoteKey reports whether key is required to be quoted. Taken from: 1.22.0 mod.go
324329func quoteKey (key string ) bool {
325330 return len (key ) == 0 || strings .ContainsAny (key , "= \t \r \n \" `" )
@@ -360,22 +365,41 @@ func validateEnv(env string) bool {
360365 return stringutil .Contains (operatorEnvironments [:], env )
361366}
362367
363- func initLogger (env string ) {
368+ func init () {
369+ InitGlobalLogger ()
370+ }
371+
372+ func InitGlobalLogger () {
373+ omOperatorEnv , _ := getOperatorEnvs ()
374+
364375 var logger * zap.Logger
365376 var e error
366377
367- switch env {
378+ switch omOperatorEnv {
368379 case "prod" :
369380 logger , e = zap .NewProduction ()
370381 case "dev" , "local" :
371382 // Overriding the default stacktrace behavior - have them only for errors but not for warnings
372383 logger , e = zap .NewDevelopment (zap .AddStacktrace (zap .ErrorLevel ))
384+ default :
385+ // if for some reason we didn't set a logger, let's be safe and default to prod
386+ fmt .Println ("No OPERATOR_ENV set, defaulting setting logger to prod" )
387+ logger , e = zap .NewProduction ()
373388 }
374389
375390 if e != nil {
376391 fmt .Println ("Failed to create logger, will use the default one" )
377392 fmt .Println (e )
393+ // in the worst case logger might stay nil, replacing everything with a nil logger,
394+ // we don't want that
395+ logger = zap .S ().Desugar ()
378396 }
397+
398+ // Set the global logger used by our operator
379399 zap .ReplaceGlobals (logger )
400+ // Set the logger for controller-runtime based on the general level of the operator
401+ kubelog .SetLogger (zapr .NewLogger (logger ))
402+
403+ // Set the logger used by main.go
380404 log = zap .S ()
381405}
0 commit comments