add missing migration's ctx

umputun · Dec 31, 2024 · c43c651 · c43c651
1 parent 7f58038
commit c43c651
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/app/storage/approved_users.go b/app/storage/approved_users.go
@@ -47,7 +47,7 @@ func NewApprovedUsers(ctx context.Context, db *Engine) (*ApprovedUsers, error) {
 	}
 
 	var exists int
-	err := db.Get(&exists, "SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name='approved_users'")
+	err := db.GetContext(ctx, &exists, "SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name='approved_users'")
 	if err != nil {
 		return nil, fmt.Errorf("failed to check for approved_users table existence: %w", err)
 	}

diff --git a/app/storage/detected_spam.go b/app/storage/detected_spam.go
@@ -57,7 +57,7 @@ func NewDetectedSpam(ctx context.Context, db *Engine) (*DetectedSpam, error) {
 	// first check if the table exists. we can't do this in a transaction
 	// because missing columns will cause the transaction to fail
 	var exists int
-	err := db.Get(&exists, "SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name='detected_spam'")
+	err := db.GetContext(ctx, &exists, "SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name='detected_spam'")
 	if err != nil {
 		return nil, fmt.Errorf("failed to check for detected_spam table existence: %w", err)
 	}