@uwdata/mosaic-core 0.17.0 → 0.18.0

package/dist/src/preagg/PreAggregator.d.ts

@@ -0,0 +1,150 @@
+import { ExprNode, SelectQuery, MaybeArray } from '@uwdata/mosaic-sql';
+import type { Coordinator } from '../Coordinator.js';
+import type { MosaicClient } from '../MosaicClient.js';
+import type { Selection } from '../Selection.js';
+import type { ClauseSource, SelectionClause } from '../SelectionClause.js';
+declare const Skip: {
+    skip: boolean;
+    result: null;
+};
+export interface PreAggregateOptions {
+    /** Database schema (namespace) in which to write pre-aggregated materialized views (default 'mosaic'). */
+    schema?: string;
+    /** Flag to enable or disable the pre-aggregation. This flag can be updated later via the `enabled` property. */
+    enabled?: boolean;
+}
+type ActivePredicate = (p?: ExprNode) => MaybeArray<ExprNode> | undefined;
+interface ActiveColumnsResult {
+    source: ClauseSource | null;
+    columns?: Record<string, ExprNode>;
+    predicate?: ActivePredicate;
+}
+interface PreAggregateInfoOptions {
+    table: string;
+    create: string;
+    active: ActiveColumnsResult;
+    select: SelectQuery;
+}
+/**
+ * Build and query optimized pre-aggregated materaialized views, for fast
+ * computation of groupby aggregate queries over compatible client queries
+ * and selections. The materialized views contains pre-aggregated data for a
+ * Mosaic client, subdivided by possible query values from an active selection
+ * clause. These materialized views are database tables that can be queried
+ * for rapid updates.
+ *
+ * Compatible client queries must consist of only groupby dimensions and
+ * supported aggregate functions. Compatible selections must contain an active
+ * clause that exposes metadata for an interval or point value predicate.
+ *
+ * Materialized views are written to a dedicated schema (namespace) that
+ * can be set using the *schema* constructor option. This schema acts as a
+ * persistent cache, and materialized view tables may be used across sessions.
+ * The `dropSchema` method issues a query to remove *all* tables within this
+ * schema. This may be needed if the original tables have updated data, but
+ * should be used with care.
+ */
+export declare class PreAggregator {
+    entries: Map<MosaicClient, PreAggregateInfo | typeof Skip | null>;
+    private active;
+    private mc;
+    private _schema;
+    private _enabled;
+    /**
+     * Create a new manager of materialized views of pre-aggregated data.
+     * @param coordinator A Mosaic coordinator.
+     * @param options Pre-aggregation options.
+     */
+    constructor(coordinator: Coordinator, { schema, enabled }?: PreAggregateOptions);
+    /**
+     * Set the enabled state of this manager. If false, any local state is
+     * cleared and subsequent request calls will return null until re-enabled.
+     * This method has no effect on any pre-aggregated tables already in the
+     * database.
+     * @param state The enabled state to set.
+     */
+    set enabled(state: boolean);
+    /**
+     * Get the enabled state of this manager.
+     * @returns The current enabled state.
+     */
+    get enabled(): boolean;
+    /**
+     * Set the database schema used for pre-aggregated materialized view tables.
+     * Upon changes, any local state is cleared. This method does _not_ drop any
+     * existing materialized views, use `dropSchema` before changing the schema
+     * to also remove existing materalized views in the database.
+     * @param schema The schema name to set.
+     */
+    set schema(schema: string);
+    /**
+     * Get the database schema used for pre-aggregated materialized view tables.
+     * @returns The current schema name.
+     */
+    get schema(): string;
+    /**
+     * Issues a query through the coordinator to drop the current schema for
+     * pre-aggregated materialized views. *All* materialized view tables in the
+     * schema will be removed and local state is cleared. Call this method if
+     * the underlying base tables have been updated, causing materialized view
+     * to become stale and inaccurate. Use this method with care! Once dropped,
+     * the schema will be repopulated by future pre-aggregation requests.
+     * @returns A query result promise.
+     */
+    dropSchema(): Promise<unknown>;
+    /**
+     * Clear the cache of pre-aggregation entries for the current active
+     * selection clause. This method does _not_ drop any existing materialized
+     * views. Use `dropSchema` to remove existing materialized view tables from
+     * the database.
+     */
+    clear(): void;
+    /**
+     * Return pre-aggregation information for the active state of a
+     * client-selection pair, or null if the client has unstable filters.
+     * This method has multiple possible side effects, including materialized
+     * view creation and updating internal caches.
+     * @param client A Mosaic client.
+     * @param selection A Mosaic selection to filter the client by.
+     * @param activeClause A representative active selection
+     *  clause for which to generate materialized views of pre-aggregates.
+     * @returns Information and query generator
+     * for pre-aggregated tables, or null if the client has unstable filters.
+     */
+    request(client: MosaicClient, selection: Selection, activeClause: SelectionClause | null): PreAggregateInfo | typeof Skip | null;
+}
+/**
+ * Metadata and query generator for materialized views of pre-aggregated data.
+ * This object provides the information needed to generate and query the
+ * materialized views for a client-selection pair relative to a specific
+ * active clause and selection state.
+ */
+export declare class PreAggregateInfo {
+    /** The name of the materialized view. */
+    table: string;
+    /** The SQL query used to generate the materialized view. */
+    create: string;
+    /** A result promise returned for the materialized view creation query. */
+    result: Promise<unknown> | null;
+    /** Definitions and predicate function for the active columns,
+     * which are dynamically filtered by the active clause. */
+    active: ActiveColumnsResult;
+    /** Select query (sans where clause) for materialized views. */
+    select: SelectQuery;
+    /** Boolean flag indicating a client that should be skipped.
+     * This value is always false for a created materialized view. */
+    skip: boolean;
+    /**
+     * Create a new pre-aggregation information instance.
+     * @param options Options object.
+     */
+    constructor({ table, create, active, select }: PreAggregateInfoOptions);
+    /**
+     * Generate a materialized view query for the given predicate.
+     * @param predicate The current active clause predicate.
+     * @returns A materialized view query.
+     */
+    query(predicate: ExprNode): SelectQuery;
+}
+export {};
+//# sourceMappingURL=PreAggregator.d.ts.map

