@tdengine/websocket 3.0.0 → 3.1.0

package/lib/example/basicSchemaless.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=basicSchemaless.d.ts.map

package/lib/example/basicSchemaless.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"basicSchemaless.d.ts","sourceRoot":"","sources":["../../example/basicSchemaless.ts"],"names":[],"mappings":""}

package/lib/example/basicSchemaless.js

@@ -0,0 +1,47 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const config_1 = require("../src/common/config");
+const wsProto_1 = require("../src/sql/wsProto");
+const src_1 = require("../src");
+let dsn = 'ws://root:taosdata@localhost:6041';
+let db = 'power';
+let influxdbData = "st,t1=3i64,t2=4f64,t3=\"t3\" c1=3i64,c3=L\"passit\",c2=false,c4=4f64 1626006833639000000";
+let telnetData = "stb0_0 1626006833 4 host=host0 interface=eth0";
+let jsonData = "{\"metric\": \"meter_current\",\"timestamp\": 1626846400,\"value\": 10.3, \"tags\": {\"groupid\": 2, \"location\": \"California.SanFrancisco\", \"id\": \"d1001\"}}";
+const dropDB = `drop database if exists ${db}`;
+async function Prepare() {
+    let conf = new config_1.WSConfig(dsn);
+    conf.setUser('root');
+    conf.setPwd('taosdata');
+    let wsSql = await (0, src_1.sqlConnect)(conf);
+    const topics = ['pwer_meters_topic'];
+    let dropTopic = `DROP TOPIC IF EXISTS ${topics[0]};`;
+    await wsSql.exec(dropTopic);
+    await wsSql.exec(dropDB);
+    await wsSql.exec('create database if not exists power KEEP 3650 DURATION 10 BUFFER 16 WAL_LEVEL 1;');
+    await wsSql.exec('CREATE STABLE if not exists power.meters (ts timestamp, current float, voltage int, phase float) TAGS (location binary(64), groupId int);');
+    wsSql.close();
+}
+(async () => {
+    let wsSchemaless = null;
+    try {
+        let conf = new config_1.WSConfig(dsn);
+        conf.setUser('root');
+        conf.setPwd('taosdata');
+        conf.setDb('power');
+        wsSchemaless = await (0, src_1.sqlConnect)(conf);
+        await wsSchemaless.schemalessInsert([influxdbData], wsProto_1.SchemalessProto.InfluxDBLineProtocol, wsProto_1.Precision.NANO_SECONDS, 0);
+        await wsSchemaless.schemalessInsert([telnetData], wsProto_1.SchemalessProto.OpenTSDBTelnetLineProtocol, wsProto_1.Precision.SECONDS, 0);
+        await wsSchemaless.schemalessInsert([jsonData], wsProto_1.SchemalessProto.OpenTSDBJsonFormatProtocol, wsProto_1.Precision.SECONDS, 0);
+        wsSchemaless.close();
+    }
+    catch (e) {
+        console.error(e);
+    }
+    finally {
+        if (wsSchemaless) {
+            await wsSchemaless.close();
+        }
+        (0, src_1.destroy)();
+    }
+})();

package/lib/example/basicSql.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=basicSql.d.ts.map

package/lib/example/basicSql.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"basicSql.d.ts","sourceRoot":"","sources":["../../example/basicSql.ts"],"names":[],"mappings":""}

package/lib/example/basicSql.js

