@tmlmobilidade/writers 20260121.2332.6 → 20260126.1852.36

package/dist/clickhouse.d.ts

@@ -95,6 +95,35 @@ export declare class ClickHouseWriter<T> {
      * @param orderBy The ORDER BY clause for the table (default: tuple())
      */
     ensureTable(schema?: ClickHouseColumn<T>[], engine?: string, orderBy?: string): Promise<void>;
+    /**
+     * Builds a WHERE clause from a filter object.
+     * Supports simple equality, range queries ($gte, $lte, $gt, $lt), and $in queries.
+     *
+     * @param filter Filter object with column names as keys
+     * @returns WHERE clause string (without the WHERE keyword)
+     */
+    private buildWhereClause;
+    /**
+     * Formats a value for use in a SQL query.
+     */
+    private formatValue;
+    /**
+     * Counts the number of documents in the table that match the given filter.
+     * Supports simple equality and range queries ($gte, $lte, $gt, $lt).
+     *
+     * @param filter Optional WHERE clause conditions (e.g., { operational_date: '2024-01-01' } or { received_at: { $gte: 1234567890 } })
+     * @returns The count of matching documents
+     */
+    countDocuments(filter?: Record<string, unknown>): Promise<number>;
+    /**
+     * Returns distinct values for a given column, optionally filtered.
+     * Supports simple equality and range queries ($gte, $lte, $gt, $lt).
+     *
+     * @param column The column name to get distinct values for
+     * @param filter Optional WHERE clause conditions
+     * @returns Array of distinct values
+     */
+    distinct<K = string>(column: string, filter?: Record<string, unknown>): Promise<K[]>;
     flush(callback?: (data?: T[]) => Promise<void>): Promise<void>;
     /**
      * Write data to the ClickHouse table.

package/dist/clickhouse.js

@@ -65,6 +65,120 @@ export class ClickHouseWriter {
         }
     }
     /* * */
+    /**
+     * Builds a WHERE clause from a filter object.
+     * Supports simple equality, range queries ($gte, $lte, $gt, $lt), and $in queries.
+     *
+     * @param filter Filter object with column names as keys
+     * @returns WHERE clause string (without the WHERE keyword)
+     */
+    buildWhereClause(filter) {
+        if (!filter || Object.keys(filter).length === 0) {
+            return null;
+        }
+        const conditions = [];
+        for (const [key, value] of Object.entries(filter)) {
+            if (value === null || value === undefined) {
+                continue;
+            }
+            // Handle range queries (MongoDB-style operators)
+            if (typeof value === 'object' && !Array.isArray(value)) {
+                const operators = value;
+                for (const [op, opValue] of Object.entries(operators)) {
+                    switch (op) {
+                        case '$gt':
+                            conditions.push(`${key} > ${this.formatValue(opValue)}`);
+                            break;
+                        case '$gte':
+                            conditions.push(`${key} >= ${this.formatValue(opValue)}`);
+                            break;
+                        case '$in':
+                            if (Array.isArray(opValue)) {
+                                const values = opValue.map(v => this.formatValue(v)).join(', ');
+                                conditions.push(`${key} IN (${values})`);
+                            }
+                            break;
+                        case '$lt':
+                            conditions.push(`${key} < ${this.formatValue(opValue)}`);
+                            break;
+                        case '$lte':
+                            conditions.push(`${key} <= ${this.formatValue(opValue)}`);
+                            break;
+                    }
+                }
+            }
+            else {
+                // Simple equality
+                conditions.push(`${key} = ${this.formatValue(value)}`);
+            }
+        }
+        return conditions.length > 0 ? conditions.join(' AND ') : null;
+    }
+    /**
+     * Formats a value for use in a SQL query.
+     */
+    formatValue(value) {
+        if (typeof value === 'string') {
+            return `'${value}'`;
+        }
+        return String(value);
+    }
+    /* * */
+    /**
+     * Counts the number of documents in the table that match the given filter.
+     * Supports simple equality and range queries ($gte, $lte, $gt, $lt).
+     *
+     * @param filter Optional WHERE clause conditions (e.g., { operational_date: '2024-01-01' } or { received_at: { $gte: 1234567890 } })
+     * @returns The count of matching documents
+     */
+    async countDocuments(filter) {
+        try {
+            let query = `SELECT count() as count FROM ${this.params.table}`;
+            const whereClause = this.buildWhereClause(filter);
+            if (whereClause) {
+                query += ` WHERE ${whereClause}`;
+            }
+            const result = await this.client.query({
+                format: 'JSONEachRow',
+                query,
+            });
+            const data = await result.json();
+            return parseInt(data[0]?.count ?? '0', 10);
+        }
+        catch (error) {
+            Logger.error(`CLICKHOUSEWRITER [${this.params.table}]: Error @ countDocuments(): ${error.message}`);
+            throw error;
+        }
+    }
+    /* * */
+    /**
+     * Returns distinct values for a given column, optionally filtered.
+     * Supports simple equality and range queries ($gte, $lte, $gt, $lt).
+     *
+     * @param column The column name to get distinct values for
+     * @param filter Optional WHERE clause conditions
+     * @returns Array of distinct values
+     */
+    async distinct(column, filter) {
+        try {
+            let query = `SELECT DISTINCT ${column} FROM ${this.params.table}`;
+            const whereClause = this.buildWhereClause(filter);
+            if (whereClause) {
+                query += ` WHERE ${whereClause}`;
+            }
+            const result = await this.client.query({
+                format: 'JSONEachRow',
+                query,
+            });
+            const data = await result.json();
+            return data.map(row => row[column]);
+        }
+        catch (error) {
+            Logger.error(`CLICKHOUSEWRITER [${this.params.table}]: Error @ distinct(): ${error.message}`);
+            throw error;
+        }
+    }
+    /* * */
     async flush(callback) {
         try {
             //

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@tmlmobilidade/writers",
-	"version": "20260121.2332.6",
+	"version": "20260126.1852.36",
 	"author": {
 		"email": "iso@tmlmobilidade.pt",
 		"name": "TML-ISO"