k6-perf-reporter 1.0.0

package/src/influx-client.ts

@@ -0,0 +1,48 @@
+import { InfluxDB, QueryApi } from "@influxdata/influxdb-client";
+import { InfluxConfig } from "./config";
+
+export interface QueryRow {
+  values: (string | number | boolean | null)[];
+  tableMeta?: {
+    columns: Array<{ label: string }>;
+  };
+}
+
+export class InfluxClient {
+  private influxDB: InfluxDB;
+  private queryApi: QueryApi;
+
+  constructor(config: InfluxConfig) {
+    this.influxDB = new InfluxDB({
+      url: config.url,
+      token: config.token,
+    });
+    this.queryApi = this.influxDB.getQueryApi(config.org);
+  }
+
+  async queryData(
+    flux: string
+  ): Promise<Array<Record<string, string | number | boolean | null>>> {
+    const result: Array<Record<string, string | number | boolean | null>> = [];
+
+    return new Promise<Array<Record<string, string | number | boolean | null>>>(
+      (resolve, reject) => {
+        this.queryApi.queryRows(flux, {
+          next: (row: string[], tableMeta) => {
+            const record: Record<string, string | number | boolean | null> = {};
+            tableMeta.columns.forEach((col, index) => {
+              record[col.label] = row[index];
+            });
+            result.push(record);
+          },
+          error: (error: Error) => {
+            reject(error);
+          },
+          complete: () => {
+            resolve(result);
+          },
+        });
+      }
+    );
+  }
+}