package/dist/src/preagg/PreAggregator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"PreAggregator.d.ts","sourceRoot":"","sources":["../../../src/preagg/PreAggregator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAgB,WAAW,EAAoB,UAAU,EAAwC,MAAM,oBAAoB,CAAC;AAC7I,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAa,YAAY,EAAoB,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAKxG,QAAA,MAAM,IAAI;;;CAA+B,CAAC;AAE1C,MAAM,WAAW,mBAAmB;IAClC,0GAA0G;IAC1G,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,gHAAgH;IAChH,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,KAAK,eAAe,GAAG,CAAC,CAAC,CAAC,EAAE,QAAQ,KAAK,UAAU,CAAC,QAAQ,CAAC,GAAG,SAAS,CAAC;AAE1E,UAAU,mBAAmB;IAC3B,MAAM,EAAE,YAAY,GAAG,IAAI,CAAC;IAC5B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACnC,SAAS,CAAC,EAAE,eAAe,CAAC;CAC7B;AAED,UAAU,uBAAuB;IAC/B,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,mBAAmB,CAAC;IAC5B,MAAM,EAAE,WAAW,CAAC;CACrB;AAED;;;;;;;;;;;;;;;;;;GAkBG;AACH,qBAAa,aAAa;IACjB,OAAO,EAAE,GAAG,CAAC,YAAY,EAAE,gBAAgB,GAAG,OAAO,IAAI,GAAG,IAAI,CAAC,CAAC;IACzE,OAAO,CAAC,MAAM,CAA6B;IAC3C,OAAO,CAAC,EAAE,CAAc;IACxB,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,QAAQ,CAAU;IAE1B;;;;OAIG;gBACS,WAAW,EAAE,WAAW,EAAE,EACpC,MAAiB,EACjB,OAAc,EACf,GAAE,mBAAwB;IAQ3B;;;;;;OAMG;IACH,IAAI,OAAO,CAAC,KAAK,EAAE,OAAO,EAKzB;IAED;;;OAGG;IACH,IAAI,OAAO,IAAI,OAAO,CAErB;IAED;;;;;;OAMG;IACH,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,EAKxB;IAED;;;OAGG;IACH,IAAI,MAAM,IAAI,MAAM,CAEnB;IAED;;;;;;;;OAQG;IACH,UAAU,IAAI,OAAO,CAAC,OAAO,CAAC;IAK9B;;;;;OAKG;IACH,KAAK,IAAI,IAAI;IAKb;;;;;;;;;;;OAWG;IACH,OAAO,CAAC,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE,YAAY,EAAE,eAAe,GAAG,IAAI,GAAG,gBAAgB,GAAG,OAAO,IAAI,GAAG,IAAI;CAoEjI;AA8LD;;;;;GAKG;AACH,qBAAa,gBAAgB;IAC3B,yCAAyC;IACzC,KAAK,EAAE,MAAM,CAAC;IACd,4DAA4D;IAC5D,MAAM,EAAE,MAAM,CAAC;IACf,0EAA0E;IAC1E,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC;IAChC;8DAC0D;IAC1D,MAAM,EAAE,mBAAmB,CAAC;IAC5B,+DAA+D;IAC/D,MAAM,EAAE,WAAW,CAAC;IACpB;qEACiE;IACjE,IAAI,EAAE,OAAO,CAAC;IAEd;;;OAGG;gBACS,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,uBAAuB;IAStE;;;;OAIG;IACH,KAAK,CAAC,SAAS,EAAE,QAAQ,GAAG,WAAW;CAGxC"}

package/dist/src/preagg/PreAggregator.js

