Change to use bool to define debug option.

drkaka · Feb 27, 2017 · e809899 · e809899
1 parent adad632
commit e809899
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lg.go b/lg.go
@@ -26,10 +26,10 @@ func Rfc3339NanoEncoder(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
 }
 
 // InitLogger must be first to be called.
-func InitLogger(debug string) {
+func InitLogger(debug bool) {
 	var cfg zap.Config
 
-	if debug == "debug" {
+	if debug {
 		cfg = zap.NewDevelopmentConfig()
 	} else {
 		cfg = zap.NewProductionConfig()

diff --git a/lg_test.go b/lg_test.go
@@ -5,6 +5,6 @@ import (
 )
 
 func TestLg(t *testing.T) {
-	InitLogger("debug")
+	InitLogger(true)
 	L(nil).Info("test message.")
 }