Set query_id to be the consumption path (#1824)

* add query ids * remove console log and accidnetal deletion
514-labs · Oct 10, 2024 · 2d49dfd · 2d49dfd
1 parent 9d3335a
commit 2d49dfd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/packages/ts-moose-lib/src/consumption-apis/helpers.ts b/packages/ts-moose-lib/src/consumption-apis/helpers.ts
@@ -128,9 +128,10 @@ function emptyIfUndefined(value: string | undefined): string {
 
 export class MooseClient {
  client: ClickHouseClient;
-
- constructor(client: ClickHouseClient) {
+ query_id: string;
+ constructor(client: ClickHouseClient, query_id: string) {
  this.client = client;
+ this.query_id = query_id;
  }
 
  async query(sql: Sql) {
@@ -161,6 +162,7 @@ export class MooseClient {
  query,
  query_params,
  format: "JSONEachRow",
+ query_id: this.query_id,
  });
  }
 }

diff --git a/packages/ts-moose-lib/src/consumption-apis/runner.ts b/packages/ts-moose-lib/src/consumption-apis/runner.ts
@@ -92,7 +92,10 @@ const apiHandler =
  const userFuncModule = require(pathName);
 
  const result = await userFuncModule.default(paramsObject, {
- client: new MooseClient(getClickhouseClient(clickhouseConfig)),
+ client: new MooseClient(
+ getClickhouseClient(clickhouseConfig),
+ fileName,
+ ),
  sql: sql,
  jwt: jwtPayload,
  });