@@ -0,0 +1,382 @@
+import { Query as QueryBuilder, and, asNode, ceil, collectColumns, createTable, float64, floor, isBetween, int32, mul, round, scaleTransform, sub, isSelectQuery, isAggregateExpression, ColumnNameRefNode } from '@uwdata/mosaic-sql';
+import { preaggColumns } from './preagg-columns.js';
+import { fnv_hash } from '../util/hash.js';
+const Skip = { skip: true, result: null };
+/**
+ * Build and query optimized pre-aggregated materaialized views, for fast
+ * computation of groupby aggregate queries over compatible client queries
+ * and selections. The materialized views contains pre-aggregated data for a
+ * Mosaic client, subdivided by possible query values from an active selection
+ * clause. These materialized views are database tables that can be queried
+ * for rapid updates.
+ *
+ * Compatible client queries must consist of only groupby dimensions and
+ * supported aggregate functions. Compatible selections must contain an active
+ * clause that exposes metadata for an interval or point value predicate.
+ *
+ * Materialized views are written to a dedicated schema (namespace) that
+ * can be set using the *schema* constructor option. This schema acts as a
+ * persistent cache, and materialized view tables may be used across sessions.
+ * The `dropSchema` method issues a query to remove *all* tables within this
+ * schema. This may be needed if the original tables have updated data, but
+ * should be used with care.
+ */
+export class PreAggregator {
+    entries;
+    active;
+    mc;
+    _schema;
+    _enabled;
+    /**
+     * Create a new manager of materialized views of pre-aggregated data.
+     * @param coordinator A Mosaic coordinator.
+     * @param options Pre-aggregation options.
+     */
+    constructor(coordinator, { schema = 'mosaic', enabled = true } = {}) {
+        this.entries = new Map();
+        this.active = null;
+        this.mc = coordinator;
+        this._schema = schema;
+        this._enabled = enabled;
+    }
+    /**
+     * Set the enabled state of this manager. If false, any local state is
+     * cleared and subsequent request calls will return null until re-enabled.
+     * This method has no effect on any pre-aggregated tables already in the
+     * database.
+     * @param state The enabled state to set.
+     */
+    set enabled(state) {
+        if (this._enabled !== state) {
+            if (!state)
+                this.clear();
+            this._enabled = state;
+        }
+    }
+    /**
+     * Get the enabled state of this manager.
+     * @returns The current enabled state.
+     */
+    get enabled() {
+        return this._enabled;
+    }
+    /**
+     * Set the database schema used for pre-aggregated materialized view tables.
+     * Upon changes, any local state is cleared. This method does _not_ drop any
+     * existing materialized views, use `dropSchema` before changing the schema
+     * to also remove existing materalized views in the database.
+     * @param schema The schema name to set.
+     */
+    set schema(schema) {
+        if (this._schema !== schema) {
+            this.clear();
+            this._schema = schema;
+        }
+    }
+    /**
+     * Get the database schema used for pre-aggregated materialized view tables.
+     * @returns The current schema name.
+     */
+    get schema() {
+        return this._schema;
+    }
+    /**
+     * Issues a query through the coordinator to drop the current schema for
+     * pre-aggregated materialized views. *All* materialized view tables in the
+     * schema will be removed and local state is cleared. Call this method if
+     * the underlying base tables have been updated, causing materialized view
+     * to become stale and inaccurate. Use this method with care! Once dropped,
+     * the schema will be repopulated by future pre-aggregation requests.
+     * @returns A query result promise.
+     */
+    dropSchema() {
+        this.clear();
+        return this.mc.exec(`DROP SCHEMA IF EXISTS "${this.schema}" CASCADE`);
+    }
+    /**
+     * Clear the cache of pre-aggregation entries for the current active
+     * selection clause. This method does _not_ drop any existing materialized
+     * views. Use `dropSchema` to remove existing materialized view tables from
+     * the database.
+     */
+    clear() {
+        this.entries.clear();
+        this.active = null;
+    }
+    /**
+     * Return pre-aggregation information for the active state of a
+     * client-selection pair, or null if the client has unstable filters.
+     * This method has multiple possible side effects, including materialized
+     * view creation and updating internal caches.
+     * @param client A Mosaic client.
+     * @param selection A Mosaic selection to filter the client by.
+     * @param activeClause A representative active selection
+     *  clause for which to generate materialized views of pre-aggregates.
+     * @returns Information and query generator
+     * for pre-aggregated tables, or null if the client has unstable filters.
+     */
+    request(client, selection, activeClause) {
+        // if not enabled, do nothing
+        if (!this.enabled || activeClause == null)
+            return null;
+        const { entries, mc, schema } = this;
+        const { source } = activeClause;
+        // if there is no clause source to track, do nothing
+        if (!source)
+            return null;
+        // if we have cached active columns, check for updates or exit
+        if (this.active) {
+            // if the active clause source has changed, clear the state
+            // this cancels outstanding requests and clears the local cache
+            // a clear also sets this.active to null
+            if (this.active.source !== source)
+                this.clear();
+            // if we've seen this source and it has unstable filters, do nothing
+            if (this.active?.source === null)
+                return null;
+        }
+        // the current active columns cache value
+        let { active } = this;
+        // if cached active columns are unset, analyze the active clause
+        if (!active) {
+            // if active clause predicate is null, we can't analyze it
+            // return null to backoff to standard client query
+            // non-null clauses may come later, so don't set active state
+            if (activeClause.predicate == null)
+                return null;
+            // generate active dimension columns to select over
+            // will return an object with null source if it has unstable filters
+            this.active = active = activeColumns(activeClause);
+            // if the active clause has unstable filters, exit now
+            if (active.source === null)
+                return null;
+        }
+        // if we have cached pre-aggregate info, return that
+        if (entries.has(client)) {
+            return entries.get(client);
+        }
+        // get non-active materialized view columns
+        const preaggCols = preaggColumns(client);
+        let info;
+        if (!preaggCols) {
+            // if client is not indexable, record null info
+            info = null;
+        }
+        else if (selection.skip(client, activeClause)) {
+            // skip client if untouched by cross-filtering
+            info = Skip;
+        }
+        else {
+            // generate materialized view table
+            const filter = selection.remove(source).predicate(client);
+            info = preaggregateInfo(client.query(filter), active, preaggCols, schema);
+            info.result = mc.exec([
+                `CREATE SCHEMA IF NOT EXISTS ${schema}`,
+                createTable(info.table, info.create, { temp: false })
+            ]);
+            info.result.catch((e) => mc.logger().error(e));
+        }
+        entries.set(client, info);
+        return info;
+    }
+}
+/**
+ * Determines the active dimension columns to select over. Returns an object
+ * with the clause source, column definitions, and a predicate generator
+ * function for the active dimensions of a pre-aggregated materialized view.
+ * If the active clause is not indexable or is missing metadata, this method
+ * returns an object with a null source property.
+ * @param clause The active selection clause to analyze.
+ */
+function activeColumns(clause) {
+    const { source, meta } = clause;
+    const clausePred = clause.predicate;
+    const clauseCols = collectColumns(clausePred).map(c => c.column);
+    let predicate;
+    let columns;
+    if (!meta || !clauseCols) {
+        return { source: null, columns, predicate };
+    }
+    switch (meta.type) {
+        case 'point':
+            predicate = x => x;
+            columns = Object.fromEntries(clauseCols.map(col => [`${col}`, asNode(col)]));
+            break;
+        case 'interval': {
+            const { scales, bin, pixelSize = 1 } = meta;
+            if (!scales)
+                break;
+            // determine pixel-level binning
+            const bins = scales.map((s) => binInterval(s, pixelSize, bin));
+            if (bins.some(b => !b)) {
+                // bail if a scale type is unsupported
+            }
+            else if (bins.length === 1) {
+                // selection clause predicate has type BetweenOpNode
+                // single interval selection
+                predicate = (p) => p
+                    ? isBetween('active0', p.extent?.map(bins[0]))
+                    : [];
+                columns = { active0: bins[0](clausePred.expr) };
+            }
+            else {
+                // selection clause predicate has type AndNode<BetweenOpNode>
+                // multiple interval selection
+                predicate = (p) => p
+                    ? and(p.clauses.map((c, i) => isBetween(`active${i}`, c.extent?.map(bins[i]))))
+                    : [];
+                columns = Object.fromEntries(clausePred.clauses.map((p, i) => [`active${i}`, bins[i](p.expr)]));
+            }
+        }
+    }
+    return { source: columns ? source : null, columns, predicate };
+}
+const BIN = { ceil, round };
+/**
+ * Returns a bin function generator to discretize a selection interval domain.
+ * @param scale A scale that maps domain values to the output range
+ *  (typically pixels).
+ * @param pixelSize The interactive pixel size. This value indicates
+ *  the bin step size and may be greater than an actual screen pixel.
+ * @param bin The binning method to apply, one of `floor`, `ceil', or `round`.
+ * @returns A bin function generator.
+ */
+function binInterval(scale, pixelSize, bin) {
+    const { type, domain, range, apply, sqlApply } = scaleTransform(scale);
+    if (!apply)
+        return; // unsupported scale type
+    const binFn = BIN[`${bin}`.toLowerCase()] || floor;
+    const dom = domain.map(x => Number(x));
+    const lo = apply(Math.min(...dom));
+    const hi = apply(Math.max(...dom));
+    const s = (type === 'identity'
+        ? 1
+        : Math.abs(range[1] - range[0]) / (hi - lo)) / pixelSize;
+    const scalar = s === 1
+        ? (x) => x
+        : (x) => mul(float64(s), x);
+    const diff = lo === 0
+        ? (x) => x
+        : (x) => sub(x, float64(lo));
+    return value => int32(binFn(scalar(diff(sqlApply(value)))));
+}
+/**
+ * Generate pre-aggregate query information.
+ * @param clientQuery The original client query.
+ * @param active Active (selected) columns.
+ * @param preaggCols Pre-aggregation columns.
+ * @param schema Database schema name.
+ * @returns Pre-aggregation information.
+ */
+function preaggregateInfo(clientQuery, active, preaggCols, schema) {
+    const { group, output, preagg } = preaggCols;
+    const { columns } = active;
+    // build materialized view construction query
+    const query = clientQuery
+        .setSelect({ ...preagg, ...columns })
+        .groupby(Object.keys(columns));
+    // ensure active clause columns are selected by subqueries
+    const [subq] = query.subqueries;
+    if (subq) {
+        const cols = Object.values(columns)
+            .flatMap(c => collectColumns(c).map(c => c.column));
+        subqueryPushdown(subq, cols);
+    }
+    // push any having or orderby criteria to output queries
+    const having = query._having;
+    const order = query._orderby;
+    query._having = [];
+    query._orderby = [];
+    // generate creation query string and hash id
+    const create = query.toString();
+    const id = (fnv_hash(create) >>> 0).toString(16);
+    const table = `${schema}.preagg_${id}`;
+    // generate preaggregate select query
+    const select = QueryBuilder
+        .select(group, output)
+        .from(table)
+        .groupby(group)
+        .having(having)
+        .orderby(order);
+    return new PreAggregateInfo({ table, create, active, select });
+}
+/**
+ * Push column selections down to subqueries.
+ * @param query The (sub)query to push down to.
+ * @param cols The column names to push down.
+ */
+function subqueryPushdown(query, cols) {
+    const memo = new Set();
+    const pushdown = (q) => {
+        // it is possible to have duplicate subqueries
+        // so we memoize and exit early if already seen
+        if (memo.has(q))
+            return;
+        memo.add(q);
+        if (isSelectQuery(q) && q._from.length) {
+            // select the pushed down columns
+            // note that the select method will deduplicate for us
+            q.select(cols);
+            if (isAggregateQuery(q)) {
+                // if an aggregation query, we need to push to groupby as well
+                // we also deduplicate as the column may already be present
+                const set = new Set(q._groupby.flatMap(x => x instanceof ColumnNameRefNode ? [x.name] : []));
+                q.groupby(cols.filter(c => !set.has(c)));
+            }
+        }
+        q.subqueries.forEach(pushdown);
+    };
+    pushdown(query);
+}
+/**
+ * Test if a query performs aggregation.
+ * @param query Select query to test.
+ * @returns True if query performs aggregation.
+ */
+function isAggregateQuery(query) {
+    return query._groupby.length > 0
+        || query._select.some(node => isAggregateExpression(node));
+}
+/**
+ * Metadata and query generator for materialized views of pre-aggregated data.
+ * This object provides the information needed to generate and query the
+ * materialized views for a client-selection pair relative to a specific
+ * active clause and selection state.
+ */
+export class PreAggregateInfo {
+    /** The name of the materialized view. */
+    table;
+    /** The SQL query used to generate the materialized view. */
+    create;
+    /** A result promise returned for the materialized view creation query. */
+    result;
+    /** Definitions and predicate function for the active columns,
+     * which are dynamically filtered by the active clause. */
+    active;
+    /** Select query (sans where clause) for materialized views. */
+    select;
+    /** Boolean flag indicating a client that should be skipped.
+     * This value is always false for a created materialized view. */
+    skip;
+    /**
+     * Create a new pre-aggregation information instance.
+     * @param options Options object.
+     */
+    constructor({ table, create, active, select }) {
+        this.table = table;
+        this.create = create;
+        this.result = null;
+        this.active = active;
+        this.select = select;
+        this.skip = false;
+    }
+    /**
+     * Generate a materialized view query for the given predicate.
+     * @param predicate The current active clause predicate.
+     * @returns A materialized view query.
+     */
+    query(predicate) {
+        return this.select.clone().where(this.active.predicate(predicate));
+    }
+}
+//# sourceMappingURL=PreAggregator.js.map

