Warm up monthly samples caches on startup

Author: pierres

internal/database/cache.go

@@ -85,6 +85,11 @@ func (c *MonthlySamplesCache) load(ctx context.Context) (map[int]int, error) {
 	return cache, nil
 }
 
+func (c *MonthlySamplesCache) Warmup(ctx context.Context) error {
+	_, err := c.load(ctx)
+	return err
+}
+
 func startOfNextMonth() time.Time {
 	now := time.Now()
 	return time.Date(now.Year(), now.Month()+1, 1, 0, 0, 0, 0, now.Location())

internal/packages/repository.go

@@ -32,6 +32,10 @@ func NewSQLiteRepository(db *sql.DB) *SQLiteRepository {
 	}
 }
 
+func (r *SQLiteRepository) WarmupCache(ctx context.Context) error {
+	return r.monthlyMaxCache.Warmup(ctx)
+}
+
 // monthRange returns the SQL WHERE fragment and bound args for a month range.
 // When startMonth is 0, no lower bound is applied (all history up to endMonth).
 func monthRange(startMonth, endMonth int) (clause string, args []any) {

internal/popularity/repository.go

@@ -206,6 +206,10 @@ func (r *Repository[T, L]) FindSeries(ctx context.Context, identifier string, st
 	return &list, nil
 }
 
+func (r *Repository[T, L]) WarmupCache(ctx context.Context) error {
+	return r.samplesCache.Warmup(ctx)
+}
+
 func (r *Repository[T, L]) queryPattern(query string) string {
 	if r.cfg.QueryContains {
 		return "%" + query + "%"

main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"log/slog"
 	"net/http"
 	"os"
@@ -85,6 +86,16 @@ func run(cfg config.Config) error {
 		return err
 	}
 
+	// Warm up caches
+	ctx := context.Background()
+	for _, repo := range []interface{ WarmupCache(context.Context) error }{
+		packagesRepo, countriesRepo, mirrorsRepo, systemArchRepo, osRepo,
+	} {
+		if err := repo.WarmupCache(ctx); err != nil {
+			slog.Warn("failed to warm up cache", "error", err)
+		}
+	}
+
 	// Setup HTTP routes
 	mux := http.NewServeMux()