Add OFF level for log

subchen · Sep 29, 2017 · 6b6d7ec · 6b6d7ec
1 parent 718ed33
commit 6b6d7ec
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/log.go b/log.go
@@ -92,6 +92,11 @@ func IsErrorEnabled() bool {
 	return std.IsErrorEnabled()
 }
 
+// Indicate whether output fatal message
+func IsFatalEnabled() bool {
+	return std.IsFatalEnabled()
+}
+
 // Output a debug message
 func Debug(obj ...interface{}) {
 	std.Debug(obj...)

diff --git a/logger.go b/logger.go
@@ -21,6 +21,7 @@ const (
 	L_WARN
 	L_ERROR
 	L_FATAL
+	L_OFF
 
 	// Bits or'ed together to control what's printed.
 	F_TIME = 1 << iota
@@ -44,13 +45,15 @@ var (
 		"WARN",
 		"ERROR",
 		"FATAL",
+		"OFF",
 	}
 	levelStrWithColor = []string{
 		"\033[34mDEBUG\033[0m",
 		"\033[32mINFO\033[0m",
 		"\033[33mWARN\033[0m",
 		"\033[31mERROR\033[0m",
 		"\033[35mFATAL\033[0m",
+		"OFF",
 	}
 
 	buffer = sync.Pool{
@@ -177,6 +180,10 @@ func (l *Logger) IsErrorEnabled() bool {
 	return l.level <= L_ERROR
 }
 
+func (l *Logger) IsFatalEnabled() bool {
+	return l.level <= L_FATAL
+}
+
 func (l *Logger) Debug(obj ...interface{}) {
 	if l.level <= L_DEBUG {
 		l.log(L_DEBUG, fmt.Sprint(obj...))
@@ -232,7 +239,9 @@ func (l *Logger) Errorf(msg string, args ...interface{}) {
 }
 
 func (l *Logger) Fatalf(msg string, args ...interface{}) {
-	l.log(L_FATAL, fmt.Sprintf(msg, args...))
+	if l.level <= L_FATAL {
+		l.log(L_FATAL, fmt.Sprintf(msg, args...))
+	}
 }
 
 func (l *Logger) log(level int, msg string) {