Restructure project

internal/logging/logger_test.go

@@ -0,0 +1,44 @@
+package logging
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestInfoLogger(t *testing.T) {
+	level := "info"
+	log := CreateLogger(level)
+	defer log.Sync()
+
+	assert.NotEmpty(t, log, "Logger should not be nil")
+	assert.Equal(t, log.Level().String(), level, fmt.Sprintf("Level should be %s", level))
+}
+
+func TestWarnLogger(t *testing.T) {
+	level := "warn"
+	log := CreateLogger(level)
+	defer log.Sync()
+
+	assert.NotEmpty(t, log, "Logger should not be nil")
+	assert.Equal(t, log.Level().String(), level, fmt.Sprintf("Level should be %s", level))
+}
+
+func TestDebugLogger(t *testing.T) {
+	level := "debug"
+	log := CreateLogger(level)
+	defer log.Sync()
+
+	assert.NotEmpty(t, log, "Logger should not be nil")
+	assert.Equal(t, log.Level().String(), level, fmt.Sprintf("Level should be %s", level))
+}
+
+func TestInvalidLogger(t *testing.T) {
+	level := "invalid"
+	log := CreateLogger(level)
+	defer log.Sync()
+
+	assert.NotEmpty(t, log, "Logger should not be nil")
+	assert.Equal(t, log.Level().String(), "info", "Level should be info")
+}

internal/logging/logging.go

@@ -0,0 +1,33 @@
+package logging
+
+import (
+	"time"
+
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+)
+
+func CreateLogger(logLevel string) *zap.Logger {
+	encoderCfg := zap.NewProductionEncoderConfig()
+	encoderCfg.TimeKey = "time"
+	encoderCfg.EncodeTime = zapcore.TimeEncoderOfLayout(time.StampMilli)
+
+	level := zap.NewAtomicLevelAt(zap.InfoLevel)
+	zapLevel, err := zap.ParseAtomicLevel(logLevel)
+	if err == nil {
+		level = zapLevel
+	}
+
+	config := zap.Config{
+		Level:             level,
+		Development:       false,
+		DisableCaller:     false,
+		DisableStacktrace: false,
+		Sampling:          nil,
+		Encoding:          "json",
+		EncoderConfig:     encoderCfg,
+		OutputPaths:       []string{"stdout"},
+		ErrorOutputPaths:  []string{"stderr"},
+	}
+	return zap.Must(config.Build())
+}