feat: use .env to set logger level

gabrielg2020 · Nov 3, 2024 · 6a31f48 · 6a31f48
1 parent d6aa7d3
commit 6a31f48
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -1,2 +1,4 @@
 # Chess API
-@TODO Complete README.md
+@TODO Complete README.md
+
+@TODO add .env explanation
diff --git a/go.mod b/go.mod
@@ -20,6 +20,7 @@ require (
 	github.com/go-playground/validator/v10 v10.22.1 // indirect
 	github.com/goccy/go-json v0.10.3 // indirect
 	github.com/google/go-cmp v0.6.0 // indirect
+	github.com/joho/godotenv v1.5.1 // indirect
 	github.com/json-iterator/go v1.1.12 // indirect
 	github.com/klauspost/cpuid/v2 v2.2.8 // indirect
 	github.com/leodido/go-urn v1.4.0 // indirect

diff --git a/go.sum b/go.sum
@@ -29,6 +29,8 @@ github.com/goccy/go-json v0.10.3/go.mod h1:oq7eo15ShAhp70Anwd5lgX2pLfOS3QCiwU/PU
 github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=
 github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
 github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
+github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
+github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
 github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
 github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
 github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=

diff --git a/pkg/logger/logger.go b/pkg/logger/logger.go
@@ -2,9 +2,11 @@ package logger
 
 import (
 	"fmt"
+	"github.com/joho/godotenv"
 	"os"
 	"path"
 	"runtime"
+	"strings"
 
 	"github.com/sirupsen/logrus"
 )
@@ -13,6 +15,12 @@ import (
 var Log *logrus.Logger
 
 func init() {
+	// Load env variables
+	err := godotenv.Load()
+	if err != nil {
+		logrus.Warn("Error loading .env file")
+	}
+
 	Log = logrus.New()
 
 	Log.SetOutput(os.Stdout)
@@ -34,5 +42,17 @@ func init() {
 		},
 		FullTimestamp: true,
 	})
-	Log.SetLevel(logrus.DebugLevel)
+
+	// Set log level
+	levelStr := os.Getenv("LOG_LEVEL")
+	if levelStr == "" {
+		levelStr = "info" // default log level
+	}
+
+	level, err := logrus.ParseLevel(strings.ToLower(levelStr))
+	if err != nil {
+		Log.Warn("Error setting log level")
+		level = logrus.InfoLevel
+	}
+	Log.SetLevel(level)
 }