@@ -0,0 +1,52 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const config_1 = require("../src/common/config");
+const src_1 = require("../src");
+let dsn = 'ws://root:taosdata@localhost:6041';
+(async () => {
+    let wsSql = null;
+    let wsRows = null;
+    let reqId = 0;
+    try {
+        (0, src_1.setLogLevel)("debug");
+        let conf = new config_1.WSConfig(dsn);
+        conf.setUser('root');
+        conf.setPwd('taosdata');
+        wsSql = await (0, src_1.sqlConnect)(conf);
+        let version = await wsSql.version();
+        console.log(version);
+        let taosResult = await wsSql.exec('show databases', reqId++);
+        console.log(taosResult);
+        taosResult = await wsSql.exec('create database if not exists power KEEP 3650 DURATION 10 BUFFER 16 WAL_LEVEL 1;', reqId++);
+        console.log(taosResult);
+        taosResult = await wsSql.exec('use power', reqId++);
+        console.log(taosResult);
+        taosResult = await wsSql.exec('CREATE STABLE if not exists meters (ts timestamp, current float, voltage int, phase float) TAGS (location binary(64), groupId int);', reqId++);
+        console.log(taosResult);
+        taosResult = await wsSql.exec('describe meters', reqId++);
+        console.log(taosResult);
+        taosResult = await wsSql.exec('INSERT INTO d1001 USING meters TAGS ("California.SanFrancisco", 3) VALUES (NOW, 10.2, 219, 0.32)', reqId++);
+        console.log(taosResult);
+        wsRows = await wsSql.query('select * from meters', reqId++);
+        let meta = wsRows.getMeta();
+        console.log("wsRow:meta:=>", meta);
+        while (await wsRows.next()) {
+            let result = wsRows.getData();
+            console.log('queryRes.Scan().then=>', result);
+        }
+        wsRows.close();
+    }
+    catch (err) {
+        console.error(err.code, err.message);
+    }
+    finally {
+        if (wsRows) {
+            await wsRows.close();
+        }
+        if (wsSql) {
+            await wsSql.close();
+        }
+        (0, src_1.destroy)();
+        console.log("finish!");
+    }
+})();

package/lib/example/basicStmt.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=basicStmt.d.ts.map

package/lib/example/basicStmt.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"basicStmt.d.ts","sourceRoot":"","sources":["../../example/basicStmt.ts"],"names":[],"mappings":""}

package/lib/example/basicStmt.js

@@ -0,0 +1,58 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const config_1 = require("../src/common/config");
+const src_1 = require("../src");
+let db = 'power';
+let stable = 'meters';
+let tags = ['California.SanFrancisco', 3];
+let multi = [
+    [1706786044994, 1706786044995, 1706786044996],
+    [10.2, 10.3, 10.4],
+    [292, 293, 294],
+    [0.32, 0.33, 0.34],
+];
+let dsn = 'ws://root:taosdata@localhost:6041';
+async function Prepare() {
+    let conf = new config_1.WSConfig(dsn);
+    let wsSql = await (0, src_1.sqlConnect)(conf);
+    await wsSql.exec(`create database if not exists ${db} KEEP 3650 DURATION 10 BUFFER 16 WAL_LEVEL 1;`);
+    await wsSql.exec(`CREATE STABLE if not exists ${db}.${stable} (ts timestamp, current float, voltage int, phase float) TAGS (location binary(64), groupId int);`);
+    wsSql.close();
+}
+(async () => {
+    let stmt = null;
+    let connector = null;
+    try {
+        await Prepare();
+        let wsConf = new config_1.WSConfig(dsn);
+        wsConf.setDb(db);
+        connector = await (0, src_1.sqlConnect)(wsConf);
+        stmt = await connector.stmtInit();
+        await stmt.prepare(`INSERT INTO ? USING ${db}.${stable} (location, groupId) TAGS (?, ?) VALUES (?, ?, ?, ?)`);
+        await stmt.setTableName('d1001');
+        let tagParams = stmt.newStmtParam();
+        tagParams.setVarchar([tags[0]]);
+        tagParams.setInt([tags[1]]);
+        await stmt.setTags(tagParams);
+        let bindParams = stmt.newStmtParam();
+        bindParams.setTimestamp(multi[0]);
+        bindParams.setFloat(multi[1]);
+        bindParams.setInt(multi[2]);
+        bindParams.setFloat(multi[3]);
+        await stmt.bind(bindParams);
+        await stmt.batch();
+        await stmt.exec();
+    }
+    catch (e) {
+        console.error(e);
+    }
+    finally {
+        if (stmt) {
+            await stmt.close();
+        }
+        if (connector) {
+            await connector.close();
+        }
+        (0, src_1.destroy)();
+    }
+})();

