fix: index for expires_at

Author: sattvikc

CHANGELOG.md

@@ -50,6 +50,7 @@ CREATE TABLE IF NOT EXISTS saml_relay_state (
 );
 
 CREATE INDEX IF NOT EXISTS saml_relay_state_app_id_tenant_id_index ON saml_relay_state (app_id, tenant_id);
+CREATE INDEX IF NOT EXISTS saml_relay_state_expires_at_index ON saml_relay_state (expires_at);
 
 CREATE TABLE IF NOT EXISTS saml_claims (
     app_id VARCHAR(64) NOT NULL DEFAULT 'public',
@@ -64,6 +65,7 @@ CREATE TABLE IF NOT EXISTS saml_claims (
 );
 
 CREATE INDEX IF NOT EXISTS saml_claims_app_id_tenant_id_index ON saml_claims (app_id, tenant_id);
+CREATE INDEX IF NOT EXISTS saml_claims_expires_at_index ON saml_claims (expires_at);
 ```
 
 ## [9.2.0]

src/main/java/io/supertokens/storage/postgresql/queries/GeneralQueries.java

@@ -711,6 +711,7 @@ public static void createTablesIfNotExists(Start start, Connection con) throws S
 
                     // indexes
                     update(con, SAMLQueries.getQueryToCreateSAMLRelayStateAppIdTenantIdIndex(start), NO_OP_SETTER);
+                    update(con, SAMLQueries.getQueryToCreateSAMLRelayStateExpiresAtIndex(start), NO_OP_SETTER);
                 }
 
                 if (!doesTableExists(start, con, Config.getConfig(start).getSAMLClaimsTable())) {
@@ -719,6 +720,7 @@ public static void createTablesIfNotExists(Start start, Connection con) throws S
 
                     // indexes
                     update(con, SAMLQueries.getQueryToCreateSAMLClaimsAppIdTenantIdIndex(start), NO_OP_SETTER);
+                    update(con, SAMLQueries.getQueryToCreateSAMLClaimsExpiresAtIndex(start), NO_OP_SETTER);
                 }
 
             } catch (Exception e) {

src/main/java/io/supertokens/storage/postgresql/queries/SAMLQueries.java

@@ -100,6 +100,11 @@ public static String getQueryToCreateSAMLRelayStateAppIdTenantIdIndex(Start star
                 + Config.getConfig(start).getSAMLRelayStateTable() + " (app_id, tenant_id)";
     }
 
+    public static String getQueryToCreateSAMLRelayStateExpiresAtIndex(Start start) {
+        return "CREATE INDEX IF NOT EXISTS saml_relay_state_expires_at_index ON "
+                + Config.getConfig(start).getSAMLRelayStateTable() + " (expires_at)";
+    }
+
     public static String getQueryToCreateSAMLClaimsTable(Start start) {
         String schema = Config.getConfig(start).getTableSchema();
         String tableName = Config.getConfig(start).getSAMLClaimsTable();
@@ -126,6 +131,11 @@ public static String getQueryToCreateSAMLClaimsAppIdTenantIdIndex(Start start) {
                 + Config.getConfig(start).getSAMLClaimsTable() + " (app_id, tenant_id)";
     }
 
+    public static String getQueryToCreateSAMLClaimsExpiresAtIndex(Start start) {
+        return "CREATE INDEX IF NOT EXISTS saml_claims_expires_at_index ON "
+                + Config.getConfig(start).getSAMLClaimsTable() + " (expires_at)";
+    }
+
     public static void saveRelayStateInfo(Start start, TenantIdentifier tenantIdentifier,
                                           String relayState, String clientId, String state, String redirectURI)
             throws StorageQueryException, SQLException {