fixup! Data model and repository for user registration tokens

Author: sandhose

crates/storage-pg/migrations/20250602212100_user_registration_tokens.sql

@@ -29,8 +29,29 @@ CREATE TABLE "user_registration_tokens" (
   "revoked_at" TIMESTAMP WITH TIME ZONE
 );
 
+-- Create a few indices on the table, as we use those for filtering
+-- They are safe to create non-concurrently, as the table is empty at this point
+CREATE INDEX "user_registration_tokens_usage_limit_idx"
+  ON "user_registration_tokens" ("usage_limit");
+  
+CREATE INDEX "user_registration_tokens_times_used_idx"
+  ON "user_registration_tokens" ("times_used");
+
+CREATE INDEX "user_registration_tokens_created_at_idx"
+  ON "user_registration_tokens" ("created_at");
+
+CREATE INDEX "user_registration_tokens_last_used_at_idx"
+  ON "user_registration_tokens" ("last_used_at");
+
+CREATE INDEX "user_registration_tokens_expires_at_idx"
+  ON "user_registration_tokens" ("expires_at");
+
+CREATE INDEX "user_registration_tokens_revoked_at_idx"
+  ON "user_registration_tokens" ("revoked_at");
+
 -- Add foreign key reference to registration tokens in user registrations
+-- A second migration will add the index for this foreign key
 ALTER TABLE "user_registrations"
   ADD COLUMN "user_registration_token_id" UUID
     REFERENCES "user_registration_tokens" ("user_registration_token_id")
-    ON DELETE SET NULL;
+    ON DELETE SET NULL;

crates/storage-pg/migrations/20250602212101_idx_user_registration_token.sql

@@ -0,0 +1,9 @@
+-- no-transaction
+-- Copyright 2025 New Vector Ltd.
+--
+-- SPDX-License-Identifier: AGPL-3.0-only
+-- Please see LICENSE in the repository root for full details.
+
+CREATE INDEX CONCURRENTLY
+  user_registrations_user_registration_token_id_fk
+  ON user_registrations (user_registration_token_id);