From 9ad7b0e67c90697893f188323a32590fbecd5a65 Mon Sep 17 00:00:00 2001 From: Anhgelus Morhtuuzh Date: Mon, 22 Dec 2025 18:51:42 +0100 Subject: fix(storage): does not store stats if request fails --- backend/storage/db.go | 6 ++++++ backend/storage/stats.go | 7 ------- 2 files changed, 6 insertions(+), 7 deletions(-) (limited to 'backend/storage') diff --git a/backend/storage/db.go b/backend/storage/db.go index 7a7c6e0..0ee50da 100644 --- a/backend/storage/db.go +++ b/backend/storage/db.go @@ -13,6 +13,8 @@ import ( _ "github.com/mattn/go-sqlite3" ) +const DBKey = "db" + //go:embed migrations var migrations embed.FS @@ -66,3 +68,7 @@ func RunMigration(ctx context.Context, db *sql.DB) error { } return nil } + +func getDB(ctx context.Context) *sql.DB { + return ctx.Value(DBKey).(*sql.DB) +} diff --git a/backend/storage/stats.go b/backend/storage/stats.go index 757168d..d95387d 100644 --- a/backend/storage/stats.go +++ b/backend/storage/stats.go @@ -2,7 +2,6 @@ package storage import ( "context" - "database/sql" "fmt" "log/slog" "net/http" @@ -13,8 +12,6 @@ import ( "time" ) -const DBKey = "db" - type loaded struct { data map[string]struct{} mu *sync.RWMutex @@ -50,10 +47,6 @@ var trimRefererReg = regexp.MustCompile(`https?://([a-z-0-9.]+(:\d+)?)/.*`) var load = newLoaded() -func getDB(ctx context.Context) *sql.DB { - return ctx.Value(DBKey).(*sql.DB) -} - func UpdateStats(ctx context.Context, r *http.Request, domain string) error { target := r.URL.Path if strings.HasPrefix(target, "/static") || strings.HasPrefix(target, "/admin") { -- cgit v1.2.3