package/lib/example/basicTmq.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=basicTmq.d.ts.map

package/lib/example/basicTmq.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"basicTmq.d.ts","sourceRoot":"","sources":["../../example/basicTmq.ts"],"names":[],"mappings":""}

package/lib/example/basicTmq.js

@@ -0,0 +1,71 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const config_1 = require("../src/common/config");
+const constant_1 = require("../src/tmq/constant");
+const src_1 = require("../src");
+const stable = 'meters';
+const db = 'power';
+const topics = ['pwer_meters_topic'];
+let dropTopic = `DROP TOPIC IF EXISTS ${topics[0]};`;
+let configMap = new Map([
+    [constant_1.TMQConstants.GROUP_ID, "gId"],
+    [constant_1.TMQConstants.CONNECT_USER, "root"],
+    [constant_1.TMQConstants.CONNECT_PASS, "taosdata"],
+    [constant_1.TMQConstants.AUTO_OFFSET_RESET, "earliest"],
+    [constant_1.TMQConstants.CLIENT_ID, 'test_tmq_client'],
+    [constant_1.TMQConstants.WS_URL, 'ws://localhost:6041'],
+    [constant_1.TMQConstants.ENABLE_AUTO_COMMIT, 'true'],
+    [constant_1.TMQConstants.AUTO_COMMIT_INTERVAL_MS, '1000']
+]);
+let dsn = 'ws://root:taosdata@localhost:6041';
+async function Prepare() {
+    let conf = new config_1.WSConfig(dsn);
+    const createDB = `create database if not exists ${db} KEEP 3650 DURATION 10 BUFFER 16 WAL_LEVEL 1;`;
+    const createStable = `CREATE STABLE if not exists ${db}.${stable} (ts timestamp, current float, voltage int, phase float) TAGS (location binary(64), groupId int);`;
+    let createTopic = `create topic if not exists ${topics[0]} as select * from ${db}.${stable}`;
+    const useDB = `use ${db}`;
+    let ws = await (0, src_1.sqlConnect)(conf);
+    await ws.exec(createDB);
+    await ws.exec(useDB);
+    await ws.exec(createStable);
+    await ws.exec(createTopic);
+    for (let i = 0; i < 10; i++) {
+        await ws.exec(`INSERT INTO d1001 USING ${stable} (location, groupId) TAGS ("California.SanFrancisco", 3) VALUES (NOW, ${10 + i}, ${200 + i}, ${0.32 + i})`);
+    }
+    ws.close();
+}
+(async () => {
+    let consumer = null;
+    try {
+        await Prepare();
+        consumer = await (0, src_1.tmqConnect)(configMap);
+        await consumer.subscribe(topics);
+        for (let i = 0; i < 5; i++) {
+            let res = await consumer.poll(500);
+            for (let [key, value] of res) {
+                console.log(key, value);
+            }
+            if (res.size == 0) {
+                break;
+            }
+            await consumer.commit();
+        }
+        let assignment = await consumer.assignment();
+        console.log(assignment);
+        await consumer.seekToBeginning(assignment);
+        assignment = await consumer.assignment();
+        for (let i in assignment) {
+            console.log("seek after:", assignment[i]);
+        }
+        await consumer.unsubscribe();
+    }
+    catch (e) {
+        console.error(e);
+    }
+    finally {
+        if (consumer) {
+            await consumer.close();
+        }
+        (0, src_1.destroy)();
+    }
+})();

package/lib/index.d.ts

