@malloydata/db-trino 0.0.424 → 0.0.426

package/dist/trino_connection.d.ts

@@ -24,7 +24,7 @@ export interface TrinoConnectionConfiguration {
 }
 export type TrinoConnectionOptions = ConnectionConfig;
 export interface BaseRunner {
-    runSQL(sql: string, options: RunSQLOptions): Promise<{
+    runSQL(sql: string, options: RunSQLOptions, queryMetadataComment?: string): Promise<{
         rows: unknown[][];
         columns: {
             name: string;

package/dist/trino_connection.js

@@ -31,11 +31,12 @@ class PrestoRunner {
         }
         this.client = new presto_js_client_1.PrestoClient(prestoClientConfig);
     }
-    async runSQL(sql, options = {}) {
+    async runSQL(sql, options = {}, queryMetadataComment = '') {
         let ret = undefined;
-        const q = options.rowLimit
-            ? `SELECT * FROM(${sql}) LIMIT ${options.rowLimit}`
-            : sql;
+        const q = queryMetadataComment +
+            (options.rowLimit
+                ? `SELECT * FROM(${sql}) LIMIT ${options.rowLimit}`
+                : sql);
         let error = undefined;
         try {
             ret = (await this.client.query(q)) || [];
@@ -79,9 +80,9 @@ class TrinoRunner {
             auth: new trino_client_1.BasicAuth(config.user, config.password || ''),
         });
     }
-    async runSQL(sql, options = {}) {
+    async runSQL(sql, options = {}, queryMetadataComment = '') {
         var _a;
-        const result = await this.client.query(sql);
+        const result = await this.client.query(queryMetadataComment + sql);
         let queryResult = await result.next();
         if (queryResult.value.error) {
             return {
@@ -180,7 +181,7 @@ class TrinoPrestoConnection extends connection_1.BaseConnection {
     // TODO(figutierrez): Use.
     _rowIndex = 0) {
         await this.ensureSetup();
-        const r = await this.client.runSQL(sqlCommand, options);
+        const r = await this.client.runSQL(sqlCommand, options, this.queryMetadataComment(options.queryMetadata));
         if (r.error) {
             throw new Error(r.error);
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@malloydata/db-trino",
-  "version": "0.0.424",
+  "version": "0.0.426",
   "license": "MIT",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -23,7 +23,7 @@
     "prepublishOnly": "npm run build"
   },
   "dependencies": {
-    "@malloydata/malloy": "0.0.424",
+    "@malloydata/malloy": "0.0.426",
     "@prestodb/presto-js-client": "^1.1.0",
     "gaxios": "^4.2.0",
     "luxon": "^3.7.2",