feat: send error logs to stderr

argoproj-labs · Oct 5, 2021 · f318ebf · f318ebf
1 parent fa64809
commit f318ebf
Showing 1 changed file with 23 additions and 5 deletions.
diff --git a/shared/util/log.go b/shared/util/log.go
@@ -1,15 +1,33 @@
 package util
 
 import (
-	"os"
-
 	"github.com/bombsimon/logrusr"
 	"github.com/go-logr/logr"
-	"github.com/sirupsen/logrus"
+	log "github.com/sirupsen/logrus"
+	"github.com/sirupsen/logrus/hooks/writer"
+	"io"
+
+	"os"
 )
 
 func NewLogger() logr.Logger {
-	l := logrus.New()
-	l.SetOutput(os.Stdout)
+	l := log.New()
+	l.SetOutput(io.Discard)
+	l.AddHook(&writer.Hook{
+		Writer: os.Stderr,
+		LogLevels: []log.Level{
+			log.PanicLevel,
+			log.FatalLevel,
+			log.ErrorLevel,
+			log.WarnLevel,
+		},
+	})
+	l.AddHook(&writer.Hook{
+		Writer: os.Stdout,
+		LogLevels: []log.Level{
+			log.InfoLevel,
+			log.DebugLevel,
+		},
+	})
 	return logrusr.NewLogger(l)
 }