@@ -0,0 +1,26 @@
+export * from "./src/client/wsClient";
+export * from "./src/client/wsConnector";
+export * from "./src/client/wsConnectorPool";
+export * from "./src/client/wsEventCallback";
+export * from "./src/client/wsResponse";
+export * from "./src/common/config";
+export * from "./src/common/constant";
+export * from "./src/common/log";
+export * from "./src/common/reqid";
+export * from "./src/common/taosResult";
+export * from "./src/common/ut8Helper";
+export * from "./src/common/utils";
+export * from "./src/common/wsError";
+export * from "./src/common/wsOptions";
+export * from "./src/index";
+export * from "./src/sql/wsProto";
+export * from "./src/sql/wsRows";
+export * from "./src/sql/wsSql";
+export * from "./src/stmt/wsParams";
+export * from "./src/stmt/wsProto";
+export * from "./src/stmt/wsStmt";
+export * from "./src/tmq/config";
+export * from "./src/tmq/constant";
+export * from "./src/tmq/tmqResponse";
+export * from "./src/tmq/wsTmq";
+//# sourceMappingURL=index.d.ts.map

package/lib/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AAAA,cAAc,uBAAuB,CAAA;AACrC,cAAc,0BAA0B,CAAA;AACxC,cAAc,8BAA8B,CAAA;AAC5C,cAAc,8BAA8B,CAAA;AAC5C,cAAc,yBAAyB,CAAA;AACvC,cAAc,qBAAqB,CAAA;AACnC,cAAc,uBAAuB,CAAA;AACrC,cAAc,kBAAkB,CAAA;AAChC,cAAc,oBAAoB,CAAA;AAClC,cAAc,yBAAyB,CAAA;AACvC,cAAc,wBAAwB,CAAA;AACtC,cAAc,oBAAoB,CAAA;AAClC,cAAc,sBAAsB,CAAA;AACpC,cAAc,wBAAwB,CAAA;AACtC,cAAc,aAAa,CAAA;AAC3B,cAAc,mBAAmB,CAAA;AACjC,cAAc,kBAAkB,CAAA;AAChC,cAAc,iBAAiB,CAAA;AAC/B,cAAc,qBAAqB,CAAA;AACnC,cAAc,oBAAoB,CAAA;AAClC,cAAc,mBAAmB,CAAA;AACjC,cAAc,kBAAkB,CAAA;AAChC,cAAc,oBAAoB,CAAA;AAClC,cAAc,uBAAuB,CAAA;AACrC,cAAc,iBAAiB,CAAA"}

package/lib/index.js

@@ -0,0 +1,41 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./src/client/wsClient"), exports);
+__exportStar(require("./src/client/wsConnector"), exports);
+__exportStar(require("./src/client/wsConnectorPool"), exports);
+__exportStar(require("./src/client/wsEventCallback"), exports);
+__exportStar(require("./src/client/wsResponse"), exports);
+__exportStar(require("./src/common/config"), exports);
+__exportStar(require("./src/common/constant"), exports);
+__exportStar(require("./src/common/log"), exports);
+__exportStar(require("./src/common/reqid"), exports);
+__exportStar(require("./src/common/taosResult"), exports);
+__exportStar(require("./src/common/ut8Helper"), exports);
+__exportStar(require("./src/common/utils"), exports);
+__exportStar(require("./src/common/wsError"), exports);
+__exportStar(require("./src/common/wsOptions"), exports);
+__exportStar(require("./src/index"), exports);
+__exportStar(require("./src/sql/wsProto"), exports);
+__exportStar(require("./src/sql/wsRows"), exports);
+__exportStar(require("./src/sql/wsSql"), exports);
+__exportStar(require("./src/stmt/wsParams"), exports);
+__exportStar(require("./src/stmt/wsProto"), exports);
+__exportStar(require("./src/stmt/wsStmt"), exports);
+__exportStar(require("./src/tmq/config"), exports);
+__exportStar(require("./src/tmq/constant"), exports);
+__exportStar(require("./src/tmq/tmqResponse"), exports);
+__exportStar(require("./src/tmq/wsTmq"), exports);

