@514labs/moose-lib 0.6.459 → 0.6.460

package/dist/moose-runner.js

@@ -553,7 +553,7 @@ var init_secrets = __esm({
   }
 });
 
-// src/consumption-apis/helpers.ts
+// src/consumption-apis/query-client.ts
 function formatElapsedTime(ms) {
   if (ms < 1e3) {
     return `${Math.round(ms)} ms`;
@@ -566,6 +566,68 @@ function formatElapsedTime(ms) {
   const remainingSeconds = seconds % 60;
   return `${minutes} minutes and ${remainingSeconds.toFixed(2)} seconds`;
 }
+var import_node_crypto3, import_perf_hooks, QueryClient;
+var init_query_client = __esm({
+  "src/consumption-apis/query-client.ts"() {
+    "use strict";
+    import_node_crypto3 = require("crypto");
+    import_perf_hooks = require("perf_hooks");
+    init_sqlHelpers();
+    QueryClient = class {
+      client;
+      query_id_prefix;
+      rowPolicyOptions;
+      constructor(client, query_id_prefix, rowPolicyOptions) {
+        this.client = client;
+        this.query_id_prefix = query_id_prefix;
+        this.rowPolicyOptions = rowPolicyOptions;
+      }
+      async execute(sql3) {
+        const [query, query_params] = toQuery(sql3);
+        console.log(`[QueryClient] | Query: ${toQueryPreview(sql3)}`);
+        const start = import_perf_hooks.performance.now();
+        const result = await this.client.query({
+          query,
+          query_params,
+          format: "JSONEachRow",
+          query_id: this.query_id_prefix + (0, import_node_crypto3.randomUUID)(),
+          // Note: wait_end_of_query deliberately NOT set here as this is used for SELECT queries
+          // where response buffering would harm streaming performance and concurrency
+          clickhouse_settings: {
+            asterisk_include_materialized_columns: 1,
+            asterisk_include_alias_columns: 1,
+            ...this.rowPolicyOptions?.clickhouse_settings
+          },
+          ...this.rowPolicyOptions && {
+            role: this.rowPolicyOptions.role
+          }
+        });
+        const elapsedMs = import_perf_hooks.performance.now() - start;
+        console.log(
+          `[QueryClient] | Query completed: ${formatElapsedTime(elapsedMs)}`
+        );
+        return result;
+      }
+      async command(sql3) {
+        const [query, query_params] = toQuery(sql3);
+        console.log(`[QueryClient] | Command: ${toQueryPreview(sql3)}`);
+        const start = import_perf_hooks.performance.now();
+        const result = await this.client.command({
+          query,
+          query_params,
+          query_id: this.query_id_prefix + (0, import_node_crypto3.randomUUID)()
+        });
+        const elapsedMs = import_perf_hooks.performance.now() - start;
+        console.log(
+          `[QueryClient] | Command completed: ${formatElapsedTime(elapsedMs)}`
+        );
+        return result;
+      }
+    };
+  }
+});
+
+// src/consumption-apis/helpers.ts
 function buildRowPolicyOptionsFromClaims(config, claims) {
   const clickhouse_settings = /* @__PURE__ */ Object.create(null);
   for (const [settingName, claimName] of Object.entries(config)) {
@@ -612,16 +674,16 @@ async function getTemporalClient(temporalUrl, namespace, clientCert, clientKey,
     return void 0;
   }
 }
-var import_client2, import_node_crypto3, import_perf_hooks, fs, MooseClient, MOOSE_RLS_ROLE, MOOSE_RLS_USER, MOOSE_RLS_SETTING_PREFIX, QueryClient, WorkflowClient;
+var import_client2, import_node_crypto4, fs, MooseClient, MOOSE_RLS_ROLE, MOOSE_RLS_USER, MOOSE_RLS_SETTING_PREFIX, WorkflowClient;
 var init_helpers = __esm({
   "src/consumption-apis/helpers.ts"() {
     "use strict";
     import_client2 = require("@temporalio/client");
-    import_node_crypto3 = require("crypto");
-    import_perf_hooks = require("perf_hooks");
+    import_node_crypto4 = require("crypto");
     fs = __toESM(require("fs"));
     init_internal();
     init_sqlHelpers();
+    init_query_client();
     MooseClient = class {
       query;
       workflow;
@@ -633,57 +695,6 @@ var init_helpers = __esm({
     MOOSE_RLS_ROLE = "moose_rls_role";
     MOOSE_RLS_USER = "moose_rls_user";
     MOOSE_RLS_SETTING_PREFIX = "SQL_moose_rls_";
-    QueryClient = class {
-      client;
-      query_id_prefix;
-      rowPolicyOptions;
-      constructor(client, query_id_prefix, rowPolicyOptions) {
-        this.client = client;
-        this.query_id_prefix = query_id_prefix;
-        this.rowPolicyOptions = rowPolicyOptions;
-      }
-      async execute(sql3) {
-        const [query, query_params] = toQuery(sql3);
-        console.log(`[QueryClient] | Query: ${toQueryPreview(sql3)}`);
-        const start = import_perf_hooks.performance.now();
-        const result = await this.client.query({
-          query,
-          query_params,
-          format: "JSONEachRow",
-          query_id: this.query_id_prefix + (0, import_node_crypto3.randomUUID)(),
-          // Note: wait_end_of_query deliberately NOT set here as this is used for SELECT queries
-          // where response buffering would harm streaming performance and concurrency
-          clickhouse_settings: {
-            asterisk_include_materialized_columns: 1,
-            asterisk_include_alias_columns: 1,
-            ...this.rowPolicyOptions?.clickhouse_settings
-          },
-          ...this.rowPolicyOptions && {
-            role: this.rowPolicyOptions.role
-          }
-        });
-        const elapsedMs = import_perf_hooks.performance.now() - start;
-        console.log(
-          `[QueryClient] | Query completed: ${formatElapsedTime(elapsedMs)}`
-        );
-        return result;
-      }
-      async command(sql3) {
-        const [query, query_params] = toQuery(sql3);
-        console.log(`[QueryClient] | Command: ${toQueryPreview(sql3)}`);
-        const start = import_perf_hooks.performance.now();
-        const result = await this.client.command({
-          query,
-          query_params,
-          query_id: this.query_id_prefix + (0, import_node_crypto3.randomUUID)()
-        });
-        const elapsedMs = import_perf_hooks.performance.now() - start;
-        console.log(
-          `[QueryClient] | Command completed: ${formatElapsedTime(elapsedMs)}`
-        );
-        return result;
-      }
-    };
     WorkflowClient = class {
       client;
       constructor(temporalClient) {
@@ -766,7 +777,7 @@ var init_helpers = __esm({
       processInputData(name, input_data) {
         let workflowId = name;
         if (input_data) {
-          const hash = (0, import_node_crypto3.createHash)("sha256").update(JSON.stringify(input_data)).digest("hex").slice(0, 16);
+          const hash = (0, import_node_crypto4.createHash)("sha256").update(JSON.stringify(input_data)).digest("hex").slice(0, 16);
           workflowId = `${name}-${hash}`;
         }
         return [input_data, workflowId];