Move cron migrations

Author: DJAndries

datastore/datastore.go

@@ -2,7 +2,6 @@ package datastore
 
 import (
 	"context"
-	"embed"
 	"errors"
 	"fmt"
 	"os"
@@ -71,14 +70,7 @@ func NewDatastore(minSessionVersion int, isKeyService bool, isTesting bool) (*Da
 		}
 	}
 
-	var files embed.FS
-	if isTesting || isKeyService {
-		files = migrations.MigrationFiles
-	} else {
-		files = migrations.MigrationFilesWithExtension
-	}
-
-	iofsDriver, err := iofs.New(files, ".")
+	iofsDriver, err := iofs.New(migrations.MigrationFiles, ".")
 	if err != nil {
 		return nil, fmt.Errorf("failed to load iofs driver for migrations: %w", err)
 	}

docker-compose.yml

@@ -15,6 +15,7 @@ services:
       - POSTGRES_PASSWORD=password
     volumes:
       - ./misc/create_additional_dbs.sql:/docker-entrypoint-initdb.d/create_additional_dbs.sql
+      - ./misc/init_pg_cron.sh:/docker-entrypoint-initdb.d/init_pg_cron.sh
   ses-local:
     build:
       context: ./misc/ses-v2-local

migrations/11_cron.down.sql

@@ -0,0 +1,11 @@
+DO $$
+BEGIN
+    IF EXISTS (SELECT 1 FROM pg_namespace WHERE nspname = 'cron') THEN
+        PERFORM cron.unschedule('remove-old-verifications');
+        PERFORM cron.unschedule('remove-old-interim-password-states');
+        PERFORM cron.unschedule('clean-job-history');
+        PERFORM cron.unschedule('remove-old-totp-used-codes');
+        PERFORM cron.unschedule('delete-unverified-accounts');
+    END IF;
+END;
+$$;

migrations/11_cron.up.sql

@@ -0,0 +1,11 @@
+DO $$
+BEGIN
+    IF EXISTS (SELECT 1 FROM pg_namespace WHERE nspname = 'cron') THEN
+        PERFORM cron.schedule('remove-old-verifications', '0 0 * * *', $q$DELETE FROM verifications WHERE created_at < CURRENT_TIMESTAMP - interval '24 hours'$q$);
+        PERFORM cron.schedule('remove-old-interim-password-states', '0 0 * * *', $q$DELETE FROM interim_password_states WHERE created_at < CURRENT_TIMESTAMP - interval '1 hours'$q$);
+        PERFORM cron.schedule('clean-job-history', '0 0 * * *', $q$DELETE FROM cron.job_run_details WHERE end_time < now() - interval '7 days'$q$);
+        PERFORM cron.schedule('remove-old-totp-used-codes', '0 0 * * *', $q$DELETE FROM totp_used_codes WHERE created_at < CURRENT_TIMESTAMP - interval '1 hours'$q$);
+        PERFORM cron.schedule('delete-unverified-accounts', '0,15,30,45 * * * *', $q$DELETE FROM accounts WHERE last_email_verified_at IS NULL AND created_at < CURRENT_TIMESTAMP - interval '30 minutes'$q$);
+    END IF;
+END;
+$$;

migrations/20260515221717_cron.down.sql

@@ -3,7 +3,4 @@ package migrations
 import "embed"
 
 //go:embed *
-var MigrationFilesWithExtension embed.FS
-
-//go:embed 20*
 var MigrationFiles embed.FS

migrations/20260515221717_cron.up.sql

@@ -0,0 +1,6 @@
+#!/bin/bash
+set -e
+
+psql -v ON_ERROR_STOP=1 --username "$POSTGRES_USER" --dbname "postgres" <<-EOSQL
+    CREATE EXTENSION IF NOT EXISTS pg_cron;
+EOSQL