package/lib/src/client/wsClient.d.ts

@@ -0,0 +1,22 @@
+import { TaosResult } from '../common/taosResult';
+import { WSQueryResponse, WSFetchResponse } from './wsResponse';
+export declare class WsClient {
+    private _wsConnector?;
+    private _timeout?;
+    private readonly _url;
+    constructor(url: URL, timeout?: number | undefined | null);
+    connect(database?: string | undefined | null): Promise<void>;
+    execNoResp(queryMsg: string): Promise<void>;
+    exec(queryMsg: string, bSqlQuery?: boolean): Promise<any>;
+    sendBinaryMsg(reqId: bigint, action: string, message: ArrayBuffer, bSqlQuery?: boolean): Promise<any>;
+    getState(): number;
+    ready(): Promise<void>;
+    fetch(res: WSQueryResponse): Promise<WSFetchResponse>;
+    fetchBlock(fetchResponse: WSFetchResponse, taosResult: TaosResult): Promise<TaosResult>;
+    sendMsg(msg: string): Promise<any>;
+    freeResult(res: WSQueryResponse): Promise<unknown>;
+    version(): Promise<string>;
+    close(): Promise<void>;
+    checkURL(url: URL): void;
+}
+//# sourceMappingURL=wsClient.d.ts.map

package/lib/src/client/wsClient.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"wsClient.d.ts","sourceRoot":"","sources":["../../../src/client/wsClient.ts"],"names":[],"mappings":"AAGA,OAAO,EAA2B,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAE3E,OAAO,EAGH,eAAe,EACf,eAAe,EAClB,MAAM,cAAc,CAAC;AAKtB,qBAAa,QAAQ;IACjB,OAAO,CAAC,YAAY,CAAC,CAAqB;IAC1C,OAAO,CAAC,QAAQ,CAAC,CAA2B;IAC5C,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAK;gBAEd,GAAG,EAAE,GAAG,EAAE,OAAQ,CAAC,EAAC,MAAM,GAAG,SAAS,GAAG,IAAI;IAOnD,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAqC5D,UAAU,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAU3C,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,GAAC,OAAc,GAAG,OAAO,CAAC,GAAG,CAAC;IAuB9D,aAAa,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAC,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,SAAS,GAAC,OAAc,GAAG,OAAO,CAAC,GAAG,CAAC;IAsB/G,QAAQ;IAQF,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAetB,KAAK,CAAC,GAAG,EAAE,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;IA0BrD,UAAU,CAAC,aAAa,EAAE,eAAe,EAAE,UAAU,EAAE,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;IA2BvF,OAAO,CAAC,GAAG,EAAC,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC;IAajC,UAAU,CAAC,GAAG,EAAE,eAAe;IAqB/B,OAAO,IAAI,OAAO,CAAC,MAAM,CAAC;IA4B1B,KAAK,IAAG,OAAO,CAAC,IAAI,CAAC;IAS3B,QAAQ,CAAC,GAAG,EAAE,GAAG;CASpB"}

package/lib/src/client/wsClient.js

