@query-doctor/core 0.1.8 → 0.1.11

package/dist/index.cjs

@@ -189,6 +189,23 @@ function parseNudges(node, stack) {
       });
     }
   }
+  if (is(node, "A_Expr")) {
+    if (node.A_Expr.kind === "AEXPR_IN") {
+      let list;
+      if (node.A_Expr.lexpr && is(node.A_Expr.lexpr, "List")) {
+        list = node.A_Expr.lexpr.List;
+      } else if (node.A_Expr.rexpr && is(node.A_Expr.rexpr, "List")) {
+        list = node.A_Expr.rexpr.List;
+      }
+      if (list?.items && list.items.length >= 10) {
+        nudges.push({
+          kind: "REPLACE_LARGE_IN_TUPLE_WITH_ANY_ARRAY",
+          message: "`in (...)` queries with large tuples can often be replaced with `= ANY($1)` using a single parameter",
+          severity: "INFO"
+        });
+      }
+    }
+  }
   return nudges;
 }
 function containsColumnRef(args) {
@@ -1165,6 +1182,24 @@ var _IndexOptimizer = class _IndexOptimizer {
       await this.dropExistingIndexes(tx);
     });
   }
+  /**
+   * Given the current indexes in the optimizer, transform them in some
+   * way to change which indexes will be assumed to exist when optimizing
+   *
+   * @example
+   * ```
+   * // resets indexes
+   * optimizer.transformIndexes(() => [])
+   *
+   * // adds new index
+   * optimizer.transformIndexes(indexes => [...indexes, newIndex])
+   * ```
+   */
+  transformIndexes(f) {
+    const newIndexes = f(this.existingIndexes);
+    this.existingIndexes = newIndexes;
+    return this;
+  }
   /**
    * Postgres has a limit of 63 characters for index names.
    * So we use this to make sure we don't derive it from a list of columns that can
@@ -1520,6 +1555,16 @@ var _Statistics = class _Statistics {
   restoreStats(tx) {
     return this.restoreStats17(tx);
   }
+  approximateTotalRows() {
+    if (!this.exportedMetadata) {
+      return 0;
+    }
+    let totalRows = 0;
+    for (const table of this.exportedMetadata) {
+      totalRows += table.reltuples;
+    }
+    return totalRows;
+  }
   /**
    * We have to cast stavaluesN to the correct type
    * This derives that type for us so it can be used in `array_in`