@mastra/clickhouse 1.5.1 → 1.5.2

package/dist/index.js

@@ -514,16 +514,11 @@ var ClickhouseDB = class extends MastraBase {
   }
   async clearTable({ tableName }) {
     try {
-      await this.client.query({
-        query: `TRUNCATE TABLE ${tableName}`,
-        clickhouse_settings: {
-          // Allows to insert serialized JS Dates (such as '2023-12-06T10:54:48.000Z')
-          date_time_input_format: "best_effort",
-          date_time_output_format: "iso",
-          use_client_time_zone: 1,
-          output_format_json_quote_64bit_integers: 0
-        }
+      await this.client.command({ query: `SYSTEM STOP MERGES ${tableName}` });
+      await this.client.command({
+        query: `TRUNCATE TABLE ${tableName}`
       });
+      await this.client.command({ query: `SYSTEM START MERGES ${tableName}` });
     } catch (error) {
       throw new MastraError(
         {
@@ -785,7 +780,7 @@ var BackgroundTasksStorageClickhouse = class extends BackgroundTasksStorage {
     const rows = await result.json();
     return rows.length > 0 ? rowToTask(rows[0]) : null;
   }
-  async listTasks(filter) {
+  buildFilterClause(filter) {
     const conditions = [];
     const params = {};
     if (filter.status) {
@@ -819,6 +814,10 @@ var BackgroundTasksStorageClickhouse = class extends BackgroundTasksStorage {
       params.var_to_date = filter.toDate.toISOString();
     }
     const where = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
+    return { where, params };
+  }
+  async listTasks(filter) {
+    const { where, params } = this.buildFilterClause(filter);
     const countResult = await this.client.query({
       query: `SELECT count() as count FROM ${TABLE_BACKGROUND_TASKS} FINAL ${where}`,
       query_params: params,
@@ -848,19 +847,18 @@ var BackgroundTasksStorageClickhouse = class extends BackgroundTasksStorage {
     return { tasks, total };
   }
   async deleteTask(taskId) {
-    await this.client.query({
-      query: `ALTER TABLE ${TABLE_BACKGROUND_TASKS} DELETE WHERE id = {var_id:String}`,
+    await this.client.command({
+      query: `DELETE FROM ${TABLE_BACKGROUND_TASKS} WHERE id = {var_id:String}`,
       query_params: { var_id: taskId }
     });
   }
   async deleteTasks(filter) {
-    const { tasks } = await this.listTasks(filter);
-    for (const task of tasks) {
-      await this.client.query({
-        query: `ALTER TABLE ${TABLE_BACKGROUND_TASKS} DELETE WHERE id = {var_id:String}`,
-        query_params: { var_id: task.id }
-      });
-    }
+    const { where, params } = this.buildFilterClause(filter);
+    if (!where) return;
+    await this.client.command({
+      query: `DELETE FROM ${TABLE_BACKGROUND_TASKS} ${where}`,
+      query_params: params
+    });
   }
   async getRunningCount() {
     const result = await this.client.query({