@@ -0,0 +1,253 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WsClient = void 0;
+const json_bigint_1 = __importDefault(require("json-bigint"));
+const wsConnectorPool_1 = require("./wsConnectorPool");
+const taosResult_1 = require("../common/taosResult");
+const wsError_1 = require("../common/wsError");
+const wsResponse_1 = require("./wsResponse");
+const reqid_1 = require("../common/reqid");
+const log_1 = __importDefault(require("../common/log"));
+class WsClient {
+    constructor(url, timeout) {
+        this.checkURL(url);
+        this._url = url;
+        this._timeout = timeout;
+    }
+    async connect(database) {
+        let _db = this._url.pathname.split('/')[3];
+        if (database) {
+            _db = database;
+        }
+        let connMsg = {
+            action: 'conn',
+            args: {
+                req_id: reqid_1.ReqId.getReqID(),
+                user: this._url.username,
+                password: this._url.password,
+                db: _db,
+            },
+        };
+        this._wsConnector = await wsConnectorPool_1.WebSocketConnectionPool.instance().getConnection(this._url, this._timeout);
+        if (this._wsConnector.readyState() > 0) {
+            return;
+        }
+        try {
+            await this._wsConnector.ready();
+            let result = await this._wsConnector.sendMsg(JSON.stringify(connMsg));
+            if (result.msg.code == 0) {
+                return;
+            }
+            throw (new wsError_1.WebSocketQueryError(result.msg.code, result.msg.message));
+        }
+        catch (e) {
+            log_1.default.error(e.code, e.message);
+            throw (new wsError_1.TDWebSocketClientError(wsError_1.ErrorCode.ERR_WEBSOCKET_CONNECTION_FAIL, `connection creation failed, url: ${this._url}`));
+        }
+    }
+    async execNoResp(queryMsg) {
+        log_1.default.debug('[wsQueryInterface.query.queryMsg]===>' + queryMsg);
+        if (this._wsConnector && this._wsConnector.readyState() > 0) {
+            await this._wsConnector.sendMsgNoResp(queryMsg);
+            return;
+        }
+        throw (new wsError_1.TDWebSocketClientError(wsError_1.ErrorCode.ERR_CONNECTION_CLOSED, "invalid websocket connect"));
+    }
+    // need to construct Response.
+    async exec(queryMsg, bSqlQuery = true) {
+        return new Promise((resolve, reject) => {
+            log_1.default.debug('[wsQueryInterface.query.queryMsg]===>' + queryMsg);
+            if (this._wsConnector && this._wsConnector.readyState() > 0) {
+                this._wsConnector.sendMsg(queryMsg).then((e) => {
+                    if (e.msg.code == 0) {
+                        if (bSqlQuery) {
+                            resolve(new wsResponse_1.WSQueryResponse(e));
+                        }
+                        else {
+                            resolve(e);
+                        }
+                    }
+                    else {
+                        reject(new wsError_1.WebSocketInterfaceError(e.msg.code, e.msg.message));
+                    }
+                }).catch((e) => { reject(e); });
+            }
+            else {
+                reject(new wsError_1.TDWebSocketClientError(wsError_1.ErrorCode.ERR_CONNECTION_CLOSED, "invalid websocket connect"));
+            }
+        });
+    }
+    // need to construct Response.
+    async sendBinaryMsg(reqId, action, message, bSqlQuery = true) {
+        return new Promise((resolve, reject) => {
+            if (this._wsConnector && this._wsConnector.readyState() > 0) {
+                this._wsConnector.sendBinaryMsg(reqId, action, message).then((e) => {
+                    if (e.msg.code == 0) {
+                        if (bSqlQuery) {
+                            resolve(new wsResponse_1.WSQueryResponse(e));
+                        }
+                        else {
+                            resolve(e);
+                        }
+                    }
+                    else {
+                        reject(new wsError_1.WebSocketInterfaceError(e.msg.code, e.msg.message));
+                    }
+                }).catch((e) => { reject(e); });
+            }
+            else {
+                reject(new wsError_1.TDWebSocketClientError(wsError_1.ErrorCode.ERR_CONNECTION_CLOSED, "invalid websocket connect"));
+            }
+        });
+    }
+    getState() {
+        if (this._wsConnector) {
+            return this._wsConnector.readyState();
+        }
+        return -1;
+    }
+    async ready() {
+        try {
+            this._wsConnector = await wsConnectorPool_1.WebSocketConnectionPool.instance().getConnection(this._url, this._timeout);
+            if (this._wsConnector.readyState() <= 0) {
+                await this._wsConnector.ready();
+            }
+            log_1.default.debug("ready status ", this._url, this._wsConnector.readyState());
+            return;
+        }
+        catch (e) {
+            log_1.default.error(e.code, e.message);
+            throw (new wsError_1.TDWebSocketClientError(wsError_1.ErrorCode.ERR_WEBSOCKET_CONNECTION_FAIL, `connection creation failed, url: ${this._url}`));
+        }
+    }
+    async fetch(res) {
+        let fetchMsg = {
+            action: 'fetch',
+            args: {
+                req_id: reqid_1.ReqId.getReqID(),
+                id: res.id,
+            },
+        };
+        return new Promise((resolve, reject) => {
+            let jsonStr = json_bigint_1.default.stringify(fetchMsg);
+            log_1.default.debug('[wsQueryInterface.fetch.fetchMsg]===>' + jsonStr);
+            if (this._wsConnector && this._wsConnector.readyState() > 0) {
+                this._wsConnector.sendMsg(jsonStr).then((e) => {
+                    if (e.msg.code == 0) {
+                        resolve(new wsResponse_1.WSFetchResponse(e));
+                    }
+                    else {
+                        reject(new wsError_1.WebSocketInterfaceError(e.msg.code, e.msg.message));
+                    }
+                }).catch((e) => { reject(e); });
+            }
+            else {
+                reject(new wsError_1.TDWebSocketClientError(wsError_1.ErrorCode.ERR_CONNECTION_CLOSED, "invalid websocket connect"));
+            }
+        });
+    }
+    async fetchBlock(fetchResponse, taosResult) {
+        let fetchBlockMsg = {
+            action: 'fetch_block',
+            args: {
+                req_id: reqid_1.ReqId.getReqID(),
+                id: fetchResponse.id,
+            },
+        };
+        return new Promise((resolve, reject) => {
+            let jsonStr = json_bigint_1.default.stringify(fetchBlockMsg);
+            log_1.default.debug("[wsQueryInterface.fetchBlock.fetchBlockMsg]===>" + jsonStr);
+            if (this._wsConnector && this._wsConnector.readyState() > 0) {
+                this._wsConnector.sendMsg(jsonStr).then((e) => {
+                    let resp = e;
+                    taosResult.addTotalTime(resp.totalTime);
+                    // if retrieve JSON then reject with message
+                    // else is binary , so parse raw block to TaosResult
+                    (0, taosResult_1.parseBlock)(fetchResponse.rows, new wsResponse_1.WSFetchBlockResponse(resp.msg), taosResult);
+                    resolve(taosResult);
+                }).catch((e) => reject(e));
+            }
+            else {
+                reject(new wsError_1.TDWebSocketClientError(wsError_1.ErrorCode.ERR_CONNECTION_CLOSED, "invalid websocket connect"));
+            }
+        });
+    }
+    async sendMsg(msg) {
+        return new Promise((resolve, reject) => {
+            log_1.default.debug("[wsQueryInterface.sendMsg]===>" + msg);
+            if (this._wsConnector && this._wsConnector.readyState() > 0) {
+                this._wsConnector.sendMsg(msg).then((e) => {
+                    resolve(e);
+                }).catch((e) => reject(e));
+            }
+            else {
+                reject(new wsError_1.TDWebSocketClientError(wsError_1.ErrorCode.ERR_CONNECTION_CLOSED, "invalid websocket connect"));
+            }
+        });
+    }
+    async freeResult(res) {
+        let freeResultMsg = {
+            action: 'free_result',
+            args: {
+                req_id: reqid_1.ReqId.getReqID(),
+                id: res.id,
+            },
+        };
+        return new Promise((resolve, reject) => {
+            let jsonStr = json_bigint_1.default.stringify(freeResultMsg);
+            log_1.default.debug("[wsQueryInterface.freeResult.freeResultMsg]===>" + jsonStr);
+            if (this._wsConnector && this._wsConnector.readyState() > 0) {
+                this._wsConnector.sendMsg(jsonStr, false)
+                    .then((e) => { resolve(e); })
+                    .catch((e) => reject(e));
+            }
+            else {
+                reject(new wsError_1.TDWebSocketClientError(wsError_1.ErrorCode.ERR_CONNECTION_CLOSED, "invalid websocket connect"));
+            }
+        });
+    }
+    async version() {
+        let versionMsg = {
+            action: 'version',
+            args: {
+                req_id: reqid_1.ReqId.getReqID()
+            },
+        };
+        if (this._wsConnector) {
+            try {
+                if (this._wsConnector.readyState() <= 0) {
+                    await this._wsConnector.ready();
+                }
+                let result = await this._wsConnector.sendMsg(json_bigint_1.default.stringify(versionMsg));
+                if (result.msg.code == 0) {
+                    return new wsResponse_1.WSVersionResponse(result).version;
+                }
+                throw (new wsError_1.WebSocketInterfaceError(result.msg.code, result.msg.message));
+            }
+            catch (e) {
+                log_1.default.error(e.code, e.message);
+                throw (new wsError_1.TDWebSocketClientError(wsError_1.ErrorCode.ERR_WEBSOCKET_CONNECTION_FAIL, `connection creation failed, url: ${this._url}`));
+            }
+        }
+        throw (wsError_1.ErrorCode.ERR_CONNECTION_CLOSED, "invalid websocket connect");
+    }
+    async close() {
+        if (this._wsConnector) {
+            await wsConnectorPool_1.WebSocketConnectionPool.instance().releaseConnection(this._wsConnector);
+            this._wsConnector = undefined;
+            // this._wsConnector.close();
+        }
+    }
+    checkURL(url) {
+        // Assert is cloud url
+        if (!url.searchParams.has('token')) {
+            if (!(url.username || url.password)) {
+                throw new wsError_1.WebSocketInterfaceError(wsError_1.ErrorCode.ERR_INVALID_AUTHENTICATION, 'invalid url, password or username needed.');
+            }
+        }
+    }
+}
+exports.WsClient = WsClient;

