Added a custom logger for dev and production.

cmd/main.go

@@ -2,8 +2,8 @@ package main
 
 import (
 	"context"
-	"fmt"
 	"log"
+	"log/slog"
 	"net/http"
 	"os"
 	"os/signal"
@@ -17,10 +17,21 @@ import (
 )
 
 func main() {
+
+	// @Logger
+	// ------------------------------------------------------------
+	var handler slog.Handler
+	if os.Getenv("ENV") == "production" {
+		handler = slog.NewJSONHandler(os.Stdout, nil)
+	} else {
+		handler = slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{Level: slog.LevelDebug})
+	}
+	logger := slog.New(handler)
+
 	ctx, ctxCancel := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
 	defer ctxCancel()
 
-	// DB init
+	// @DB
 	// ------------------------------------------------------------
 	connStr := os.Getenv("DATABASE_URL")
 	if connStr == "" {
@@ -46,23 +57,23 @@ func main() {
 	if err := pool.Ping(ctx); err != nil {
 		log.Fatalf("Unable to reach database: %v", err)
 	}
-	fmt.Println("Database connection established")
+	logger.Info("database connection established", "max_conns", poolConfig.MaxConns, "min_conns", poolConfig.MinConns)
 
 	defer func() {
 		pool.Close()
-		fmt.Println("Database pool closed")
+		logger.Info("database pool closed")
 	}()
 
 	// Create Store sqlc wrapper
 	store := db.NewStore(pool)
 
-	// Server Init
+	// @Server
 	// ------------------------------------------------------------
 	jwtSecret := os.Getenv("JWT_SECRET")
 	if jwtSecret == "" {
 		log.Fatal("JWT_SECRET environment variable is required")
 	}
-	server := web.NewServer(store, jwtSecret)
+	server := web.NewServer(store, logger, jwtSecret)
 
 	// We run it in a goroutine so it doesn't block main from reaching the signal listener.
 	go func() {
@@ -70,17 +81,16 @@ func main() {
 		if connStr == "" {
 			log.Fatal("SERVER_PORT environment variable is not set")
 		}
-		fmt.Printf("Server starting on %s\n", addr)
 
 		if err := server.Start(addr); err != nil && err != http.ErrServerClosed {
 			log.Fatalf("Server failed: %v", err)
 		}
 	}()
 
-	// Graceful Shutdown
+	// @GracefulShutdown
 	// ------------------------------------------------------------
 	<-ctx.Done()
-	fmt.Println("\nShutdown signal received. Starting graceful exit...")
+	logger.Warn("shutdown signal received", "signal", "SIGTERM/Interrupt")
 
 	shutdownCtx, shutdownCtxCancel := context.WithTimeout(context.Background(), 10*time.Second)
 	defer shutdownCtxCancel()
@@ -89,5 +99,5 @@ func main() {
 		log.Fatalf("Graceful shutdown failed: %v", err)
 	}
 
-	fmt.Println("Server exited properly")
+	logger.Info("Server shutdown successful")
 }