Adds min-level option to SetupLoggerv0.3.0

1 files changed, 8 insertions, 3 deletions

diff --git a/logger.go b/logger.go
index d6d5278..11fd5c1 100644
--- a/logger.go
+++ b/logger.go
@@ -2,15 +2,20 @@ package logger
 
 import (
 	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
 )
 
-func SetupLogger() (*zap.Logger, error) {
+func SetupLogger(level zapcore.Level) (*zap.Logger, error) {
 	var logger *zap.Logger
 	var err error
 	if Production {
-		logger, err = zap.NewProduction()
+		cfg := zap.NewProductionConfig()
+		cfg.Level.SetLevel(level)
+		logger, err = cfg.Build()
 	} else {
-		logger, err = zap.NewDevelopment()
+		cfg := zap.NewDevelopmentConfig()
+		cfg.Level.SetLevel(level)
+		logger, err = cfg.Build()
 	}
 	if err != nil {
 		return nil, err