refactor: logger

backend/cmd/main.go

@@ -15,8 +15,7 @@ func main() {
 	}
 
 	// Setup logger
-	logger.InitLogger()
+  defer logger.Sync()
-    defer logger.GetLogger().Sync()
 
 	// Connect & migrate database
 	// models.Init()

backend/internal/logger/logger.go

@@ -1,29 +1,34 @@
 package logger
 
 import (
-    "go.uber.org/zap"
+	"sync"
-	  "easywish/config"
+
+	"go.uber.org/zap"
+	"easywish/config"
 )
 
-var logger *zap.Logger
+var (
-
+	logger *zap.Logger
-func InitLogger() {
+	once   sync.Once
-    var err error
+)
-	cfg := config.GetConfig()
-
-    if cfg.Environment == "production" {
-        logger, err = zap.NewProduction()
-    } else {
-        logger, err = zap.NewDevelopment()
-    }
-    if err != nil {
-        panic(err)
-    }
-
-		defer logger.Sync()
-}
 
 func GetLogger() *zap.Logger {
-    return logger
+	once.Do(func() {
+		var err error
+		cfg := config.GetConfig()
+
+		if cfg.Environment == "production" {
+			logger, err = zap.NewProduction()
+		} else {
+			logger, err = zap.NewDevelopment()
+		}
+		if err != nil {
+			panic(err)
+		}
+	})
+	return logger
 }
 
+func Sync() error {
+	return logger.Sync()
+}