package/lib/src/client/wsConnector.d.ts

@@ -0,0 +1,17 @@
+export declare class WebSocketConnector {
+    private _wsConn;
+    private _wsURL;
+    _timeout: number;
+    constructor(url: URL, timeout: number | undefined | null);
+    ready(): Promise<unknown>;
+    private _onclose;
+    private _onmessage;
+    close(): void;
+    readyState(): number;
+    sendMsgNoResp(message: string): Promise<void>;
+    sendMsg(message: string, register?: Boolean): Promise<unknown>;
+    sendBinaryMsg(reqId: bigint, action: string, message: ArrayBuffer, register?: Boolean): Promise<unknown>;
+    configTimeout(ms: number): void;
+    getWsURL(): URL;
+}
+//# sourceMappingURL=wsConnector.d.ts.map

package/lib/src/client/wsConnector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"wsConnector.d.ts","sourceRoot":"","sources":["../../../src/client/wsConnector.ts"],"names":[],"mappings":"AAMA,qBAAa,kBAAkB;IAC3B,OAAO,CAAC,OAAO,CAAe;IAC9B,OAAO,CAAC,MAAM,CAAM;IACpB,QAAQ,SAAQ;gBAGJ,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI;IAsBlD,KAAK;YAeG,QAAQ;IAKtB,OAAO,CAAC,UAAU;IA0BlB,KAAK;IASL,UAAU,IAAI,MAAM;IAId,aAAa,CAAC,OAAO,EAAE,MAAM,GAAE,OAAO,CAAC,IAAI,CAAC;IAmB5C,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,GAAE,OAAc;IAuBjD,aAAa,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAC,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,QAAQ,GAAE,OAAc;IAgBzF,aAAa,CAAC,EAAE,EAAE,MAAM;IAIxB,QAAQ,IAAI,GAAG;CAGzB"}