@cleverbrush/knex-clickhouse 1.1.7 → 1.1.8

package/dist/index.d.ts

@@ -1,9 +1,21 @@
 import Knex from 'knex';
 import ClickhouseDriver, { type ClickHouseSettings } from '@clickhouse/client';
+import { retry } from '@cleverbrush/async';
+type AdditionalClientOptions = {
+    /**
+     * Options for retrying the query,
+     * by default it will retry 10 times with a delay factor of 2 and a minimum delay of
+     * 100ms and delay randomization of 10%.
+     * and will retry only if the error code is 'ECONNRESET',
+     * you can override this behavior by passing your own options.
+     * If this option is set to `null` then the query will not be retried.
+     */
+    retry?: Parameters<typeof retry>[1];
+};
 export declare class ClickhouseKnexClient extends Knex.Client {
     #private;
     _driver(): typeof ClickhouseDriver;
-    constructor(config?: {});
+    constructor(config?: Knex.Knex.Config<any> & AdditionalClientOptions);
     _escapeBinding(value: any): any;
     /**
      *
@@ -26,4 +38,5 @@ export declare class ClickhouseKnexClient extends Knex.Client {
  *
  * @returns {import('knex').Knex<any, unknown[]>}
  */
-export declare const getClickhouseConnection: (config?: Knex.Knex.Config<any>, clickHouseSettings?: ClickHouseSettings) => Knex.Knex<any, unknown[]>;
+export declare const getClickhouseConnection: (config?: Knex.Knex.Config<any> & AdditionalClientOptions, clickHouseSettings?: ClickHouseSettings) => Knex.Knex<any, unknown[]>;
+export {};

package/dist/index.js

@@ -1,5 +1,7 @@
 import Knex from 'knex';
 import ClickhouseDriver from '@clickhouse/client';
+import { retry } from '@cleverbrush/async';
+import { deepExtend } from '@cleverbrush/deep';
 import { makeEscape } from './makeEscape.js';
 const parseDate = (value) => new Date(Date.parse(value));
 // this function is taken from knex npm package
@@ -28,8 +30,19 @@ export class ClickhouseKnexClient extends Knex.Client {
     _driver() {
         return ClickhouseDriver;
     }
+    #retryOptions;
     constructor(config = {}) {
+        const { retry: retryOptions } = config;
         super(config);
+        this.#retryOptions = retryOptions
+            ? deepExtend({
+                maxRetries: 10,
+                delayFactor: 2,
+                minDelay: 100,
+                delayRandomizationPercent: 0.1,
+                shouldRetry: (error) => error.code === 'ECONNRESET'
+            }, retryOptions)
+            : null;
     }
     #typeParsers = {
         Date: parseDate,
@@ -125,14 +138,18 @@ export class ClickhouseKnexClient extends Knex.Client {
             case 'first':
             case 'pluck':
                 {
-                    const p = await connection.query(queryParams);
+                    const p = (await (this.#retryOptions
+                        ? retry(() => connection.query(queryParams), this.#retryOptions)
+                        : connection.query(queryParams)));
                     response = await p.json();
                     const { data, meta, ...rest } = response;
                     obj.response = [data, meta, { ...rest }];
                 }
                 break;
             default:
-                response = await connection.exec(queryParams);
+                response = await (this.#retryOptions
+                    ? retry(() => connection.exec(queryParams), this.#retryOptions)
+                    : connection.exec(queryParams));
         }
         return obj;
     }

package/package.json

@@ -5,6 +5,8 @@
         "email": "andrew_zol@cleverbrush.com"
     },
     "dependencies": {
+        "@cleverbrush/async": "^1.1.8",
+        "@cleverbrush/deep": "^1.1.8",
         "@clickhouse/client": "^1.0.1"
     },
     "peerDependencies": {
@@ -35,5 +37,5 @@
     },
     "type": "module",
     "types": "./dist/index.d.ts",
-    "version": "1.1.7"
+    "version": "1.1.8"
 }