@mastra/pg 1.11.0 → 1.11.1-alpha.0

package/dist/index.js

@@ -11733,18 +11733,22 @@ function rowToTrigger(row) {
   if (metadata !== void 0) trigger.metadata = metadata;
   return trigger;
 }
-var SchedulesPG = class extends SchedulesStorage {
+var SchedulesPG = class _SchedulesPG extends SchedulesStorage {
   #db;
   #client;
   #schema;
+  #skipDefaultIndexes;
+  #indexes;
   /** Tables managed by this domain */
   static MANAGED_TABLES = [TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS];
   constructor(config) {
     super();
-    const { client, schemaName, skipDefaultIndexes } = resolvePgConfig(config);
+    const { client, schemaName, skipDefaultIndexes, indexes } = resolvePgConfig(config);
     this.#client = client;
     this.#db = new PgDB({ client, schemaName, skipDefaultIndexes });
     this.#schema = schemaName || "public";
+    this.#skipDefaultIndexes = skipDefaultIndexes;
+    this.#indexes = indexes?.filter((idx) => _SchedulesPG.MANAGED_TABLES.includes(idx.table));
   }
   async init() {
     await this.#db.createTable({
@@ -11755,22 +11759,79 @@ var SchedulesPG = class extends SchedulesStorage {
       tableName: TABLE_SCHEDULE_TRIGGERS,
       schema: TABLE_SCHEMAS[TABLE_SCHEDULE_TRIGGERS]
     });
+    await this.createDefaultIndexes();
+    await this.createCustomIndexes();
   }
-  static getExportDDL(schemaName) {
+  /**
+   * Returns default index definitions for the schedules domain.
+   * @param schemaPrefix - Prefix for index names (e.g. "my_schema_" or "")
+   */
+  static getDefaultIndexDefs(schemaPrefix) {
     return [
+      {
+        name: `${schemaPrefix}idx_mastra_schedules_status_next_fire`,
+        table: TABLE_SCHEDULES,
+        columns: ["status", "next_fire_at"]
+      },
+      {
+        name: `${schemaPrefix}idx_mastra_schedule_triggers_schedule_fire`,
+        table: TABLE_SCHEDULE_TRIGGERS,
+        columns: ["schedule_id", "actual_fire_at DESC"]
+      }
+    ];
+  }
+  getDefaultIndexDefinitions() {
+    const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+    return _SchedulesPG.getDefaultIndexDefs(schemaPrefix);
+  }
+  async createDefaultIndexes() {
+    if (this.#skipDefaultIndexes) {
+      return;
+    }
+    for (const indexDef of this.getDefaultIndexDefinitions()) {
+      try {
+        await this.#db.createIndex(indexDef);
+      } catch (error) {
+        this.logger?.warn?.(`Failed to create index ${indexDef.name}:`, error);
+      }
+    }
+  }
+  async createCustomIndexes() {
+    if (!this.#indexes || this.#indexes.length === 0) {
+      return;
+    }
+    for (const indexDef of this.#indexes) {
+      try {
+        await this.#db.createIndex(indexDef);
+      } catch (error) {
+        this.logger?.warn?.(`Failed to create custom index ${indexDef.name}:`, error);
+      }
+    }
+  }
+  static getExportDDL(schemaName) {
+    const statements = [];
+    const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+    const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
+    statements.push(
       generateTableSQL({
         tableName: TABLE_SCHEDULES,
         schema: TABLE_SCHEMAS[TABLE_SCHEDULES],
         schemaName,
         includeAllConstraints: true
-      }),
+      })
+    );
+    statements.push(
       generateTableSQL({
         tableName: TABLE_SCHEDULE_TRIGGERS,
         schema: TABLE_SCHEMAS[TABLE_SCHEDULE_TRIGGERS],
         schemaName,
         includeAllConstraints: true
       })
-    ];
+    );
+    for (const idx of _SchedulesPG.getDefaultIndexDefs(schemaPrefix)) {
+      statements.push(generateIndexSQL(idx, schemaName));
+    }
+    return statements;
   }
   async dangerouslyClearAll() {
     await this.#db.clearTable({ tableName: TABLE_SCHEDULE_TRIGGERS });