diff --git a/.gitignore b/.gitignore index 5623649..f6246e3 100644 --- a/.gitignore +++ b/.gitignore @@ -3,5 +3,4 @@ .vscode/ bin/ coverage.out -go-loose diff --git a/cmd/go-loose/main.go b/cmd/go-loose/main.go new file mode 100644 index 0000000..4d2b695 --- /dev/null +++ b/cmd/go-loose/main.go @@ -0,0 +1,77 @@ +package main + +import ( + "context" + "errors" + "io" + "log/slog" + "net/http" + "os" + "os/signal" + "syscall" + "time" + + "github.com/danyel/go-loose/internal/config" + "github.com/danyel/go-loose/internal/database" + "github.com/danyel/go-loose/internal/server" +) + +func main() { + if len(os.Args) == 2 && os.Args[1] == "-healthcheck" { + runHealthcheck() + return + } + logger := slog.New(slog.NewJSONHandler(os.Stdout, nil)) + cfg, err := config.Load() + if err != nil { + logger.Error("invalid configuration", "error", err) + os.Exit(1) + } + ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM) + defer stop() + db, err := database.Open(ctx, cfg.DatabaseURL) + if err != nil { + logger.Error("database connection failed", "error", err) + os.Exit(1) + } + defer db.Close() + if err := database.Migrate(ctx, db); err != nil { + logger.Error("database migration failed", "error", err) + os.Exit(1) + } + app, err := server.New(ctx, cfg, db, logger) + if err != nil { + logger.Error("server initialization failed", "error", err) + os.Exit(1) + } + httpServer := &http.Server{ + Addr: cfg.Addr, Handler: app.Handler(), ReadHeaderTimeout: 5 * time.Second, + ReadTimeout: 15 * time.Second, WriteTimeout: 30 * time.Second, IdleTimeout: 90 * time.Second, + } + go func() { + logger.Info("Go Loose listening", "address", cfg.Addr) + if err := httpServer.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) { + logger.Error("HTTP server failed", "error", err) + stop() + } + }() + <-ctx.Done() + shutdownContext, cancel := context.WithTimeout(context.Background(), 10*time.Second) + defer cancel() + if err := httpServer.Shutdown(shutdownContext); err != nil { + logger.Error("graceful shutdown failed", "error", err) + } +} + +func runHealthcheck() { + client := &http.Client{Timeout: 2 * time.Second} + response, err := client.Get("http://127.0.0.1:8080/healthz") + if err != nil { + os.Exit(1) + } + defer response.Body.Close() + _, _ = io.Copy(io.Discard, response.Body) + if response.StatusCode != http.StatusOK { + os.Exit(1) + } +}