@dbos-inc/dbos-sdk 5.1.3-preview → 5.1.5-preview

package/dist/src/system_database.js

@@ -2560,7 +2560,7 @@ class SystemDatabase {
         return rows[0].workflow_uuid;
     }
     async getQueuePartitions(queueName) {
-        // Recursive-CTE loose index scan: SELECT DISTINCT would scan every ENQUEUED row, whereas each iteration here is one seek on idx_workflow_status_partition_dequeue_v2, so cost scales with the number of partitions rather than the backlog depth.
+        // Recursive-CTE loose index scan: SELECT DISTINCT would scan every ENQUEUED row, whereas each iteration here is one seek on idx_workflow_status_partition_dequeue_v3, so cost scales with the number of partitions rather than the backlog depth.
         const params = [queueName, workflow_1.StatusString.ENQUEUED];
         // Only partitions this application can actually dequeue from.
         const scope = this.#appNameFilter('application_name', this.appName, params);
@@ -2615,7 +2615,7 @@ class SystemDatabase {
             };
             /**
              * Workflows already running, which peer workers count against too. Kept as its own query per
-             * scope: the partition-scoped predicate rides idx_workflow_status_partition_dequeue_v2, which
+             * scope: the partition-scoped predicate rides idx_workflow_status_partition_dequeue_v3, which
              * a queue-wide scan loses.
              */
             const pendingCount = async (partitionScoped) => {
@@ -3066,8 +3066,18 @@ class SystemDatabase {
         addFilter('executor_id', input.executorId);
         const whereClause = whereClauses.length > 0 ? `WHERE ${whereClauses.join(' AND ')}` : '';
         const orderClause = `ORDER BY created_at ${input.sortDesc ? 'DESC' : 'ASC'}`;
-        const limitClause = input.limit ? `LIMIT ${input.limit}` : '';
-        const offsetClause = input.offset ? `OFFSET ${input.offset}` : '';
+        let limitClause = '';
+        if (input.limit) {
+            limitClause = `LIMIT $${paramCounter}`;
+            params.push(input.limit);
+            paramCounter++;
+        }
+        let offsetClause = '';
+        if (input.offset) {
+            offsetClause = `OFFSET $${paramCounter}`;
+            params.push(input.offset);
+            paramCounter++;
+        }
         const projection = [...selectColumns.map((c) => `workflow_status.${c}`), ...payloadColumns].join(', ');
         const query = `
       SELECT ${projection}
@@ -3106,9 +3116,12 @@ class SystemDatabase {
                 groupSelectColumns.push(col);
             }
         }
+        const params = [];
         if (input.timeBucketSizeMs !== undefined) {
             // Bucket on created_at — the indexed wall-clock timestamp on workflow_status.
-            const bucket = input.timeBucketSizeMs;
+            // One placeholder shared by SELECT and GROUP BY, so Postgres sees the two expressions as identical.
+            params.push(input.timeBucketSizeMs);
+            const bucket = `$${params.length}::bigint`;
             const bucketExpr = `(CAST(FLOOR(created_at / ${bucket}) AS BIGINT) * ${bucket})`;
             groupNames.push('time_bucket');
             groupColumns.push(bucketExpr);
@@ -3138,8 +3151,7 @@ class SystemDatabase {
             throw new Error('At least one select_ flag must be set to True');
         }
         const whereClauses = [];
-        const params = [];
-        let paramIdx = 1;
+        let paramIdx = params.length + 1;
         const addFilter = (column, values) => {
             if (!values || values.length === 0)
                 return;
@@ -3261,10 +3273,12 @@ class SystemDatabase {
                 groupSelectColumns.push(`${expr} AS ${colName}`);
             }
         }
+        const params = [];
         if (input.timeBucketSizeMs !== undefined) {
             // Bucket on completed_at_epoch_ms — it's the indexed timestamp on
             // this table.
-            const bucket = input.timeBucketSizeMs;
+            params.push(input.timeBucketSizeMs);
+            const bucket = `$${params.length}::bigint`;
             const bucketExpr = `(CAST(FLOOR(completed_at_epoch_ms / ${bucket}) AS BIGINT) * ${bucket})`;
             groupNames.push('time_bucket');
             groupColumns.push(bucketExpr);
@@ -3292,8 +3306,7 @@ class SystemDatabase {
             throw new Error('At least one select_ flag must be set to True');
         }
         const whereClauses = [];
-        const params = [];
-        let paramIdx = 1;
+        let paramIdx = params.length + 1;
         if (input.status && input.status.length > 0) {
             const placeholders = input.status.map((_, i) => `$${paramIdx + i}`).join(', ');
             whereClauses.push(`(${statusExpr}) IN (${placeholders})`);