package/dist/src/preagg/PreAggregator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"PreAggregator.js","sourceRoot":"","sources":["../../../src/preagg/PreAggregator.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,KAAK,IAAI,YAAY,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,cAAc,EAAE,GAAG,EAAE,aAAa,EAAE,qBAAqB,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvO,OAAO,EAAE,aAAa,EAAuB,MAAM,qBAAqB,CAAC;AACzE,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAE3C,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;AAwB1C;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAM,OAAO,aAAa;IACjB,OAAO,CAA2D;IACjE,MAAM,CAA6B;IACnC,EAAE,CAAc;IAChB,OAAO,CAAS;IAChB,QAAQ,CAAU;IAE1B;;;;OAIG;IACH,YAAY,WAAwB,EAAE,EACpC,MAAM,GAAG,QAAQ,EACjB,OAAO,GAAG,IAAI,KACS,EAAE;QACzB,IAAI,CAAC,OAAO,GAAG,IAAI,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,EAAE,GAAG,WAAW,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;IAC1B,CAAC;IAED;;;;;;OAMG;IACH,IAAI,OAAO,CAAC,KAAc;QACxB,IAAI,IAAI,CAAC,QAAQ,KAAK,KAAK,EAAE,CAAC;YAC5B,IAAI,CAAC,KAAK;gBAAE,IAAI,CAAC,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC;QACxB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED;;;;;;OAMG;IACH,IAAI,MAAM,CAAC,MAAc;QACvB,IAAI,IAAI,CAAC,OAAO,KAAK,MAAM,EAAE,CAAC;YAC5B,IAAI,CAAC,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACxB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED;;;;;;;;OAQG;IACH,UAAU;QACR,IAAI,CAAC,KAAK,EAAE,CAAC;QACb,OAAO,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,0BAA0B,IAAI,CAAC,MAAM,WAAW,CAAC,CAAC;IACxE,CAAC;IAED;;;;;OAKG;IACH,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAED;;;;;;;;;;;OAWG;IACH,OAAO,CAAC,MAAoB,EAAE,SAAoB,EAAE,YAAoC;QACtF,6BAA6B;QAC7B,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,YAAY,IAAI,IAAI;YAAE,OAAO,IAAI,CAAC;QAEvD,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACrC,MAAM,EAAE,MAAM,EAAE,GAAG,YAAY,CAAC;QAEhC,oDAAoD;QACpD,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC;QAEzB,8DAA8D;QAC9D,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,2DAA2D;YAC3D,+DAA+D;YAC/D,wCAAwC;YACxC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,MAAM;gBAAE,IAAI,CAAC,KAAK,EAAE,CAAC;YAChD,oEAAoE;YACpE,IAAI,IAAI,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI;gBAAE,OAAO,IAAI,CAAC;QAChD,CAAC;QAED,yCAAyC;QACzC,IAAI,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QAEtB,gEAAgE;QAChE,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,0DAA0D;YAC1D,kDAAkD;YAClD,6DAA6D;YAC7D,IAAI,YAAY,CAAC,SAAS,IAAI,IAAI;gBAAE,OAAO,IAAI,CAAC;YAChD,mDAAmD;YACnD,oEAAoE;YACpE,IAAI,CAAC,MAAM,GAAG,MAAM,GAAG,aAAa,CAAC,YAAY,CAAC,CAAC;YACnD,sDAAsD;YACtD,IAAI,MAAM,CAAC,MAAM,KAAK,IAAI;gBAAE,OAAO,IAAI,CAAC;QAC1C,CAAC;QAED,oDAAoD;QACpD,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YACxB,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC;QAC9B,CAAC;QAED,2CAA2C;QAC3C,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QAEzC,IAAI,IAA2C,CAAC;QAChD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,+CAA+C;YAC/C,IAAI,GAAG,IAAI,CAAC;QACd,CAAC;aAAM,IAAI,SAAS,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,EAAE,CAAC;YAChD,8CAA8C;YAC9C,IAAI,GAAG,IAAI,CAAC;QACd,CAAC;aAAM,CAAC;YACN,mCAAmC;YACnC,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAC1D,IAAI,GAAG,gBAAgB,CACrB,MAAM,CAAC,KAAK,CAAC,MAAM,CAAgB,EACnC,MAAM,EAAE,UAAU,EAAE,MAAM,CAC3B,CAAC;YACF,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC;gBACpB,+BAA+B,MAAM,EAAE;gBACvC,WAAW,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;aACtD,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QACxD,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1B,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAED;;;;;;;GAOG;AACH,SAAS,aAAa,CAAC,MAAuB;IAC5C,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,CAAC;IAChC,MAAM,UAAU,GAAG,MAAM,CAAC,SAAS,CAAC;IACpC,MAAM,UAAU,GAAG,cAAc,CAAC,UAAW,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;IAClE,IAAI,SAAsC,CAAC;IAC3C,IAAI,OAA6C,CAAC;IAElD,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;QACzB,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;IAC9C,CAAC;IAED,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,OAAO;YACV,SAAS,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACnB,OAAO,GAAG,MAAM,CAAC,WAAW,CAC1B,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,GAAG,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAC/C,CAAC;YACF,MAAM;QACR,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,GAAG,CAAC,EAAE,GAAI,IAAyB,CAAC;YAClE,IAAI,CAAC,MAAM;gBAAE,MAAM;YAEnB,gCAAgC;YAChC,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAe,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,CAAC,CAAC;YAC7E,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvB,sCAAsC;YACxC,CAAC;iBAAM,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,oDAAoD;gBACpD,4BAA4B;gBAC5B,SAAS,GAAG,CAAC,CAAY,EAAE,EAAE,CAAC,CAAC;oBAC7B,CAAC,CAAC,SAAS,CAAC,SAAS,EAAG,CAAmB,CAAC,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,CAAC;oBAClE,CAAC,CAAC,EAAE,CAAC;gBACP,OAAO,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAE,CAAE,UAA4B,CAAC,IAAI,CAAC,EAAE,CAAC;YACtE,CAAC;iBAAM,CAAC;gBACN,6DAA6D;gBAC7D,8BAA8B;gBAC9B,SAAS,GAAG,CAAC,CAAY,EAAE,EAAE,CAAC,CAAC;oBAC7B,CAAC,CAAC,GAAG,CAAE,CAA4B,CAAC,OAAO,CAAC,GAAG,CAC3C,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,CAAC,CAC3D,CAAC;oBACJ,CAAC,CAAC,EAAE,CAAC;gBACP,OAAO,GAAG,MAAM,CAAC,WAAW,CACzB,UAAqC,CAAC,OAAO,CAAC,GAAG,CAChD,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAC3C,CACF,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;AACjE,CAAC;AAED,MAAM,GAAG,GAAsD,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;AAE/E;;;;;;;;GAQG;AACH,SAAS,WAAW,CAClB,KAAmB,EACnB,SAAiB,EACjB,GAAe;IAEf,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,cAAc,CAAC,KAAK,CAAE,CAAC;IACxE,IAAI,CAAC,KAAK;QAAE,OAAO,CAAC,yBAAyB;IAC7C,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC,IAAI,KAAK,CAAC;IACnD,MAAM,GAAG,GAAG,MAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IACxC,MAAM,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;IACnC,MAAM,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;IACnC,MAAM,CAAC,GAAG,CAAC,IAAI,KAAK,UAAU;QAC5B,CAAC,CAAC,CAAC;QACH,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,KAAM,CAAC,CAAC,CAAC,GAAG,KAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,SAAS,CAAC;IAC7D,MAAM,MAAM,GAAG,CAAC,KAAK,CAAC;QACpB,CAAC,CAAC,CAAC,CAAY,EAAE,EAAE,CAAC,CAAC;QACrB,CAAC,CAAC,CAAC,CAAY,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACzC,MAAM,IAAI,GAAG,EAAE,KAAK,CAAC;QACnB,CAAC,CAAC,CAAC,CAAY,EAAE,EAAE,CAAC,CAAC;QACrB,CAAC,CAAC,CAAC,CAAY,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1C,OAAO,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAC9D,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,gBAAgB,CACvB,WAAwB,EACxB,MAA2B,EAC3B,UAA+B,EAC/B,MAAc;IAEd,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC;IAC7C,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC;IAE3B,6CAA6C;IAC7C,MAAM,KAAK,GAAG,WAAW;SACtB,SAAS,CAAC,EAAE,GAAG,MAAM,EAAE,GAAG,OAAO,EAAE,CAAC;SACpC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,OAAQ,CAAC,CAAC,CAAC;IAElC,0DAA0D;IAC1D,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,UAAU,CAAC;IAChC,IAAI,IAAI,EAAE,CAAC;QACT,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,OAAQ,CAAC;aACjC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;QACtD,gBAAgB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAC/B,CAAC;IAED,wDAAwD;IACxD,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC;IAC7B,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,CAAC;IAC7B,KAAK,CAAC,OAAO,GAAG,EAAE,CAAC;IACnB,KAAK,CAAC,QAAQ,GAAG,EAAE,CAAC;IAEpB,6CAA6C;IAC7C,MAAM,MAAM,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;IAChC,MAAM,EAAE,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjD,MAAM,KAAK,GAAG,GAAG,MAAM,WAAW,EAAE,EAAE,CAAC;IAEvC,qCAAqC;IACrC,MAAM,MAAM,GAAG,YAAY;SACxB,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC;SACrB,IAAI,CAAC,KAAK,CAAC;SACX,OAAO,CAAC,KAAK,CAAC;SACd,MAAM,CAAC,MAAM,CAAC;SACd,OAAO,CAAC,KAAK,CAAC,CAAC;IAElB,OAAO,IAAI,gBAAgB,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;AACjE,CAAC;AAED;;;;GAIG;AACH,SAAS,gBAAgB,CAAC,KAAY,EAAE,IAAc;IACpD,MAAM,IAAI,GAAG,IAAI,GAAG,EAAE,CAAC;IACvB,MAAM,QAAQ,GAAG,CAAC,CAAQ,EAAE,EAAE;QAC5B,8CAA8C;QAC9C,+CAA+C;QAC/C,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YAAE,OAAO;QACxB,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAEZ,IAAI,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;YACvC,iCAAiC;YACjC,sDAAsD;YACtD,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACf,IAAI,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC;gBACxB,8DAA8D;gBAC9D,2DAA2D;gBAC3D,MAAM,GAAG,GAAG,IAAI,GAAG,CACjB,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,YAAY,iBAAiB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CACxE,CAAC;gBACF,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;QACD,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACjC,CAAC,CAAC;IACF,QAAQ,CAAC,KAAK,CAAC,CAAC;AAClB,CAAC;AAED;;;;GAIG;AACH,SAAS,gBAAgB,CAAC,KAAkB;IAC1C,OAAO,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC;WAC3B,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC;AAC/D,CAAC;AAED;;;;;GAKG;AACH,MAAM,OAAO,gBAAgB;IAC3B,yCAAyC;IACzC,KAAK,CAAS;IACd,4DAA4D;IAC5D,MAAM,CAAS;IACf,0EAA0E;IAC1E,MAAM,CAA0B;IAChC;8DAC0D;IAC1D,MAAM,CAAsB;IAC5B,+DAA+D;IAC/D,MAAM,CAAc;IACpB;qEACiE;IACjE,IAAI,CAAU;IAEd;;;OAGG;IACH,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAA2B;QACpE,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC;IACpB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,SAAmB;QACvB,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,SAAU,CAAC,SAAS,CAAE,CAAC,CAAC;IACvE,CAAC;CACF"}

