fix: add logs methods (#311)

* fix: add log methods * delete some method
erda-project · Jan 12, 2024 · e4c6c25 · e4c6c25
1 parent 7847b1f
commit e4c6c25
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/base/logs/logrusx/log.go b/base/logs/logrusx/log.go
@@ -15,6 +15,8 @@
 package logrusx
 
 import (
+	"io"
+
 	"github.com/sirupsen/logrus"
 
 	"github.com/erda-project/erda-infra/base/logs"
@@ -59,6 +61,10 @@ func (l *Logger) SetLevel(lvl string) error {
 	return nil
 }
 
+func (l *Logger) SetOutput(output io.Writer) {
+	l.Logger.SetOutput(output)
+}
+
 func processOptions(logr *logrus.Logger, logger *Logger, opt interface{}) {
 	switch val := opt.(type) {
 	case setNameOption:

diff --git a/base/logs/logs.go b/base/logs/logs.go
@@ -14,6 +14,10 @@
 
 package logs
 
+import (
+	"io"
+)
+
 // Logger .
 type Logger interface {
 	Sub(name string) Logger
@@ -32,4 +36,5 @@ type Logger interface {
 	Fatalf(template string, args ...interface{})
 
 	SetLevel(lvl string) error
+	SetOutput(output io.Writer)
 }