@tanstack/db 0.0.23 → 0.0.24
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/cjs/proxy.cjs +21 -0
- package/dist/cjs/proxy.cjs.map +1 -1
- package/dist/cjs/query/builder/index.cjs +72 -0
- package/dist/cjs/query/builder/index.cjs.map +1 -1
- package/dist/cjs/query/builder/index.d.cts +64 -0
- package/dist/cjs/query/compiler/index.cjs +44 -8
- package/dist/cjs/query/compiler/index.cjs.map +1 -1
- package/dist/cjs/query/compiler/index.d.cts +4 -7
- package/dist/cjs/query/compiler/joins.cjs +14 -6
- package/dist/cjs/query/compiler/joins.cjs.map +1 -1
- package/dist/cjs/query/compiler/joins.d.cts +4 -8
- package/dist/cjs/query/compiler/types.d.cts +10 -0
- package/dist/cjs/query/optimizer.cjs +283 -0
- package/dist/cjs/query/optimizer.cjs.map +1 -0
- package/dist/cjs/query/optimizer.d.cts +42 -0
- package/dist/cjs/utils.cjs +42 -0
- package/dist/cjs/utils.cjs.map +1 -0
- package/dist/cjs/utils.d.cts +18 -0
- package/dist/esm/proxy.js +21 -0
- package/dist/esm/proxy.js.map +1 -1
- package/dist/esm/query/builder/index.d.ts +64 -0
- package/dist/esm/query/builder/index.js +72 -0
- package/dist/esm/query/builder/index.js.map +1 -1
- package/dist/esm/query/compiler/index.d.ts +4 -7
- package/dist/esm/query/compiler/index.js +44 -8
- package/dist/esm/query/compiler/index.js.map +1 -1
- package/dist/esm/query/compiler/joins.d.ts +4 -8
- package/dist/esm/query/compiler/joins.js +14 -6
- package/dist/esm/query/compiler/joins.js.map +1 -1
- package/dist/esm/query/compiler/types.d.ts +10 -0
- package/dist/esm/query/optimizer.d.ts +42 -0
- package/dist/esm/query/optimizer.js +283 -0
- package/dist/esm/query/optimizer.js.map +1 -0
- package/dist/esm/utils.d.ts +18 -0
- package/dist/esm/utils.js +42 -0
- package/dist/esm/utils.js.map +1 -0
- package/package.json +1 -1
- package/src/proxy.ts +24 -0
- package/src/query/builder/index.ts +104 -0
- package/src/query/compiler/index.ts +85 -18
- package/src/query/compiler/joins.ts +21 -13
- package/src/query/compiler/types.ts +12 -0
- package/src/query/optimizer.ts +738 -0
- package/src/utils.ts +86 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"optimizer.js","sources":["../../../src/query/optimizer.ts"],"sourcesContent":["/**\n * # Query Optimizer\n *\n * The query optimizer improves query performance by implementing predicate pushdown optimization.\n * It rewrites the intermediate representation (IR) to push WHERE clauses as close to the data\n * source as possible, reducing the amount of data processed during joins.\n *\n * ## How It Works\n *\n * The optimizer follows a 4-step process:\n *\n * ### 1. AND Clause Splitting\n * Splits AND clauses at the root level into separate WHERE clauses for granular optimization.\n * ```javascript\n * // Before: WHERE and(eq(users.department_id, 1), gt(users.age, 25))\n * // After: WHERE eq(users.department_id, 1) + WHERE gt(users.age, 25)\n * ```\n *\n * ### 2. Source Analysis\n * Analyzes each WHERE clause to determine which table sources it references:\n * - Single-source clauses: Touch only one table (e.g., `users.department_id = 1`)\n * - Multi-source clauses: Touch multiple tables (e.g., `users.id = posts.user_id`)\n *\n * ### 3. Clause Grouping\n * Groups WHERE clauses by the sources they touch:\n * - Single-source clauses are grouped by their respective table\n * - Multi-source clauses are combined for the main query\n *\n * ### 4. Subquery Creation\n * Lifts single-source WHERE clauses into subqueries that wrap the original table references.\n *\n * ## Safety & Edge Cases\n *\n * The optimizer includes targeted safety checks to prevent predicate pushdown when it could\n * break query semantics:\n *\n * ### Always Safe Operations\n * - **Creating new subqueries**: Wrapping collection references in subqueries with WHERE clauses\n * - **Main query optimizations**: Moving single-source WHERE clauses from main query to subqueries\n * - **Queries with aggregates/ORDER BY/HAVING**: Can still create new filtered subqueries\n *\n * ### Unsafe Operations (blocked by safety checks)\n * Pushing WHERE clauses **into existing subqueries** that have:\n * - **Aggregates**: GROUP BY, HAVING, or aggregate functions in SELECT (would change aggregation)\n * - **Ordering + Limits**: ORDER BY combined with LIMIT/OFFSET (would change result set)\n * - **Functional Operations**: fnSelect, fnWhere, fnHaving (potential side effects)\n *\n * The optimizer tracks which clauses were actually optimized and only removes those from the\n * main query. Subquery reuse is handled safely through immutable query copies.\n *\n * ## Example Optimizations\n *\n * ### Basic Query with Joins\n * **Original Query:**\n * ```javascript\n * query\n * .from({ users: usersCollection })\n * .join({ posts: postsCollection }, ({users, posts}) => eq(users.id, posts.user_id))\n * .where(({users}) => eq(users.department_id, 1))\n * .where(({posts}) => gt(posts.views, 100))\n * .where(({users, posts}) => eq(users.id, posts.author_id))\n * ```\n *\n * **Optimized Query:**\n * ```javascript\n * query\n * .from({\n * users: subquery\n * .from({ users: usersCollection })\n * .where(({users}) => eq(users.department_id, 1))\n * })\n * .join({\n * posts: subquery\n * .from({ posts: postsCollection })\n * .where(({posts}) => gt(posts.views, 100))\n * }, ({users, posts}) => eq(users.id, posts.user_id))\n * .where(({users, posts}) => eq(users.id, posts.author_id))\n * ```\n *\n * ### Query with Aggregates (Now Optimizable!)\n * **Original Query:**\n * ```javascript\n * query\n * .from({ users: usersCollection })\n * .join({ posts: postsCollection }, ({users, posts}) => eq(users.id, posts.user_id))\n * .where(({users}) => eq(users.department_id, 1))\n * .groupBy(['users.department_id'])\n * .select({ count: agg('count', '*') })\n * ```\n *\n * **Optimized Query:**\n * ```javascript\n * query\n * .from({\n * users: subquery\n * .from({ users: usersCollection })\n * .where(({users}) => eq(users.department_id, 1))\n * })\n * .join({ posts: postsCollection }, ({users, posts}) => eq(users.id, posts.user_id))\n * .groupBy(['users.department_id'])\n * .select({ count: agg('count', '*') })\n * ```\n *\n * ## Benefits\n *\n * - **Reduced Data Processing**: Filters applied before joins reduce intermediate result size\n * - **Better Performance**: Smaller datasets lead to faster query execution\n * - **Automatic Optimization**: No manual query rewriting required\n * - **Preserves Semantics**: Optimized queries return identical results\n * - **Safe by Design**: Comprehensive checks prevent semantic-breaking optimizations\n *\n * ## Integration\n *\n * The optimizer is automatically called during query compilation before the IR is\n * transformed into a D2Mini pipeline.\n */\n\nimport { deepEquals } from \"../utils.js\"\nimport {\n CollectionRef as CollectionRefClass,\n Func,\n QueryRef as QueryRefClass,\n} from \"./ir.js\"\nimport type { BasicExpression, From, QueryIR } from \"./ir.js\"\n\n/**\n * Represents a WHERE clause after source analysis\n */\nexport interface AnalyzedWhereClause {\n /** The WHERE expression */\n expression: BasicExpression<boolean>\n /** Set of table/source aliases that this WHERE clause touches */\n touchedSources: Set<string>\n}\n\n/**\n * Represents WHERE clauses grouped by the sources they touch\n */\nexport interface GroupedWhereClauses {\n /** WHERE clauses that touch only a single source, grouped by source alias */\n singleSource: Map<string, BasicExpression<boolean>>\n /** WHERE clauses that touch multiple sources, combined into one expression */\n multiSource?: BasicExpression<boolean>\n}\n\n/**\n * Main query optimizer entry point that lifts WHERE clauses into subqueries.\n *\n * This function implements multi-level predicate pushdown optimization by recursively\n * moving WHERE clauses through nested subqueries to get them as close to the data\n * sources as possible, then removing redundant subqueries.\n *\n * @param query - The QueryIR to optimize\n * @returns A new QueryIR with optimizations applied (or original if no optimization possible)\n *\n * @example\n * ```typescript\n * const originalQuery = {\n * from: new CollectionRef(users, 'u'),\n * join: [{ from: new CollectionRef(posts, 'p'), ... }],\n * where: [eq(u.dept_id, 1), gt(p.views, 100)]\n * }\n *\n * const optimized = optimizeQuery(originalQuery)\n * // Result: Single-source clauses moved to deepest possible subqueries\n * ```\n */\nexport function optimizeQuery(query: QueryIR): QueryIR {\n // Apply multi-level predicate pushdown with iterative convergence\n let optimized = query\n let previousOptimized: QueryIR | undefined\n let iterations = 0\n const maxIterations = 10 // Prevent infinite loops\n\n // Keep optimizing until no more changes occur or max iterations reached\n while (\n iterations < maxIterations &&\n !deepEquals(optimized, previousOptimized)\n ) {\n previousOptimized = optimized\n optimized = applyRecursiveOptimization(optimized)\n iterations++\n }\n\n // Remove redundant subqueries\n const cleaned = removeRedundantSubqueries(optimized)\n\n return cleaned\n}\n\n/**\n * Applies recursive predicate pushdown optimization.\n *\n * @param query - The QueryIR to optimize\n * @returns A new QueryIR with optimizations applied\n */\nfunction applyRecursiveOptimization(query: QueryIR): QueryIR {\n // First, recursively optimize any existing subqueries\n const subqueriesOptimized = {\n ...query,\n from:\n query.from.type === `queryRef`\n ? new QueryRefClass(\n applyRecursiveOptimization(query.from.query),\n query.from.alias\n )\n : query.from,\n join: query.join?.map((joinClause) => ({\n ...joinClause,\n from:\n joinClause.from.type === `queryRef`\n ? new QueryRefClass(\n applyRecursiveOptimization(joinClause.from.query),\n joinClause.from.alias\n )\n : joinClause.from,\n })),\n }\n\n // Then apply single-level optimization to this query\n return applySingleLevelOptimization(subqueriesOptimized)\n}\n\n/**\n * Applies single-level predicate pushdown optimization (existing logic)\n */\nfunction applySingleLevelOptimization(query: QueryIR): QueryIR {\n // Skip optimization if no WHERE clauses exist\n if (!query.where || query.where.length === 0) {\n return query\n }\n\n // Skip optimization if there are no joins - predicate pushdown only benefits joins\n // Single-table queries don't benefit from this optimization\n if (!query.join || query.join.length === 0) {\n return query\n }\n\n // Step 1: Split all AND clauses at the root level for granular optimization\n const splitWhereClauses = splitAndClauses(query.where)\n\n // Step 2: Analyze each WHERE clause to determine which sources it touches\n const analyzedClauses = splitWhereClauses.map((clause) =>\n analyzeWhereClause(clause)\n )\n\n // Step 3: Group clauses by single-source vs multi-source\n const groupedClauses = groupWhereClauses(analyzedClauses)\n\n // Step 4: Apply optimizations by lifting single-source clauses into subqueries\n return applyOptimizations(query, groupedClauses)\n}\n\n/**\n * Removes redundant subqueries that don't add value.\n * A subquery is redundant if it only wraps another query without adding\n * WHERE, SELECT, GROUP BY, HAVING, ORDER BY, or LIMIT/OFFSET clauses.\n *\n * @param query - The QueryIR to process\n * @returns A new QueryIR with redundant subqueries removed\n */\nfunction removeRedundantSubqueries(query: QueryIR): QueryIR {\n return {\n ...query,\n from: removeRedundantFromClause(query.from),\n join: query.join?.map((joinClause) => ({\n ...joinClause,\n from: removeRedundantFromClause(joinClause.from),\n })),\n }\n}\n\n/**\n * Removes redundant subqueries from a FROM clause.\n *\n * @param from - The FROM clause to process\n * @returns A FROM clause with redundant subqueries removed\n */\nfunction removeRedundantFromClause(from: From): From {\n if (from.type === `collectionRef`) {\n return from\n }\n\n const processedQuery = removeRedundantSubqueries(from.query)\n\n // Check if this subquery is redundant\n if (isRedundantSubquery(processedQuery)) {\n // Return the inner query's FROM clause with this alias\n const innerFrom = removeRedundantFromClause(processedQuery.from)\n if (innerFrom.type === `collectionRef`) {\n return new CollectionRefClass(innerFrom.collection, from.alias)\n } else {\n return new QueryRefClass(innerFrom.query, from.alias)\n }\n }\n\n return new QueryRefClass(processedQuery, from.alias)\n}\n\n/**\n * Determines if a subquery is redundant (adds no value).\n *\n * @param query - The query to check\n * @returns True if the query is redundant and can be removed\n */\nfunction isRedundantSubquery(query: QueryIR): boolean {\n return (\n (!query.where || query.where.length === 0) &&\n !query.select &&\n (!query.groupBy || query.groupBy.length === 0) &&\n (!query.having || query.having.length === 0) &&\n (!query.orderBy || query.orderBy.length === 0) &&\n (!query.join || query.join.length === 0) &&\n query.limit === undefined &&\n query.offset === undefined &&\n !query.fnSelect &&\n (!query.fnWhere || query.fnWhere.length === 0) &&\n (!query.fnHaving || query.fnHaving.length === 0)\n )\n}\n\n/**\n * Step 1: Split all AND clauses recursively into separate WHERE clauses.\n *\n * This enables more granular optimization by treating each condition independently.\n * OR clauses are preserved as they cannot be split without changing query semantics.\n *\n * @param whereClauses - Array of WHERE expressions to split\n * @returns Flattened array with AND clauses split into separate expressions\n *\n * @example\n * ```typescript\n * // Input: [and(eq(a, 1), gt(b, 2)), eq(c, 3)]\n * // Output: [eq(a, 1), gt(b, 2), eq(c, 3)]\n * ```\n */\nfunction splitAndClauses(\n whereClauses: Array<BasicExpression<boolean>>\n): Array<BasicExpression<boolean>> {\n const result: Array<BasicExpression<boolean>> = []\n\n for (const clause of whereClauses) {\n if (clause.type === `func` && clause.name === `and`) {\n // Recursively split nested AND clauses to handle complex expressions\n const splitArgs = splitAndClauses(\n clause.args as Array<BasicExpression<boolean>>\n )\n result.push(...splitArgs)\n } else {\n // Preserve non-AND clauses as-is (including OR clauses)\n result.push(clause)\n }\n }\n\n return result\n}\n\n/**\n * Step 2: Analyze which table sources a WHERE clause touches.\n *\n * This determines whether a clause can be pushed down to a specific table\n * or must remain in the main query (for multi-source clauses like join conditions).\n *\n * @param clause - The WHERE expression to analyze\n * @returns Analysis result with the expression and touched source aliases\n *\n * @example\n * ```typescript\n * // eq(users.department_id, 1) -> touches ['users']\n * // eq(users.id, posts.user_id) -> touches ['users', 'posts']\n * ```\n */\nfunction analyzeWhereClause(\n clause: BasicExpression<boolean>\n): AnalyzedWhereClause {\n const touchedSources = new Set<string>()\n\n /**\n * Recursively collect all table aliases referenced in an expression\n */\n function collectSources(expr: BasicExpression | any): void {\n switch (expr.type) {\n case `ref`:\n // PropRef path has the table alias as the first element\n if (expr.path && expr.path.length > 0) {\n const firstElement = expr.path[0]\n if (firstElement) {\n touchedSources.add(firstElement)\n }\n }\n break\n case `func`:\n // Recursively analyze function arguments (e.g., eq, gt, and, or)\n if (expr.args) {\n expr.args.forEach(collectSources)\n }\n break\n case `val`:\n // Values don't reference any sources\n break\n case `agg`:\n // Aggregates can reference sources in their arguments\n if (expr.args) {\n expr.args.forEach(collectSources)\n }\n break\n }\n }\n\n collectSources(clause)\n\n return {\n expression: clause,\n touchedSources,\n }\n}\n\n/**\n * Step 3: Group WHERE clauses by the sources they touch.\n *\n * Single-source clauses can be pushed down to subqueries for optimization.\n * Multi-source clauses must remain in the main query to preserve join semantics.\n *\n * @param analyzedClauses - Array of analyzed WHERE clauses\n * @returns Grouped clauses ready for optimization\n */\nfunction groupWhereClauses(\n analyzedClauses: Array<AnalyzedWhereClause>\n): GroupedWhereClauses {\n const singleSource = new Map<string, Array<BasicExpression<boolean>>>()\n const multiSource: Array<BasicExpression<boolean>> = []\n\n // Categorize each clause based on how many sources it touches\n for (const clause of analyzedClauses) {\n if (clause.touchedSources.size === 1) {\n // Single source clause - can be optimized\n const source = Array.from(clause.touchedSources)[0]!\n if (!singleSource.has(source)) {\n singleSource.set(source, [])\n }\n singleSource.get(source)!.push(clause.expression)\n } else if (clause.touchedSources.size > 1) {\n // Multi-source clause - must stay in main query\n multiSource.push(clause.expression)\n }\n // Skip clauses that touch no sources (constants) - they don't need optimization\n }\n\n // Combine multiple clauses for each source with AND\n const combinedSingleSource = new Map<string, BasicExpression<boolean>>()\n for (const [source, clauses] of singleSource) {\n combinedSingleSource.set(source, combineWithAnd(clauses))\n }\n\n // Combine multi-source clauses with AND\n const combinedMultiSource =\n multiSource.length > 0 ? combineWithAnd(multiSource) : undefined\n\n return {\n singleSource: combinedSingleSource,\n multiSource: combinedMultiSource,\n }\n}\n\n/**\n * Step 4: Apply optimizations by lifting single-source clauses into subqueries.\n *\n * Creates a new QueryIR with single-source WHERE clauses moved to subqueries\n * that wrap the original table references. This ensures immutability and prevents\n * infinite recursion issues.\n *\n * @param query - Original QueryIR to optimize\n * @param groupedClauses - WHERE clauses grouped by optimization strategy\n * @returns New QueryIR with optimizations applied\n */\nfunction applyOptimizations(\n query: QueryIR,\n groupedClauses: GroupedWhereClauses\n): QueryIR {\n // Track which single-source clauses were actually optimized\n const actuallyOptimized = new Set<string>()\n\n // Optimize the main FROM clause and track what was optimized\n const optimizedFrom = optimizeFromWithTracking(\n query.from,\n groupedClauses.singleSource,\n actuallyOptimized\n )\n\n // Optimize JOIN clauses and track what was optimized\n const optimizedJoins = query.join\n ? query.join.map((joinClause) => ({\n ...joinClause,\n from: optimizeFromWithTracking(\n joinClause.from,\n groupedClauses.singleSource,\n actuallyOptimized\n ),\n }))\n : undefined\n\n // Build the remaining WHERE clauses: multi-source + any single-source that weren't optimized\n const remainingWhereClauses: Array<BasicExpression<boolean>> = []\n\n // Add multi-source clauses\n if (groupedClauses.multiSource) {\n remainingWhereClauses.push(groupedClauses.multiSource)\n }\n\n // Add single-source clauses that weren't actually optimized\n for (const [source, clause] of groupedClauses.singleSource) {\n if (!actuallyOptimized.has(source)) {\n remainingWhereClauses.push(clause)\n }\n }\n\n // Create a completely new query object to ensure immutability\n const optimizedQuery: QueryIR = {\n // Copy all non-optimized fields as-is\n select: query.select,\n groupBy: query.groupBy ? [...query.groupBy] : undefined,\n having: query.having ? [...query.having] : undefined,\n orderBy: query.orderBy ? [...query.orderBy] : undefined,\n limit: query.limit,\n offset: query.offset,\n fnSelect: query.fnSelect,\n fnWhere: query.fnWhere ? [...query.fnWhere] : undefined,\n fnHaving: query.fnHaving ? [...query.fnHaving] : undefined,\n\n // Use the optimized FROM and JOIN clauses\n from: optimizedFrom,\n join: optimizedJoins,\n\n // Only include WHERE clauses that weren't successfully optimized\n where: remainingWhereClauses.length > 0 ? remainingWhereClauses : [],\n }\n\n return optimizedQuery\n}\n\n/**\n * Helper function to create a deep copy of a QueryIR object for immutability.\n *\n * This ensures that all optimizations create new objects rather than modifying\n * existing ones, preventing infinite recursion and shared reference issues.\n *\n * @param query - QueryIR to deep copy\n * @returns New QueryIR object with all nested objects copied\n */\nfunction deepCopyQuery(query: QueryIR): QueryIR {\n return {\n // Recursively copy the FROM clause\n from:\n query.from.type === `collectionRef`\n ? new CollectionRefClass(query.from.collection, query.from.alias)\n : new QueryRefClass(deepCopyQuery(query.from.query), query.from.alias),\n\n // Copy all other fields, creating new arrays where necessary\n select: query.select,\n join: query.join\n ? query.join.map((joinClause) => ({\n type: joinClause.type,\n left: joinClause.left,\n right: joinClause.right,\n from:\n joinClause.from.type === `collectionRef`\n ? new CollectionRefClass(\n joinClause.from.collection,\n joinClause.from.alias\n )\n : new QueryRefClass(\n deepCopyQuery(joinClause.from.query),\n joinClause.from.alias\n ),\n }))\n : undefined,\n where: query.where ? [...query.where] : undefined,\n groupBy: query.groupBy ? [...query.groupBy] : undefined,\n having: query.having ? [...query.having] : undefined,\n orderBy: query.orderBy ? [...query.orderBy] : undefined,\n limit: query.limit,\n offset: query.offset,\n fnSelect: query.fnSelect,\n fnWhere: query.fnWhere ? [...query.fnWhere] : undefined,\n fnHaving: query.fnHaving ? [...query.fnHaving] : undefined,\n }\n}\n\n/**\n * Helper function to optimize a FROM clause while tracking what was actually optimized.\n *\n * @param from - FROM clause to optimize\n * @param singleSourceClauses - Map of source aliases to their WHERE clauses\n * @param actuallyOptimized - Set to track which sources were actually optimized\n * @returns New FROM clause, potentially wrapped in a subquery\n */\nfunction optimizeFromWithTracking(\n from: From,\n singleSourceClauses: Map<string, BasicExpression<boolean>>,\n actuallyOptimized: Set<string>\n): From {\n const whereClause = singleSourceClauses.get(from.alias)\n\n if (!whereClause) {\n // No optimization needed, but return a copy to maintain immutability\n if (from.type === `collectionRef`) {\n return new CollectionRefClass(from.collection, from.alias)\n }\n // Must be queryRef due to type system\n return new QueryRefClass(deepCopyQuery(from.query), from.alias)\n }\n\n if (from.type === `collectionRef`) {\n // Create a new subquery with the WHERE clause for the collection\n // This is always safe since we're creating a new subquery\n const subQuery: QueryIR = {\n from: new CollectionRefClass(from.collection, from.alias),\n where: [whereClause],\n }\n actuallyOptimized.add(from.alias) // Mark as successfully optimized\n return new QueryRefClass(subQuery, from.alias)\n }\n\n // Must be queryRef due to type system\n\n // SAFETY CHECK: Only check safety when pushing WHERE clauses into existing subqueries\n // We need to be careful about pushing WHERE clauses into subqueries that already have\n // aggregates, HAVING, or ORDER BY + LIMIT since that could change their semantics\n if (!isSafeToPushIntoExistingSubquery(from.query)) {\n // Return a copy without optimization to maintain immutability\n // Do NOT mark as optimized since we didn't actually optimize it\n return new QueryRefClass(deepCopyQuery(from.query), from.alias)\n }\n\n // Add the WHERE clause to the existing subquery\n // Create a deep copy to ensure immutability\n const existingWhere = from.query.where || []\n const optimizedSubQuery: QueryIR = {\n ...deepCopyQuery(from.query),\n where: [...existingWhere, whereClause],\n }\n actuallyOptimized.add(from.alias) // Mark as successfully optimized\n return new QueryRefClass(optimizedSubQuery, from.alias)\n}\n\n/**\n * Determines if it's safe to push WHERE clauses into an existing subquery.\n *\n * Pushing WHERE clauses into existing subqueries can break semantics in several cases:\n *\n * 1. **Aggregates**: Pushing predicates before GROUP BY changes what gets aggregated\n * 2. **ORDER BY + LIMIT/OFFSET**: Pushing predicates before sorting+limiting changes the result set\n * 3. **HAVING clauses**: These operate on aggregated data, predicates should not be pushed past them\n * 4. **Functional operations**: fnSelect, fnWhere, fnHaving could have side effects\n *\n * Note: This safety check only applies when pushing WHERE clauses into existing subqueries.\n * Creating new subqueries from collection references is always safe.\n *\n * @param query - The existing subquery to check for safety\n * @returns True if it's safe to push WHERE clauses into this subquery, false otherwise\n *\n * @example\n * ```typescript\n * // UNSAFE: has GROUP BY - pushing WHERE could change aggregation\n * { from: users, groupBy: [dept], select: { count: agg('count', '*') } }\n *\n * // UNSAFE: has ORDER BY + LIMIT - pushing WHERE could change \"top 10\"\n * { from: users, orderBy: [salary desc], limit: 10 }\n *\n * // SAFE: plain SELECT without aggregates/limits\n * { from: users, select: { id, name } }\n * ```\n */\nfunction isSafeToPushIntoExistingSubquery(query: QueryIR): boolean {\n // Check for aggregates in SELECT clause\n if (query.select) {\n const hasAggregates = Object.values(query.select).some(\n (expr) => expr.type === `agg`\n )\n if (hasAggregates) {\n return false\n }\n }\n\n // Check for GROUP BY clause\n if (query.groupBy && query.groupBy.length > 0) {\n return false\n }\n\n // Check for HAVING clause\n if (query.having && query.having.length > 0) {\n return false\n }\n\n // Check for ORDER BY with LIMIT or OFFSET (dangerous combination)\n if (query.orderBy && query.orderBy.length > 0) {\n if (query.limit !== undefined || query.offset !== undefined) {\n return false\n }\n }\n\n // Check for functional variants that might have side effects\n if (\n query.fnSelect ||\n (query.fnWhere && query.fnWhere.length > 0) ||\n (query.fnHaving && query.fnHaving.length > 0)\n ) {\n return false\n }\n\n // If none of the unsafe conditions are present, it's safe to optimize\n return true\n}\n\n/**\n * Helper function to combine multiple expressions with AND.\n *\n * If there's only one expression, it's returned as-is.\n * If there are multiple expressions, they're combined with an AND function.\n *\n * @param expressions - Array of expressions to combine\n * @returns Single expression representing the AND combination\n * @throws Error if the expressions array is empty\n */\nfunction combineWithAnd(\n expressions: Array<BasicExpression<boolean>>\n): BasicExpression<boolean> {\n if (expressions.length === 0) {\n throw new Error(`Cannot combine empty expression list`)\n }\n\n if (expressions.length === 1) {\n return expressions[0]!\n }\n\n // Create an AND function with all expressions as arguments\n return new Func(`and`, expressions)\n}\n"],"names":["QueryRefClass","CollectionRefClass"],"mappings":";;AAuKO,SAAS,cAAc,OAAyB;AAErD,MAAI,YAAY;AAChB,MAAI;AACJ,MAAI,aAAa;AACjB,QAAM,gBAAgB;AAGtB,SACE,aAAa,iBACb,CAAC,WAAW,WAAW,iBAAiB,GACxC;AACA,wBAAoB;AACpB,gBAAY,2BAA2B,SAAS;AAChD;AAAA,EACF;AAGA,QAAM,UAAU,0BAA0B,SAAS;AAEnD,SAAO;AACT;AAQA,SAAS,2BAA2B,OAAyB;;AAE3D,QAAM,sBAAsB;AAAA,IAC1B,GAAG;AAAA,IACH,MACE,MAAM,KAAK,SAAS,aAChB,IAAIA;AAAAA,MACF,2BAA2B,MAAM,KAAK,KAAK;AAAA,MAC3C,MAAM,KAAK;AAAA,IAAA,IAEb,MAAM;AAAA,IACZ,OAAM,WAAM,SAAN,mBAAY,IAAI,CAAC,gBAAgB;AAAA,MACrC,GAAG;AAAA,MACH,MACE,WAAW,KAAK,SAAS,aACrB,IAAIA;AAAAA,QACF,2BAA2B,WAAW,KAAK,KAAK;AAAA,QAChD,WAAW,KAAK;AAAA,MAAA,IAElB,WAAW;AAAA,IAAA;AAAA,EACjB;AAIJ,SAAO,6BAA6B,mBAAmB;AACzD;AAKA,SAAS,6BAA6B,OAAyB;AAE7D,MAAI,CAAC,MAAM,SAAS,MAAM,MAAM,WAAW,GAAG;AAC5C,WAAO;AAAA,EACT;AAIA,MAAI,CAAC,MAAM,QAAQ,MAAM,KAAK,WAAW,GAAG;AAC1C,WAAO;AAAA,EACT;AAGA,QAAM,oBAAoB,gBAAgB,MAAM,KAAK;AAGrD,QAAM,kBAAkB,kBAAkB;AAAA,IAAI,CAAC,WAC7C,mBAAmB,MAAM;AAAA,EAAA;AAI3B,QAAM,iBAAiB,kBAAkB,eAAe;AAGxD,SAAO,mBAAmB,OAAO,cAAc;AACjD;AAUA,SAAS,0BAA0B,OAAyB;;AAC1D,SAAO;AAAA,IACL,GAAG;AAAA,IACH,MAAM,0BAA0B,MAAM,IAAI;AAAA,IAC1C,OAAM,WAAM,SAAN,mBAAY,IAAI,CAAC,gBAAgB;AAAA,MACrC,GAAG;AAAA,MACH,MAAM,0BAA0B,WAAW,IAAI;AAAA,IAAA;AAAA,EAC/C;AAEN;AAQA,SAAS,0BAA0B,MAAkB;AACnD,MAAI,KAAK,SAAS,iBAAiB;AACjC,WAAO;AAAA,EACT;AAEA,QAAM,iBAAiB,0BAA0B,KAAK,KAAK;AAG3D,MAAI,oBAAoB,cAAc,GAAG;AAEvC,UAAM,YAAY,0BAA0B,eAAe,IAAI;AAC/D,QAAI,UAAU,SAAS,iBAAiB;AACtC,aAAO,IAAIC,cAAmB,UAAU,YAAY,KAAK,KAAK;AAAA,IAChE,OAAO;AACL,aAAO,IAAID,SAAc,UAAU,OAAO,KAAK,KAAK;AAAA,IACtD;AAAA,EACF;AAEA,SAAO,IAAIA,SAAc,gBAAgB,KAAK,KAAK;AACrD;AAQA,SAAS,oBAAoB,OAAyB;AACpD,UACG,CAAC,MAAM,SAAS,MAAM,MAAM,WAAW,MACxC,CAAC,MAAM,WACN,CAAC,MAAM,WAAW,MAAM,QAAQ,WAAW,OAC3C,CAAC,MAAM,UAAU,MAAM,OAAO,WAAW,OACzC,CAAC,MAAM,WAAW,MAAM,QAAQ,WAAW,OAC3C,CAAC,MAAM,QAAQ,MAAM,KAAK,WAAW,MACtC,MAAM,UAAU,UAChB,MAAM,WAAW,UACjB,CAAC,MAAM,aACN,CAAC,MAAM,WAAW,MAAM,QAAQ,WAAW,OAC3C,CAAC,MAAM,YAAY,MAAM,SAAS,WAAW;AAElD;AAiBA,SAAS,gBACP,cACiC;AACjC,QAAM,SAA0C,CAAA;AAEhD,aAAW,UAAU,cAAc;AACjC,QAAI,OAAO,SAAS,UAAU,OAAO,SAAS,OAAO;AAEnD,YAAM,YAAY;AAAA,QAChB,OAAO;AAAA,MAAA;AAET,aAAO,KAAK,GAAG,SAAS;AAAA,IAC1B,OAAO;AAEL,aAAO,KAAK,MAAM;AAAA,IACpB;AAAA,EACF;AAEA,SAAO;AACT;AAiBA,SAAS,mBACP,QACqB;AACrB,QAAM,qCAAqB,IAAA;AAK3B,WAAS,eAAe,MAAmC;AACzD,YAAQ,KAAK,MAAA;AAAA,MACX,KAAK;AAEH,YAAI,KAAK,QAAQ,KAAK,KAAK,SAAS,GAAG;AACrC,gBAAM,eAAe,KAAK,KAAK,CAAC;AAChC,cAAI,cAAc;AAChB,2BAAe,IAAI,YAAY;AAAA,UACjC;AAAA,QACF;AACA;AAAA,MACF,KAAK;AAEH,YAAI,KAAK,MAAM;AACb,eAAK,KAAK,QAAQ,cAAc;AAAA,QAClC;AACA;AAAA,MACF,KAAK;AAEH;AAAA,MACF,KAAK;AAEH,YAAI,KAAK,MAAM;AACb,eAAK,KAAK,QAAQ,cAAc;AAAA,QAClC;AACA;AAAA,IAAA;AAAA,EAEN;AAEA,iBAAe,MAAM;AAErB,SAAO;AAAA,IACL,YAAY;AAAA,IACZ;AAAA,EAAA;AAEJ;AAWA,SAAS,kBACP,iBACqB;AACrB,QAAM,mCAAmB,IAAA;AACzB,QAAM,cAA+C,CAAA;AAGrD,aAAW,UAAU,iBAAiB;AACpC,QAAI,OAAO,eAAe,SAAS,GAAG;AAEpC,YAAM,SAAS,MAAM,KAAK,OAAO,cAAc,EAAE,CAAC;AAClD,UAAI,CAAC,aAAa,IAAI,MAAM,GAAG;AAC7B,qBAAa,IAAI,QAAQ,EAAE;AAAA,MAC7B;AACA,mBAAa,IAAI,MAAM,EAAG,KAAK,OAAO,UAAU;AAAA,IAClD,WAAW,OAAO,eAAe,OAAO,GAAG;AAEzC,kBAAY,KAAK,OAAO,UAAU;AAAA,IACpC;AAAA,EAEF;AAGA,QAAM,2CAA2B,IAAA;AACjC,aAAW,CAAC,QAAQ,OAAO,KAAK,cAAc;AAC5C,yBAAqB,IAAI,QAAQ,eAAe,OAAO,CAAC;AAAA,EAC1D;AAGA,QAAM,sBACJ,YAAY,SAAS,IAAI,eAAe,WAAW,IAAI;AAEzD,SAAO;AAAA,IACL,cAAc;AAAA,IACd,aAAa;AAAA,EAAA;AAEjB;AAaA,SAAS,mBACP,OACA,gBACS;AAET,QAAM,wCAAwB,IAAA;AAG9B,QAAM,gBAAgB;AAAA,IACpB,MAAM;AAAA,IACN,eAAe;AAAA,IACf;AAAA,EAAA;AAIF,QAAM,iBAAiB,MAAM,OACzB,MAAM,KAAK,IAAI,CAAC,gBAAgB;AAAA,IAC9B,GAAG;AAAA,IACH,MAAM;AAAA,MACJ,WAAW;AAAA,MACX,eAAe;AAAA,MACf;AAAA,IAAA;AAAA,EACF,EACA,IACF;AAGJ,QAAM,wBAAyD,CAAA;AAG/D,MAAI,eAAe,aAAa;AAC9B,0BAAsB,KAAK,eAAe,WAAW;AAAA,EACvD;AAGA,aAAW,CAAC,QAAQ,MAAM,KAAK,eAAe,cAAc;AAC1D,QAAI,CAAC,kBAAkB,IAAI,MAAM,GAAG;AAClC,4BAAsB,KAAK,MAAM;AAAA,IACnC;AAAA,EACF;AAGA,QAAM,iBAA0B;AAAA;AAAA,IAE9B,QAAQ,MAAM;AAAA,IACd,SAAS,MAAM,UAAU,CAAC,GAAG,MAAM,OAAO,IAAI;AAAA,IAC9C,QAAQ,MAAM,SAAS,CAAC,GAAG,MAAM,MAAM,IAAI;AAAA,IAC3C,SAAS,MAAM,UAAU,CAAC,GAAG,MAAM,OAAO,IAAI;AAAA,IAC9C,OAAO,MAAM;AAAA,IACb,QAAQ,MAAM;AAAA,IACd,UAAU,MAAM;AAAA,IAChB,SAAS,MAAM,UAAU,CAAC,GAAG,MAAM,OAAO,IAAI;AAAA,IAC9C,UAAU,MAAM,WAAW,CAAC,GAAG,MAAM,QAAQ,IAAI;AAAA;AAAA,IAGjD,MAAM;AAAA,IACN,MAAM;AAAA;AAAA,IAGN,OAAO,sBAAsB,SAAS,IAAI,wBAAwB,CAAA;AAAA,EAAC;AAGrE,SAAO;AACT;AAWA,SAAS,cAAc,OAAyB;AAC9C,SAAO;AAAA;AAAA,IAEL,MACE,MAAM,KAAK,SAAS,kBAChB,IAAIC,cAAmB,MAAM,KAAK,YAAY,MAAM,KAAK,KAAK,IAC9D,IAAID,SAAc,cAAc,MAAM,KAAK,KAAK,GAAG,MAAM,KAAK,KAAK;AAAA;AAAA,IAGzE,QAAQ,MAAM;AAAA,IACd,MAAM,MAAM,OACR,MAAM,KAAK,IAAI,CAAC,gBAAgB;AAAA,MAC9B,MAAM,WAAW;AAAA,MACjB,MAAM,WAAW;AAAA,MACjB,OAAO,WAAW;AAAA,MAClB,MACE,WAAW,KAAK,SAAS,kBACrB,IAAIC;AAAAA,QACF,WAAW,KAAK;AAAA,QAChB,WAAW,KAAK;AAAA,MAAA,IAElB,IAAID;AAAAA,QACF,cAAc,WAAW,KAAK,KAAK;AAAA,QACnC,WAAW,KAAK;AAAA,MAAA;AAAA,IAClB,EACN,IACF;AAAA,IACJ,OAAO,MAAM,QAAQ,CAAC,GAAG,MAAM,KAAK,IAAI;AAAA,IACxC,SAAS,MAAM,UAAU,CAAC,GAAG,MAAM,OAAO,IAAI;AAAA,IAC9C,QAAQ,MAAM,SAAS,CAAC,GAAG,MAAM,MAAM,IAAI;AAAA,IAC3C,SAAS,MAAM,UAAU,CAAC,GAAG,MAAM,OAAO,IAAI;AAAA,IAC9C,OAAO,MAAM;AAAA,IACb,QAAQ,MAAM;AAAA,IACd,UAAU,MAAM;AAAA,IAChB,SAAS,MAAM,UAAU,CAAC,GAAG,MAAM,OAAO,IAAI;AAAA,IAC9C,UAAU,MAAM,WAAW,CAAC,GAAG,MAAM,QAAQ,IAAI;AAAA,EAAA;AAErD;AAUA,SAAS,yBACP,MACA,qBACA,mBACM;AACN,QAAM,cAAc,oBAAoB,IAAI,KAAK,KAAK;AAEtD,MAAI,CAAC,aAAa;AAEhB,QAAI,KAAK,SAAS,iBAAiB;AACjC,aAAO,IAAIC,cAAmB,KAAK,YAAY,KAAK,KAAK;AAAA,IAC3D;AAEA,WAAO,IAAID,SAAc,cAAc,KAAK,KAAK,GAAG,KAAK,KAAK;AAAA,EAChE;AAEA,MAAI,KAAK,SAAS,iBAAiB;AAGjC,UAAM,WAAoB;AAAA,MACxB,MAAM,IAAIC,cAAmB,KAAK,YAAY,KAAK,KAAK;AAAA,MACxD,OAAO,CAAC,WAAW;AAAA,IAAA;AAErB,sBAAkB,IAAI,KAAK,KAAK;AAChC,WAAO,IAAID,SAAc,UAAU,KAAK,KAAK;AAAA,EAC/C;AAOA,MAAI,CAAC,iCAAiC,KAAK,KAAK,GAAG;AAGjD,WAAO,IAAIA,SAAc,cAAc,KAAK,KAAK,GAAG,KAAK,KAAK;AAAA,EAChE;AAIA,QAAM,gBAAgB,KAAK,MAAM,SAAS,CAAA;AAC1C,QAAM,oBAA6B;AAAA,IACjC,GAAG,cAAc,KAAK,KAAK;AAAA,IAC3B,OAAO,CAAC,GAAG,eAAe,WAAW;AAAA,EAAA;AAEvC,oBAAkB,IAAI,KAAK,KAAK;AAChC,SAAO,IAAIA,SAAc,mBAAmB,KAAK,KAAK;AACxD;AA8BA,SAAS,iCAAiC,OAAyB;AAEjE,MAAI,MAAM,QAAQ;AAChB,UAAM,gBAAgB,OAAO,OAAO,MAAM,MAAM,EAAE;AAAA,MAChD,CAAC,SAAS,KAAK,SAAS;AAAA,IAAA;AAE1B,QAAI,eAAe;AACjB,aAAO;AAAA,IACT;AAAA,EACF;AAGA,MAAI,MAAM,WAAW,MAAM,QAAQ,SAAS,GAAG;AAC7C,WAAO;AAAA,EACT;AAGA,MAAI,MAAM,UAAU,MAAM,OAAO,SAAS,GAAG;AAC3C,WAAO;AAAA,EACT;AAGA,MAAI,MAAM,WAAW,MAAM,QAAQ,SAAS,GAAG;AAC7C,QAAI,MAAM,UAAU,UAAa,MAAM,WAAW,QAAW;AAC3D,aAAO;AAAA,IACT;AAAA,EACF;AAGA,MACE,MAAM,YACL,MAAM,WAAW,MAAM,QAAQ,SAAS,KACxC,MAAM,YAAY,MAAM,SAAS,SAAS,GAC3C;AACA,WAAO;AAAA,EACT;AAGA,SAAO;AACT;AAYA,SAAS,eACP,aAC0B;AAC1B,MAAI,YAAY,WAAW,GAAG;AAC5B,UAAM,IAAI,MAAM,sCAAsC;AAAA,EACxD;AAEA,MAAI,YAAY,WAAW,GAAG;AAC5B,WAAO,YAAY,CAAC;AAAA,EACtB;AAGA,SAAO,IAAI,KAAK,OAAO,WAAW;AACpC;"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Generic utility functions
|
|
3
|
+
*/
|
|
4
|
+
/**
|
|
5
|
+
* Deep equality function that compares two values recursively
|
|
6
|
+
*
|
|
7
|
+
* @param a - First value to compare
|
|
8
|
+
* @param b - Second value to compare
|
|
9
|
+
* @returns True if the values are deeply equal, false otherwise
|
|
10
|
+
*
|
|
11
|
+
* @example
|
|
12
|
+
* ```typescript
|
|
13
|
+
* deepEquals({ a: 1, b: 2 }, { b: 2, a: 1 }) // true (property order doesn't matter)
|
|
14
|
+
* deepEquals([1, { x: 2 }], [1, { x: 2 }]) // true
|
|
15
|
+
* deepEquals({ a: 1 }, { a: 2 }) // false
|
|
16
|
+
* ```
|
|
17
|
+
*/
|
|
18
|
+
export declare function deepEquals(a: any, b: any): boolean;
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
function deepEquals(a, b) {
|
|
2
|
+
return deepEqualsInternal(a, b, /* @__PURE__ */ new Map());
|
|
3
|
+
}
|
|
4
|
+
function deepEqualsInternal(a, b, visited) {
|
|
5
|
+
if (a === b) return true;
|
|
6
|
+
if (a == null || b == null) return false;
|
|
7
|
+
if (typeof a !== typeof b) return false;
|
|
8
|
+
if (Array.isArray(a)) {
|
|
9
|
+
if (!Array.isArray(b) || a.length !== b.length) return false;
|
|
10
|
+
if (visited.has(a)) {
|
|
11
|
+
return visited.get(a) === b;
|
|
12
|
+
}
|
|
13
|
+
visited.set(a, b);
|
|
14
|
+
const result = a.every(
|
|
15
|
+
(item, index) => deepEqualsInternal(item, b[index], visited)
|
|
16
|
+
);
|
|
17
|
+
visited.delete(a);
|
|
18
|
+
return result;
|
|
19
|
+
}
|
|
20
|
+
if (typeof a === `object`) {
|
|
21
|
+
if (visited.has(a)) {
|
|
22
|
+
return visited.get(a) === b;
|
|
23
|
+
}
|
|
24
|
+
visited.set(a, b);
|
|
25
|
+
const keysA = Object.keys(a);
|
|
26
|
+
const keysB = Object.keys(b);
|
|
27
|
+
if (keysA.length !== keysB.length) {
|
|
28
|
+
visited.delete(a);
|
|
29
|
+
return false;
|
|
30
|
+
}
|
|
31
|
+
const result = keysA.every(
|
|
32
|
+
(key) => key in b && deepEqualsInternal(a[key], b[key], visited)
|
|
33
|
+
);
|
|
34
|
+
visited.delete(a);
|
|
35
|
+
return result;
|
|
36
|
+
}
|
|
37
|
+
return false;
|
|
38
|
+
}
|
|
39
|
+
export {
|
|
40
|
+
deepEquals
|
|
41
|
+
};
|
|
42
|
+
//# sourceMappingURL=utils.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"utils.js","sources":["../../src/utils.ts"],"sourcesContent":["/**\n * Generic utility functions\n */\n\n/**\n * Deep equality function that compares two values recursively\n *\n * @param a - First value to compare\n * @param b - Second value to compare\n * @returns True if the values are deeply equal, false otherwise\n *\n * @example\n * ```typescript\n * deepEquals({ a: 1, b: 2 }, { b: 2, a: 1 }) // true (property order doesn't matter)\n * deepEquals([1, { x: 2 }], [1, { x: 2 }]) // true\n * deepEquals({ a: 1 }, { a: 2 }) // false\n * ```\n */\nexport function deepEquals(a: any, b: any): boolean {\n return deepEqualsInternal(a, b, new Map())\n}\n\n/**\n * Internal implementation with cycle detection to prevent infinite recursion\n */\nfunction deepEqualsInternal(\n a: any,\n b: any,\n visited: Map<object, object>\n): boolean {\n // Handle strict equality (primitives, same reference)\n if (a === b) return true\n\n // Handle null/undefined\n if (a == null || b == null) return false\n\n // Handle different types\n if (typeof a !== typeof b) return false\n\n // Handle arrays\n if (Array.isArray(a)) {\n if (!Array.isArray(b) || a.length !== b.length) return false\n\n // Check for circular references\n if (visited.has(a)) {\n return visited.get(a) === b\n }\n visited.set(a, b)\n\n const result = a.every((item, index) =>\n deepEqualsInternal(item, b[index], visited)\n )\n visited.delete(a)\n return result\n }\n\n // Handle objects\n if (typeof a === `object`) {\n // Check for circular references\n if (visited.has(a)) {\n return visited.get(a) === b\n }\n visited.set(a, b)\n\n // Get all keys from both objects\n const keysA = Object.keys(a)\n const keysB = Object.keys(b)\n\n // Check if they have the same number of keys\n if (keysA.length !== keysB.length) {\n visited.delete(a)\n return false\n }\n\n // Check if all keys exist in both objects and their values are equal\n const result = keysA.every(\n (key) => key in b && deepEqualsInternal(a[key], b[key], visited)\n )\n\n visited.delete(a)\n return result\n }\n\n // For primitives that aren't strictly equal\n return false\n}\n"],"names":[],"mappings":"AAkBO,SAAS,WAAW,GAAQ,GAAiB;AAClD,SAAO,mBAAmB,GAAG,GAAG,oBAAI,KAAK;AAC3C;AAKA,SAAS,mBACP,GACA,GACA,SACS;AAET,MAAI,MAAM,EAAG,QAAO;AAGpB,MAAI,KAAK,QAAQ,KAAK,KAAM,QAAO;AAGnC,MAAI,OAAO,MAAM,OAAO,EAAG,QAAO;AAGlC,MAAI,MAAM,QAAQ,CAAC,GAAG;AACpB,QAAI,CAAC,MAAM,QAAQ,CAAC,KAAK,EAAE,WAAW,EAAE,OAAQ,QAAO;AAGvD,QAAI,QAAQ,IAAI,CAAC,GAAG;AAClB,aAAO,QAAQ,IAAI,CAAC,MAAM;AAAA,IAC5B;AACA,YAAQ,IAAI,GAAG,CAAC;AAEhB,UAAM,SAAS,EAAE;AAAA,MAAM,CAAC,MAAM,UAC5B,mBAAmB,MAAM,EAAE,KAAK,GAAG,OAAO;AAAA,IAAA;AAE5C,YAAQ,OAAO,CAAC;AAChB,WAAO;AAAA,EACT;AAGA,MAAI,OAAO,MAAM,UAAU;AAEzB,QAAI,QAAQ,IAAI,CAAC,GAAG;AAClB,aAAO,QAAQ,IAAI,CAAC,MAAM;AAAA,IAC5B;AACA,YAAQ,IAAI,GAAG,CAAC;AAGhB,UAAM,QAAQ,OAAO,KAAK,CAAC;AAC3B,UAAM,QAAQ,OAAO,KAAK,CAAC;AAG3B,QAAI,MAAM,WAAW,MAAM,QAAQ;AACjC,cAAQ,OAAO,CAAC;AAChB,aAAO;AAAA,IACT;AAGA,UAAM,SAAS,MAAM;AAAA,MACnB,CAAC,QAAQ,OAAO,KAAK,mBAAmB,EAAE,GAAG,GAAG,EAAE,GAAG,GAAG,OAAO;AAAA,IAAA;AAGjE,YAAQ,OAAO,CAAC;AAChB,WAAO;AAAA,EACT;AAGA,SAAO;AACT;"}
|
package/package.json
CHANGED
package/src/proxy.ts
CHANGED
|
@@ -461,6 +461,30 @@ export function createChangeProxy<
|
|
|
461
461
|
|
|
462
462
|
// If the value is a function, bind it to the ptarget
|
|
463
463
|
if (typeof value === `function`) {
|
|
464
|
+
// For Array methods that modify the array
|
|
465
|
+
if (Array.isArray(ptarget)) {
|
|
466
|
+
const methodName = prop.toString()
|
|
467
|
+
const modifyingMethods = new Set([
|
|
468
|
+
`pop`,
|
|
469
|
+
`push`,
|
|
470
|
+
`shift`,
|
|
471
|
+
`unshift`,
|
|
472
|
+
`splice`,
|
|
473
|
+
`sort`,
|
|
474
|
+
`reverse`,
|
|
475
|
+
`fill`,
|
|
476
|
+
`copyWithin`,
|
|
477
|
+
])
|
|
478
|
+
|
|
479
|
+
if (modifyingMethods.has(methodName)) {
|
|
480
|
+
return function (...args: Array<unknown>) {
|
|
481
|
+
const result = value.apply(changeTracker.copy_, args)
|
|
482
|
+
markChanged(changeTracker)
|
|
483
|
+
return result
|
|
484
|
+
}
|
|
485
|
+
}
|
|
486
|
+
}
|
|
487
|
+
|
|
464
488
|
// For Map and Set methods that modify the collection
|
|
465
489
|
if (ptarget instanceof Map || ptarget instanceof Set) {
|
|
466
490
|
const methodName = prop.toString()
|
|
@@ -184,6 +184,110 @@ export class BaseQueryBuilder<TContext extends Context = Context> {
|
|
|
184
184
|
}) as any
|
|
185
185
|
}
|
|
186
186
|
|
|
187
|
+
/**
|
|
188
|
+
* Perform a LEFT JOIN with another table or subquery
|
|
189
|
+
*
|
|
190
|
+
* @param source - An object with a single key-value pair where the key is the table alias and the value is a Collection or subquery
|
|
191
|
+
* @param onCallback - A function that receives table references and returns the join condition
|
|
192
|
+
* @returns A QueryBuilder with the left joined table available
|
|
193
|
+
*
|
|
194
|
+
* @example
|
|
195
|
+
* ```ts
|
|
196
|
+
* // Left join users with posts
|
|
197
|
+
* query
|
|
198
|
+
* .from({ users: usersCollection })
|
|
199
|
+
* .leftJoin({ posts: postsCollection }, ({users, posts}) => eq(users.id, posts.userId))
|
|
200
|
+
* ```
|
|
201
|
+
*/
|
|
202
|
+
leftJoin<TSource extends Source>(
|
|
203
|
+
source: TSource,
|
|
204
|
+
onCallback: JoinOnCallback<
|
|
205
|
+
MergeContext<TContext, SchemaFromSource<TSource>>
|
|
206
|
+
>
|
|
207
|
+
): QueryBuilder<
|
|
208
|
+
MergeContextWithJoinType<TContext, SchemaFromSource<TSource>, `left`>
|
|
209
|
+
> {
|
|
210
|
+
return this.join(source, onCallback, `left`)
|
|
211
|
+
}
|
|
212
|
+
|
|
213
|
+
/**
|
|
214
|
+
* Perform a RIGHT JOIN with another table or subquery
|
|
215
|
+
*
|
|
216
|
+
* @param source - An object with a single key-value pair where the key is the table alias and the value is a Collection or subquery
|
|
217
|
+
* @param onCallback - A function that receives table references and returns the join condition
|
|
218
|
+
* @returns A QueryBuilder with the right joined table available
|
|
219
|
+
*
|
|
220
|
+
* @example
|
|
221
|
+
* ```ts
|
|
222
|
+
* // Right join users with posts
|
|
223
|
+
* query
|
|
224
|
+
* .from({ users: usersCollection })
|
|
225
|
+
* .rightJoin({ posts: postsCollection }, ({users, posts}) => eq(users.id, posts.userId))
|
|
226
|
+
* ```
|
|
227
|
+
*/
|
|
228
|
+
rightJoin<TSource extends Source>(
|
|
229
|
+
source: TSource,
|
|
230
|
+
onCallback: JoinOnCallback<
|
|
231
|
+
MergeContext<TContext, SchemaFromSource<TSource>>
|
|
232
|
+
>
|
|
233
|
+
): QueryBuilder<
|
|
234
|
+
MergeContextWithJoinType<TContext, SchemaFromSource<TSource>, `right`>
|
|
235
|
+
> {
|
|
236
|
+
return this.join(source, onCallback, `right`)
|
|
237
|
+
}
|
|
238
|
+
|
|
239
|
+
/**
|
|
240
|
+
* Perform an INNER JOIN with another table or subquery
|
|
241
|
+
*
|
|
242
|
+
* @param source - An object with a single key-value pair where the key is the table alias and the value is a Collection or subquery
|
|
243
|
+
* @param onCallback - A function that receives table references and returns the join condition
|
|
244
|
+
* @returns A QueryBuilder with the inner joined table available
|
|
245
|
+
*
|
|
246
|
+
* @example
|
|
247
|
+
* ```ts
|
|
248
|
+
* // Inner join users with posts
|
|
249
|
+
* query
|
|
250
|
+
* .from({ users: usersCollection })
|
|
251
|
+
* .innerJoin({ posts: postsCollection }, ({users, posts}) => eq(users.id, posts.userId))
|
|
252
|
+
* ```
|
|
253
|
+
*/
|
|
254
|
+
innerJoin<TSource extends Source>(
|
|
255
|
+
source: TSource,
|
|
256
|
+
onCallback: JoinOnCallback<
|
|
257
|
+
MergeContext<TContext, SchemaFromSource<TSource>>
|
|
258
|
+
>
|
|
259
|
+
): QueryBuilder<
|
|
260
|
+
MergeContextWithJoinType<TContext, SchemaFromSource<TSource>, `inner`>
|
|
261
|
+
> {
|
|
262
|
+
return this.join(source, onCallback, `inner`)
|
|
263
|
+
}
|
|
264
|
+
|
|
265
|
+
/**
|
|
266
|
+
* Perform a FULL JOIN with another table or subquery
|
|
267
|
+
*
|
|
268
|
+
* @param source - An object with a single key-value pair where the key is the table alias and the value is a Collection or subquery
|
|
269
|
+
* @param onCallback - A function that receives table references and returns the join condition
|
|
270
|
+
* @returns A QueryBuilder with the full joined table available
|
|
271
|
+
*
|
|
272
|
+
* @example
|
|
273
|
+
* ```ts
|
|
274
|
+
* // Full join users with posts
|
|
275
|
+
* query
|
|
276
|
+
* .from({ users: usersCollection })
|
|
277
|
+
* .fullJoin({ posts: postsCollection }, ({users, posts}) => eq(users.id, posts.userId))
|
|
278
|
+
* ```
|
|
279
|
+
*/
|
|
280
|
+
fullJoin<TSource extends Source>(
|
|
281
|
+
source: TSource,
|
|
282
|
+
onCallback: JoinOnCallback<
|
|
283
|
+
MergeContext<TContext, SchemaFromSource<TSource>>
|
|
284
|
+
>
|
|
285
|
+
): QueryBuilder<
|
|
286
|
+
MergeContextWithJoinType<TContext, SchemaFromSource<TSource>, `full`>
|
|
287
|
+
> {
|
|
288
|
+
return this.join(source, onCallback, `full`)
|
|
289
|
+
}
|
|
290
|
+
|
|
187
291
|
/**
|
|
188
292
|
* Filter rows based on a condition
|
|
189
293
|
*
|
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import { distinct, filter, map } from "@electric-sql/d2mini"
|
|
2
|
+
import { optimizeQuery } from "../optimizer.js"
|
|
2
3
|
import { compileExpression } from "./evaluators.js"
|
|
3
4
|
import { processJoins } from "./joins.js"
|
|
4
5
|
import { processGroupBy } from "./group-by.js"
|
|
@@ -10,30 +11,35 @@ import type {
|
|
|
10
11
|
NamespacedAndKeyedStream,
|
|
11
12
|
ResultStream,
|
|
12
13
|
} from "../../types.js"
|
|
13
|
-
|
|
14
|
-
/**
|
|
15
|
-
* Cache for compiled subqueries to avoid duplicate compilation
|
|
16
|
-
*/
|
|
17
|
-
type QueryCache = WeakMap<QueryIR, ResultStream>
|
|
14
|
+
import type { QueryCache, QueryMapping } from "./types.js"
|
|
18
15
|
|
|
19
16
|
/**
|
|
20
17
|
* Compiles a query2 IR into a D2 pipeline
|
|
21
|
-
* @param
|
|
18
|
+
* @param rawQuery The query IR to compile
|
|
22
19
|
* @param inputs Mapping of collection names to input streams
|
|
23
20
|
* @param cache Optional cache for compiled subqueries (used internally for recursion)
|
|
21
|
+
* @param queryMapping Optional mapping from optimized queries to original queries
|
|
24
22
|
* @returns A stream builder representing the compiled query
|
|
25
23
|
*/
|
|
26
24
|
export function compileQuery(
|
|
27
|
-
|
|
25
|
+
rawQuery: QueryIR,
|
|
28
26
|
inputs: Record<string, KeyedStream>,
|
|
29
|
-
cache: QueryCache = new WeakMap()
|
|
27
|
+
cache: QueryCache = new WeakMap(),
|
|
28
|
+
queryMapping: QueryMapping = new WeakMap()
|
|
30
29
|
): ResultStream {
|
|
31
|
-
// Check if
|
|
32
|
-
const cachedResult = cache.get(
|
|
30
|
+
// Check if the original raw query has already been compiled
|
|
31
|
+
const cachedResult = cache.get(rawQuery)
|
|
33
32
|
if (cachedResult) {
|
|
34
33
|
return cachedResult
|
|
35
34
|
}
|
|
36
35
|
|
|
36
|
+
// Optimize the query before compilation
|
|
37
|
+
const query = optimizeQuery(rawQuery)
|
|
38
|
+
|
|
39
|
+
// Create mapping from optimized query to original for caching
|
|
40
|
+
queryMapping.set(query, rawQuery)
|
|
41
|
+
mapNestedQueries(query, rawQuery, queryMapping)
|
|
42
|
+
|
|
37
43
|
// Create a copy of the inputs map to avoid modifying the original
|
|
38
44
|
const allInputs = { ...inputs }
|
|
39
45
|
|
|
@@ -44,7 +50,8 @@ export function compileQuery(
|
|
|
44
50
|
const { alias: mainTableAlias, input: mainInput } = processFrom(
|
|
45
51
|
query.from,
|
|
46
52
|
allInputs,
|
|
47
|
-
cache
|
|
53
|
+
cache,
|
|
54
|
+
queryMapping
|
|
48
55
|
)
|
|
49
56
|
tables[mainTableAlias] = mainInput
|
|
50
57
|
|
|
@@ -68,7 +75,8 @@ export function compileQuery(
|
|
|
68
75
|
tables,
|
|
69
76
|
mainTableAlias,
|
|
70
77
|
allInputs,
|
|
71
|
-
cache
|
|
78
|
+
cache,
|
|
79
|
+
queryMapping
|
|
72
80
|
)
|
|
73
81
|
}
|
|
74
82
|
|
|
@@ -218,8 +226,8 @@ export function compileQuery(
|
|
|
218
226
|
)
|
|
219
227
|
|
|
220
228
|
const result = resultPipeline
|
|
221
|
-
// Cache the result before returning
|
|
222
|
-
cache.set(
|
|
229
|
+
// Cache the result before returning (use original query as key)
|
|
230
|
+
cache.set(rawQuery, result)
|
|
223
231
|
return result
|
|
224
232
|
} else if (query.limit !== undefined || query.offset !== undefined) {
|
|
225
233
|
// If there's a limit or offset without orderBy, throw an error
|
|
@@ -241,8 +249,8 @@ export function compileQuery(
|
|
|
241
249
|
)
|
|
242
250
|
|
|
243
251
|
const result = resultPipeline
|
|
244
|
-
// Cache the result before returning
|
|
245
|
-
cache.set(
|
|
252
|
+
// Cache the result before returning (use original query as key)
|
|
253
|
+
cache.set(rawQuery, result)
|
|
246
254
|
return result
|
|
247
255
|
}
|
|
248
256
|
|
|
@@ -252,7 +260,8 @@ export function compileQuery(
|
|
|
252
260
|
function processFrom(
|
|
253
261
|
from: CollectionRef | QueryRef,
|
|
254
262
|
allInputs: Record<string, KeyedStream>,
|
|
255
|
-
cache: QueryCache
|
|
263
|
+
cache: QueryCache,
|
|
264
|
+
queryMapping: QueryMapping
|
|
256
265
|
): { alias: string; input: KeyedStream } {
|
|
257
266
|
switch (from.type) {
|
|
258
267
|
case `collectionRef`: {
|
|
@@ -265,8 +274,16 @@ function processFrom(
|
|
|
265
274
|
return { alias: from.alias, input }
|
|
266
275
|
}
|
|
267
276
|
case `queryRef`: {
|
|
277
|
+
// Find the original query for caching purposes
|
|
278
|
+
const originalQuery = queryMapping.get(from.query) || from.query
|
|
279
|
+
|
|
268
280
|
// Recursively compile the sub-query with cache
|
|
269
|
-
const subQueryInput = compileQuery(
|
|
281
|
+
const subQueryInput = compileQuery(
|
|
282
|
+
originalQuery,
|
|
283
|
+
allInputs,
|
|
284
|
+
cache,
|
|
285
|
+
queryMapping
|
|
286
|
+
)
|
|
270
287
|
|
|
271
288
|
// Subqueries may return [key, [value, orderByIndex]] (with ORDER BY) or [key, value] (without ORDER BY)
|
|
272
289
|
// We need to extract just the value for use in parent queries
|
|
@@ -283,3 +300,53 @@ function processFrom(
|
|
|
283
300
|
throw new Error(`Unsupported FROM type: ${(from as any).type}`)
|
|
284
301
|
}
|
|
285
302
|
}
|
|
303
|
+
|
|
304
|
+
/**
|
|
305
|
+
* Recursively maps optimized subqueries to their original queries for proper caching.
|
|
306
|
+
* This ensures that when we encounter the same QueryRef object in different contexts,
|
|
307
|
+
* we can find the original query to check the cache.
|
|
308
|
+
*/
|
|
309
|
+
function mapNestedQueries(
|
|
310
|
+
optimizedQuery: QueryIR,
|
|
311
|
+
originalQuery: QueryIR,
|
|
312
|
+
queryMapping: QueryMapping
|
|
313
|
+
): void {
|
|
314
|
+
// Map the FROM clause if it's a QueryRef
|
|
315
|
+
if (
|
|
316
|
+
optimizedQuery.from.type === `queryRef` &&
|
|
317
|
+
originalQuery.from.type === `queryRef`
|
|
318
|
+
) {
|
|
319
|
+
queryMapping.set(optimizedQuery.from.query, originalQuery.from.query)
|
|
320
|
+
// Recursively map nested queries
|
|
321
|
+
mapNestedQueries(
|
|
322
|
+
optimizedQuery.from.query,
|
|
323
|
+
originalQuery.from.query,
|
|
324
|
+
queryMapping
|
|
325
|
+
)
|
|
326
|
+
}
|
|
327
|
+
|
|
328
|
+
// Map JOIN clauses if they exist
|
|
329
|
+
if (optimizedQuery.join && originalQuery.join) {
|
|
330
|
+
for (
|
|
331
|
+
let i = 0;
|
|
332
|
+
i < optimizedQuery.join.length && i < originalQuery.join.length;
|
|
333
|
+
i++
|
|
334
|
+
) {
|
|
335
|
+
const optimizedJoin = optimizedQuery.join[i]!
|
|
336
|
+
const originalJoin = originalQuery.join[i]!
|
|
337
|
+
|
|
338
|
+
if (
|
|
339
|
+
optimizedJoin.from.type === `queryRef` &&
|
|
340
|
+
originalJoin.from.type === `queryRef`
|
|
341
|
+
) {
|
|
342
|
+
queryMapping.set(optimizedJoin.from.query, originalJoin.from.query)
|
|
343
|
+
// Recursively map nested queries in joins
|
|
344
|
+
mapNestedQueries(
|
|
345
|
+
optimizedJoin.from.query,
|
|
346
|
+
originalJoin.from.query,
|
|
347
|
+
queryMapping
|
|
348
|
+
)
|
|
349
|
+
}
|
|
350
|
+
}
|
|
351
|
+
}
|
|
352
|
+
}
|
|
@@ -7,18 +7,13 @@ import {
|
|
|
7
7
|
import { compileExpression } from "./evaluators.js"
|
|
8
8
|
import { compileQuery } from "./index.js"
|
|
9
9
|
import type { IStreamBuilder, JoinType } from "@electric-sql/d2mini"
|
|
10
|
-
import type { CollectionRef, JoinClause,
|
|
10
|
+
import type { CollectionRef, JoinClause, QueryRef } from "../ir.js"
|
|
11
11
|
import type {
|
|
12
12
|
KeyedStream,
|
|
13
13
|
NamespacedAndKeyedStream,
|
|
14
14
|
NamespacedRow,
|
|
15
|
-
ResultStream,
|
|
16
15
|
} from "../../types.js"
|
|
17
|
-
|
|
18
|
-
/**
|
|
19
|
-
* Cache for compiled subqueries to avoid duplicate compilation
|
|
20
|
-
*/
|
|
21
|
-
type QueryCache = WeakMap<QueryIR, ResultStream>
|
|
16
|
+
import type { QueryCache, QueryMapping } from "./types.js"
|
|
22
17
|
|
|
23
18
|
/**
|
|
24
19
|
* Processes all join clauses in a query
|
|
@@ -29,7 +24,8 @@ export function processJoins(
|
|
|
29
24
|
tables: Record<string, KeyedStream>,
|
|
30
25
|
mainTableAlias: string,
|
|
31
26
|
allInputs: Record<string, KeyedStream>,
|
|
32
|
-
cache: QueryCache
|
|
27
|
+
cache: QueryCache,
|
|
28
|
+
queryMapping: QueryMapping
|
|
33
29
|
): NamespacedAndKeyedStream {
|
|
34
30
|
let resultPipeline = pipeline
|
|
35
31
|
|
|
@@ -40,7 +36,8 @@ export function processJoins(
|
|
|
40
36
|
tables,
|
|
41
37
|
mainTableAlias,
|
|
42
38
|
allInputs,
|
|
43
|
-
cache
|
|
39
|
+
cache,
|
|
40
|
+
queryMapping
|
|
44
41
|
)
|
|
45
42
|
}
|
|
46
43
|
|
|
@@ -56,13 +53,15 @@ function processJoin(
|
|
|
56
53
|
tables: Record<string, KeyedStream>,
|
|
57
54
|
mainTableAlias: string,
|
|
58
55
|
allInputs: Record<string, KeyedStream>,
|
|
59
|
-
cache: QueryCache
|
|
56
|
+
cache: QueryCache,
|
|
57
|
+
queryMapping: QueryMapping
|
|
60
58
|
): NamespacedAndKeyedStream {
|
|
61
59
|
// Get the joined table alias and input stream
|
|
62
60
|
const { alias: joinedTableAlias, input: joinedInput } = processJoinSource(
|
|
63
61
|
joinClause.from,
|
|
64
62
|
allInputs,
|
|
65
|
-
cache
|
|
63
|
+
cache,
|
|
64
|
+
queryMapping
|
|
66
65
|
)
|
|
67
66
|
|
|
68
67
|
// Add the joined table to the tables map
|
|
@@ -128,7 +127,8 @@ function processJoin(
|
|
|
128
127
|
function processJoinSource(
|
|
129
128
|
from: CollectionRef | QueryRef,
|
|
130
129
|
allInputs: Record<string, KeyedStream>,
|
|
131
|
-
cache: QueryCache
|
|
130
|
+
cache: QueryCache,
|
|
131
|
+
queryMapping: QueryMapping
|
|
132
132
|
): { alias: string; input: KeyedStream } {
|
|
133
133
|
switch (from.type) {
|
|
134
134
|
case `collectionRef`: {
|
|
@@ -141,8 +141,16 @@ function processJoinSource(
|
|
|
141
141
|
return { alias: from.alias, input }
|
|
142
142
|
}
|
|
143
143
|
case `queryRef`: {
|
|
144
|
+
// Find the original query for caching purposes
|
|
145
|
+
const originalQuery = queryMapping.get(from.query) || from.query
|
|
146
|
+
|
|
144
147
|
// Recursively compile the sub-query with cache
|
|
145
|
-
const subQueryInput = compileQuery(
|
|
148
|
+
const subQueryInput = compileQuery(
|
|
149
|
+
originalQuery,
|
|
150
|
+
allInputs,
|
|
151
|
+
cache,
|
|
152
|
+
queryMapping
|
|
153
|
+
)
|
|
146
154
|
|
|
147
155
|
// Subqueries may return [key, [value, orderByIndex]] (with ORDER BY) or [key, value] (without ORDER BY)
|
|
148
156
|
// We need to extract just the value for use in parent queries
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { QueryIR } from "../ir.js"
|
|
2
|
+
import type { ResultStream } from "../../types.js"
|
|
3
|
+
|
|
4
|
+
/**
|
|
5
|
+
* Cache for compiled subqueries to avoid duplicate compilation
|
|
6
|
+
*/
|
|
7
|
+
export type QueryCache = WeakMap<QueryIR, ResultStream>
|
|
8
|
+
|
|
9
|
+
/**
|
|
10
|
+
* Mapping from optimized queries back to their original queries for caching
|
|
11
|
+
*/
|
|
12
|
+
export type QueryMapping = WeakMap<QueryIR, QueryIR>
|