package/dist/src/preagg/preagg-columns.d.ts

@@ -0,0 +1,16 @@
+import type { ExprNode } from '@uwdata/mosaic-sql';
+import type { MosaicClient } from '../MosaicClient.js';
+export interface PreAggColumnsResult {
+    group: string[];
+    preagg: Record<string, ExprNode>;
+    output: Record<string, ExprNode>;
+}
+/**
+ * Determine pre-aggregation columns for a given Mosaic client.
+ * @param client The Mosaic client.
+ * @returns An object with necessary column data to generate pre-aggregated
+ *  columns, or null if the client can't be optimized or the client query
+ *  contains an invalid or unsupported expression.
+ */
+export declare function preaggColumns(client: MosaicClient): PreAggColumnsResult | null;
+//# sourceMappingURL=preagg-columns.d.ts.map

package/dist/src/preagg/preagg-columns.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"preagg-columns.d.ts","sourceRoot":"","sources":["../../../src/preagg/preagg-columns.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAgC,QAAQ,EAAsB,MAAM,oBAAoB,CAAC;AACrG,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAIvD,MAAM,WAAW,mBAAmB;IAClC,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACjC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;CAClC;AAED;;;;;;GAMG;AACH,wBAAgB,aAAa,CAAC,MAAM,EAAE,YAAY,GAAG,mBAAmB,GAAG,IAAI,CA0D9E"}

