main.go

package main

import (
	"context"
	"errors"
	"log/slog"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/go-chi/chi/v5"
	"github.com/jackc/pgx/v5/pgxpool"
)

func main() {
	logger := slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{
		Level: slog.LevelInfo,
	}))

	dsn := getEnv("DATABASE_URL", "postgres://admin:secret@localhost:5432/todo")
	addr := getEnv("ADDR", ":8080")

	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
	defer stop()

	pool, err := pgxpool.New(ctx, dsn)
	if err != nil {
		logger.Error("не удалось создать пул соединений", slog.String("error", err.Error()))
		os.Exit(1)
	}
	defer pool.Close()

	if err := pool.Ping(ctx); err != nil {
		logger.Error("не удалось выполнить ping базы данных", slog.String("error", err.Error()))
		os.Exit(1)
	}
	logger.Info("подключен к базе данных")

	storage := NewStorage(pool)
	handler := NewHandler(storage)

	r := chi.NewRouter()

	r.Use(RecoverMiddleware(logger))
	r.Use(RequestIDMiddleware)
	r.Use(LoggerMiddleware(logger))

	r.Post("/tasks", handler.handlerCreateTask)
	r.Get("/tasks/{id}", handler.handlerPrintTask)
	r.Put("/tasks/{id}", handler.handlerUpdateTask)
	r.Get("/tasks", handler.handlerPrintAllTasks)
	r.Delete("/tasks/{id}", handler.handlerDeleteTask)

	srv := &http.Server{
		Addr:         addr,
		Handler:      r,
		ReadTimeout:  15 * time.Second,
		WriteTimeout: 15 * time.Second,
		IdleTimeout:  60 * time.Second,
	}

	go func() {
		logger.Info("сервер стартанул", slog.String("addr", addr))
		if err := srv.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
			logger.Error("ошибка сервера", slog.String("error", err.Error()))
			os.Exit(1)
		}
	}()

	<-ctx.Done()
	logger.Info("получен сигнал отключения")
	
	shutdownCtx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()

	if err := srv.Shutdown(shutdownCtx); err != nil {
		logger.Error("ошибка завершения работы", slog.String("error", err.Error()))
	}

	logger.Info("сервер завершил работу в нормальном режиме")
}

func getEnv(key, defaultVal string) string {
	if val := os.Getenv(key); val != "" {
		return val
	}
	return defaultVal
}

func NewHandler(storage TaskStorage) *Handler {
	return &Handler{storage: storage}
}