Add SetLogLevel to control logs output

iFaceless · May 14, 2021 · 05e89e8 · 05e89e8
1 parent f7ca803
commit 05e89e8
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -190,6 +190,7 @@ func New(opts ...Option) (*Chell, error)
 func Dump(dst, src interface{}, opts ...Option) error 
 func DumpWithContext(ctx context.Context, dst, src interface{}, opts ...Option)
 func SetDebug(v bool)
+func SetLevel(level LogLevel)
 func SetMaxPoolSize(size int)
 func CleanUp()
 ```

diff --git a/README_CN.md b/README_CN.md
@@ -188,6 +188,7 @@ func New(opts ...Option) (*Chell, error)
 func Dump(dst, src interface{}, opts ...Option) error 
 func DumpWithContext(ctx context.Context, dst, src interface{}, opts ...Option)
 func SetDebug(v bool)
+func SetLogLevel(level LogLevel)
 func SetMaxPoolSize(size int)
 func CleanUp()
 ```

diff --git a/log.go b/log.go
@@ -15,3 +15,17 @@ func SetDebug(v bool) {
 		logger.SetLevel(logrus.InfoLevel)
 	}
 }
+
+type LogLevel = logrus.Level
+
+const (
+	ErrorLevel = logrus.ErrorLevel
+	WarnLevel  = logrus.WarnLevel
+	InfoLevel  = logrus.InfoLevel
+	DebugLevel = logrus.DebugLevel
+)
+
+// SetLogLevel block the logs who's level is lower
+func SetLogLevel(level LogLevel) {
+	logger.SetLevel(level)
+}