fix(storage): does not store stats if request fails

2 files changed, 6 insertions, 7 deletions

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") {