package/dist/src/preagg/preagg-columns.js

@@ -0,0 +1,95 @@
+import { collectAggregates, isAggregateExpression, isSelectQuery, isTableRef, rewrite, sql } from '@uwdata/mosaic-sql';
+import { sufficientStatistics } from './sufficient-statistics.js';
+/**
+ * Determine pre-aggregation columns for a given Mosaic client.
+ * @param client The Mosaic client.
+ * @returns An object with necessary column data to generate pre-aggregated
+ *  columns, or null if the client can't be optimized or the client query
+ *  contains an invalid or unsupported expression.
+ */
+export function preaggColumns(client) {
+    if (!client.filterStable)
+        return null;
+    const q = client.query();
+    // bail if query is not analyzable
+    if (!isSelectQuery(q))
+        return null;
+    // bail if no base table
+    const from = getBase(q, q => {
+        const ref = q._from[0]?.expr;
+        return isTableRef(ref) ? ref.name : ref;
+    });
+    if (typeof from !== 'string')
+        return null;
+    const aggrs = new Map();
+    const preagg = {};
+    const output = {};
+    const group = []; // list of grouping dimension columns
+    // generate a scalar subquery for a global average
+    const avg = (ref) => {
+        const name = ref.column;
+        const expr = getBase(q, q => q._select.find(c => c.alias === name)?.expr);
+        return sql `(SELECT avg(${expr ?? ref}) FROM "${from}")`;
+    };
+    // iterate over select clauses and analyze expressions
+    for (const { alias, expr } of q._select) {
+        // bail if there is an aggregate we can't analyze
+        // a value > 1 indicates an aggregate in verbatim text
+        if (isAggregateExpression(expr) > 1)
+            return null;
+        const nodes = collectAggregates(expr);
+        if (nodes.length === 0) {
+            // if no aggregates, expr is a groupby dimension
+            group.push(alias);
+            preagg[alias] = expr;
+        }
+        else {
+            for (const node of nodes) {
+                // bail if distinct aggregate
+                if (node.isDistinct)
+                    return null;
+                // bail if aggregate function is unsupported
+                // otherwise add output aggregate to rewrite map
+                const agg = sufficientStatistics(node, preagg, avg);
+                if (!agg)
+                    return null;
+                aggrs.set(node, agg);
+            }
+            // rewrite original select clause to use preaggregates
+            output[alias] = rewrite(expr, aggrs);
+        }
+    }
+    // bail if the query has no aggregates
+    if (!aggrs.size)
+        return null;
+    return { group, preagg, output };
+}
+/**
+ * Identify a shared base (source) query and extract a value from it.
+ * This method is used to find a shared base table name or extract
+ * the original column name within a base table.
+ * @param query The input query.
+ * @param get A getter function to extract
+ *  a value from a base query.
+ * @returns the base query value, or
+ *  `undefined` if there is no source table, or `NaN` if the
+ *  query operates over multiple source tables.
+ */
+function getBase(query, get) {
+    const subq = query.subqueries;
+    // select query
+    if (isSelectQuery(query) && subq.length === 0) {
+        return get(query);
+    }
+    // handle set operations / subqueries
+    const base = getBase(subq[0], get);
+    for (let i = 1; i < subq.length; ++i) {
+        const value = getBase(subq[i], get);
+        if (value === undefined)
+            continue;
+        if (value !== base)
+            return NaN;
+    }
+    return base;
+}
+//# sourceMappingURL=preagg-columns.js.map