@malloydata/malloy 0.0.38-dev230601002801 → 0.0.38-dev230605182241

package/dist/dialect/standardsql.js

@@ -364,13 +364,13 @@ ${(0, utils_1.indent)(sql)}
     }
     sqlMeasureTime(from, to, units) {
         const measureMap = {
-            microsecond: { use: 'microsecond', ratio: 1 },
-            millisecond: { use: 'microsecond', ratio: 1000 },
-            second: { use: 'millisecond', ratio: 1000 },
-            minute: { use: 'second', ratio: 60 },
-            hour: { use: 'minute', ratio: 60 },
-            day: { use: 'hour', ratio: 24 },
-            week: { use: 'day', ratio: 7 },
+            'microsecond': { use: 'microsecond', ratio: 1 },
+            'millisecond': { use: 'microsecond', ratio: 1000 },
+            'second': { use: 'millisecond', ratio: 1000 },
+            'minute': { use: 'second', ratio: 60 },
+            'hour': { use: 'minute', ratio: 60 },
+            'day': { use: 'hour', ratio: 24 },
+            'week': { use: 'day', ratio: 7 },
         };
         let lVal = from.value;
         let rVal = to.value;

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-export type { QueryDataRow, Fragment, StructDef, StructRelationship, NamedStructDefs, MalloyQueryData, AtomicFieldType as AtomicFieldTypeInner, DateUnit, ExtractUnit, TimestampUnit, TimeFieldType, QueryData, FieldTypeDef, Expr, DialectFragment, TimeValue, FilterExpression, SQLBlock, FieldDef, FilteredAliasedName, PipeSegment, QueryFieldDef, TurtleDef, SearchValueMapResult, SearchIndexResult, ModelDef, Query, NamedQuery, NamedModelObject, ExpressionType, DocumentLocation, DocumentRange, DocumentPosition, } from './model';
+export type { QueryDataRow, Fragment, StructDef, StructRelationship, NamedStructDefs, MalloyQueryData, AtomicFieldType as AtomicFieldTypeInner, DateUnit, ExtractUnit, TimestampUnit, TimeFieldType, QueryData, FieldTypeDef, Expr, DialectFragment, TimeValue, FilterExpression, SQLBlock, FieldDef, FilteredAliasedName, PipeSegment, QueryFieldDef, TurtleDef, SearchValueMapResult, SearchIndexResult, ModelDef, Query, QueryRunStats, NamedQuery, NamedModelObject, ExpressionType, DocumentLocation, DocumentRange, DocumentPosition, } from './model';
 export { Segment, isFilteredAliasedName, flattenQuery, expressionIsCalculation, } from './model';
 export { HighlightType, MalloyTranslator, } from './lang';
 export type { LogMessage, TranslateResponse } from './lang';

package/dist/malloy.d.ts

@@ -121,6 +121,14 @@ export declare class Malloy {
         sqlStruct: SQLBlockStructDef;
         options?: RunSQLOptions;
     }): AsyncIterableIterator<DataRecord>;
+    static estimateQueryCost(params: {
+        connections: LookupConnection<Connection>;
+        preparedResult: PreparedResult;
+    }): Promise<QueryRunStats>;
+    static estimateQueryCost(params: {
+        connections: LookupConnection<Connection>;
+        sqlStruct: SQLBlockStructDef;
+    }): Promise<QueryRunStats>;
 }
 /**
  * A Malloy error, which may contain log messages produced during compilation.
@@ -930,6 +938,12 @@ export declare class QueryMaterializer extends FluentState<PreparedQuery> {
      * @return A promise of the `PreparedQuery`.
      */
     getPreparedQuery(): Promise<PreparedQuery>;
+    /**
+     * Estimates the cost of this loaded `Query`.
+     *
+     * @return The estimated cost of running this loaded query.
+     */
+    estimateQueryCost(): Promise<QueryRunStats>;
 }
 /**
  * An object representing the task of loading a `PreparedResult`, capable of
@@ -986,6 +1000,7 @@ export declare class SQLBlockMaterializer extends FluentState<SQLBlockStructDef>
      * @return A promise to the SQL string.
      */
     getSQL(): Promise<string>;
+    estimateQueryCost(): Promise<QueryRunStats>;
 }
 /**
  * An object representing the task of loading an `Explore`, capable of

package/dist/malloy.js

@@ -281,6 +281,23 @@ class Malloy {
             index += 1;
         }
     }
+    static async estimateQueryCost({ connections, preparedResult, sqlStruct, }) {
+        const sqlBlock = sqlStruct === null || sqlStruct === void 0 ? void 0 : sqlStruct.structSource.sqlBlock;
+        if (!connections) {
+            throw new Error('Internal Error: Connection or LookupConnection<Connection> must be provided.');
+        }
+        const connectionName = (sqlBlock === null || sqlBlock === void 0 ? void 0 : sqlBlock.connection) || (preparedResult === null || preparedResult === void 0 ? void 0 : preparedResult.connectionName);
+        const connection = await connections.lookupConnection(connectionName);
+        if (sqlBlock) {
+            return await connection.estimateQueryCost(sqlBlock === null || sqlBlock === void 0 ? void 0 : sqlBlock.selectStr);
+        }
+        else if (preparedResult) {
+            return await connection.estimateQueryCost(preparedResult === null || preparedResult === void 0 ? void 0 : preparedResult.sql);
+        }
+        else {
+            throw new Error('Internal error: sqlStruct or preparedResult must be provided.');
+        }
+    }
 }
 exports.Malloy = Malloy;
 /**
@@ -1878,6 +1895,16 @@ class QueryMaterializer extends FluentState {
     getPreparedQuery() {
         return this.materialize();
     }
+    /**
+     * Estimates the cost of this loaded `Query`.
+     *
+     * @return The estimated cost of running this loaded query.
+     */
+    async estimateQueryCost() {
+        const connections = this.runtime.connections;
+        const preparedResult = await this.getPreparedResult();
+        return Malloy.estimateQueryCost({ connections, preparedResult });
+    }
 }
 exports.QueryMaterializer = QueryMaterializer;
 /**
@@ -1967,6 +1994,11 @@ class SQLBlockMaterializer extends FluentState {
         const sqlStruct = await this.getSQLBlock();
         return sqlStruct.structSource.sqlBlock.selectStr;
     }
+    async estimateQueryCost() {
+        const connections = this.runtime.connections;
+        const sqlStruct = await this.getSQLBlock();
+        return Malloy.estimateQueryCost({ connections, sqlStruct });
+    }
 }
 exports.SQLBlockMaterializer = SQLBlockMaterializer;
 /**

package/dist/runtime_types.d.ts

@@ -1,5 +1,5 @@
 import { RunSQLOptions } from './run_sql_options';
-import { MalloyQueryData, QueryDataRow, SQLBlock, StructDef } from './model/malloy_types';
+import { MalloyQueryData, QueryDataRow, QueryRunStats, SQLBlock, StructDef } from './model/malloy_types';
 /**
  * The contents of a Malloy query document.
  */
@@ -77,6 +77,7 @@ export interface Connection extends InfoConnection {
     canPersist(): this is PersistSQLResults;
     canStream(): this is StreamingConnection;
     close(): Promise<void>;
+    estimateQueryCost(sqlCommand: string): Promise<QueryRunStats>;
 }
 export interface TestableConnection extends Connection {
     test(): Promise<void>;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@malloydata/malloy",
-  "version": "0.0.38-dev230601002801",
+  "version": "0.0.38-dev230605182241",
   "license": "MIT",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",