@tdengine/websocket 3.2.3 → 3.3.0

package/lib/test/client/wsClient.recovery.test.js

@@ -0,0 +1,122 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const websocket_1 = require("websocket");
+const wsClient_1 = require("@src/client/wsClient");
+const wsConnectorPool_1 = require("@src/client/wsConnectorPool");
+const dsn_1 = require("@src/common/dsn");
+const wsError_1 = require("@src/common/wsError");
+function createMockConnector() {
+    return {
+        readyState: jest.fn(() => websocket_1.w3cwebsocket.OPEN),
+        ready: jest.fn(async () => { }),
+        setSessionRecoveryHook: jest.fn(),
+        sendMsgDirect: jest.fn(async (_message) => ({
+            msg: { code: 0, message: "" }
+        })),
+        sendMsg: jest.fn(async () => ({ msg: { code: 0, message: "" } })),
+        sendMsgNoResp: jest.fn(async () => { }),
+    };
+}
+describe("WsClient recovery hook", () => {
+    afterEach(() => {
+        jest.restoreAllMocks();
+    });
+    test("uses dsn endpoint when creating connection", async () => {
+        const dsn = (0, dsn_1.parse)("ws://root:taosdata@localhost:6041");
+        dsn.endpoint = dsn_1.WS_TMQ_ENDPOINT;
+        const connector = createMockConnector();
+        const getConnectionSpy = jest
+            .spyOn(wsConnectorPool_1.WebSocketConnectionPool.instance(), "getConnection")
+            .mockResolvedValue(connector);
+        const client = new wsClient_1.WsClient(dsn, 4321);
+        await client.ready();
+        expect(getConnectionSpy).toHaveBeenCalledWith(dsn, 4321);
+    });
+    test("runs custom recovery hook without sql conn recovery on tmq path", async () => {
+        const dsn = (0, dsn_1.parse)("ws://root:taosdata@localhost:6041");
+        dsn.endpoint = dsn_1.WS_TMQ_ENDPOINT;
+        const connector = createMockConnector();
+        jest
+            .spyOn(wsConnectorPool_1.WebSocketConnectionPool.instance(), "getConnection")
+            .mockResolvedValue(connector);
+        const customRecoveryHook = jest.fn(async () => { });
+        const client = new wsClient_1.WsClient(dsn, 5000);
+        client.setSessionRecoveryHook(customRecoveryHook);
+        await client.ready();
+        const hookCalls = connector.setSessionRecoveryHook.mock.calls;
+        const hook = hookCalls[hookCalls.length - 1]?.[0];
+        expect(hook).toBeTruthy();
+        await hook();
+        expect(connector.sendMsgDirect).not.toHaveBeenCalled();
+        expect(customRecoveryHook).toHaveBeenCalledTimes(1);
+    });
+    test("keeps sql conn recovery and then runs custom recovery on sql endpoint", async () => {
+        const dsn = (0, dsn_1.parse)("ws://root:taosdata@localhost:6041");
+        const connector = createMockConnector();
+        const callOrder = [];
+        connector.sendMsgDirect.mockImplementation(async (message) => {
+            const action = JSON.parse(message).action;
+            callOrder.push(action);
+            return { msg: { code: 0, message: "" } };
+        });
+        jest
+            .spyOn(wsConnectorPool_1.WebSocketConnectionPool.instance(), "getConnection")
+            .mockResolvedValue(connector);
+        const client = new wsClient_1.WsClient(dsn, 5000);
+        client._connectedDatabase = "db_recovery";
+        client.setSessionRecoveryHook(async () => {
+            callOrder.push("custom");
+        });
+        await client.ready();
+        const hookCalls = connector.setSessionRecoveryHook.mock.calls;
+        const hook = hookCalls[hookCalls.length - 1]?.[0];
+        expect(hook).toBeTruthy();
+        await hook();
+        expect(connector.sendMsgDirect).toHaveBeenCalledTimes(1);
+        const firstCall = connector.sendMsgDirect.mock.calls[0];
+        const connMsg = JSON.parse(firstCall[0]);
+        expect(connMsg.action).toBe("conn");
+        expect(connMsg.args.db).toBe("db_recovery");
+        expect(callOrder).toEqual(["conn", "custom"]);
+    });
+    test("throws WebSocketQueryError when sql recovery direct call fails", async () => {
+        const dsn = (0, dsn_1.parse)("ws://root:taosdata@localhost:6041");
+        const connector = createMockConnector();
+        connector.sendMsgDirect.mockResolvedValue({
+            msg: {
+                code: 9001,
+                message: "conn failed",
+            },
+        });
+        jest
+            .spyOn(wsConnectorPool_1.WebSocketConnectionPool.instance(), "getConnection")
+            .mockResolvedValue(connector);
+        const customRecoveryHook = jest.fn(async () => { });
+        const client = new wsClient_1.WsClient(dsn, 5000);
+        client.setSessionRecoveryHook(customRecoveryHook);
+        await client.ready();
+        const hookCalls = connector.setSessionRecoveryHook.mock.calls;
+        const hook = hookCalls[hookCalls.length - 1]?.[0];
+        expect(hook).toBeTruthy();
+        await expect(hook()).rejects.toBeInstanceOf(wsError_1.WebSocketQueryError);
+        expect(customRecoveryHook).not.toHaveBeenCalled();
+    });
+    test("restores default information_schema during sql recovery when no db provided", async () => {
+        const dsn = (0, dsn_1.parse)("ws://root:taosdata@localhost:6041");
+        const connector = createMockConnector();
+        jest
+            .spyOn(wsConnectorPool_1.WebSocketConnectionPool.instance(), "getConnection")
+            .mockResolvedValue(connector);
+        const client = new wsClient_1.WsClient(dsn, 5000);
+        await client.connect();
+        const hookCalls = connector.setSessionRecoveryHook.mock.calls;
+        const hook = hookCalls[hookCalls.length - 1]?.[0];
+        expect(hook).toBeTruthy();
+        await hook();
+        expect(connector.sendMsgDirect).toHaveBeenCalledTimes(1);
+        const firstCall = connector.sendMsgDirect.mock.calls[0];
+        const connMsg = JSON.parse(firstCall[0]);
+        expect(connMsg.action).toBe("conn");
+        expect(connMsg.args.db).toBe("information_schema");
+    });
+});

package/lib/test/client/wsConnectPool.test.d.ts

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

package/lib/test/client/wsConnectPool.test.d.ts.map

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

package/lib/test/client/wsConnectPool.test.js

@@ -0,0 +1,147 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const wsConnectorPool_1 = require("@src/client/wsConnectorPool");
+const config_1 = require("@src/common/config");
+const reqid_1 = require("@src/common/reqid");
+const wsSql_1 = require("@src/sql/wsSql");
+const constant_1 = require("@src/tmq/constant");
+const wsTmq_1 = require("@src/tmq/wsTmq");
+const utils_1 = require("@test-helpers/utils");
+const log_1 = require("@src/common/log");
+let dsn = `ws://${(0, utils_1.testUsername)()}:${(0, utils_1.testPassword)()}@localhost:6041`;
+let tags = ["California.SanFrancisco", 3];
+let multi = [
+    [1709183268567, 1709183268568, 1709183268569],
+    [10.2, 10.3, 10.4],
+    [292, 293, 294],
+    [0.32, 0.33, 0.34],
+];
+let configMap = new Map([
+    [constant_1.TMQConstants.GROUP_ID, "gId"],
+    [constant_1.TMQConstants.CONNECT_USER, (0, utils_1.testUsername)()],
+    [constant_1.TMQConstants.CONNECT_PASS, (0, utils_1.testPassword)()],
+    [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"],
+]);
+const stable = "meters";
+const db = "power_connect";
+const topics = ["pwer_meters_topic"];
+let createTopic = `create topic if not exists ${topics[0]} as select * from ${db}.${stable}`;
+let stmtIds = [];
+(0, log_1.setLevel)("debug");
+async function connect() {
+    let dsn = `ws://${(0, utils_1.testUsername)()}:${(0, utils_1.testPassword)()}@localhost:6041`;
+    let wsSql = null;
+    let conf = new config_1.WSConfig(dsn);
+    conf.setDb(db);
+    wsSql = await wsSql_1.WsSql.open(conf);
+    expect(wsSql.state()).toBeGreaterThan(0);
+    console.log(await wsSql.version());
+    await wsSql.close();
+}
+async function stmtConnect() {
+    let dsn = `ws://${(0, utils_1.testUsername)()}:${(0, utils_1.testPassword)()}@localhost:6041`;
+    let wsConf = new config_1.WSConfig(dsn, "100.100.100.100");
+    wsConf.setDb(db);
+    let connector = await wsSql_1.WsSql.open(wsConf);
+    let stmt = (await connector.stmtInit());
+    let id = stmt.getStmtId();
+    if (id) {
+        stmtIds.push(id);
+    }
+    expect(stmt).toBeTruthy();
+    await stmt.prepare(`INSERT INTO ? USING ${stable} (location, groupId) TAGS (?, ?) VALUES (?, ?, ?, ?)`);
+    await stmt.setTableName("d1001");
+    await stmt.setJsonTags(tags);
+    let lastTs = 0;
+    const allp = [];
+    for (let i = 0; i < 10; i++) {
+        for (let j = 0; j < multi[0].length; j++) {
+            multi[0][j] = multi[0][0] + j;
+            lastTs = multi[0][j];
+        }
+        allp.push(stmt.jsonBind(multi));
+        multi[0][0] = lastTs + 1;
+    }
+    await Promise.all(allp);
+    await stmt.batch();
+    await stmt.exec();
+    expect(stmt.getLastAffected()).toEqual(30);
+    await stmt.close();
+    await connector.close();
+}
+async function tmqConnect() {
+    let consumer = null;
+    try {
+        consumer = await wsTmq_1.WsConsumer.newConsumer(configMap);
+        await consumer.subscribe(topics);
+        let res = await consumer.poll(100);
+        for (let [key, value] of res) {
+            console.log(key, value.getMeta());
+            let data = value.getData();
+            if (data == null || data.length == 0) {
+                break;
+            }
+            for (let record of data) {
+                console.log(record);
+            }
+        }
+        await consumer.commit();
+        let assignment = await consumer.assignment();
+        console.log(assignment);
+        if (arguments && arguments.length > 0)
+            await consumer.seekToBeginning(assignment);
+        await consumer.unsubscribe();
+    }
+    catch (e) {
+        console.error(e);
+    }
+    finally {
+        if (consumer) {
+            await consumer.close();
+        }
+    }
+}
+beforeAll(async () => {
+    let conf = new config_1.WSConfig(dsn);
+    let ws = await wsSql_1.WsSql.open(conf);
+    await ws.exec(`create database if not exists ${db} KEEP 3650 DURATION 10 BUFFER 16 WAL_LEVEL 1;`);
+    await (0, utils_1.Sleep)(100);
+    await ws.exec(`CREATE STABLE if not exists ${db}.${stable} (ts timestamp, current float, voltage int, phase float) TAGS (location binary(64), groupId int);`);
+    await (0, utils_1.Sleep)(100);
+    await ws.exec(createTopic, reqid_1.ReqId.getReqID());
+    await ws.close();
+});
+describe("TDWebSocket.WsSql()", () => {
+    jest.setTimeout(60 * 1000);
+    test("ReqId", async () => {
+        const allp = [];
+        for (let i = 0; i < 10; i++) {
+            allp.push(console.log(reqid_1.ReqId.getReqID()));
+        }
+        await Promise.all(allp);
+    });
+    test("normal connect", async () => {
+        const allp = [];
+        for (let i = 0; i < 20; i++) {
+            allp.push(connect());
+            allp.push(stmtConnect());
+            allp.push(tmqConnect());
+        }
+        await Promise.all(allp);
+        console.log(stmtIds);
+    });
+});
+afterAll(async () => {
+    let conf = new config_1.WSConfig(dsn);
+    conf.setUser((0, utils_1.testUsername)());
+    conf.setPwd((0, utils_1.testPassword)());
+    let wsSql = await wsSql_1.WsSql.open(conf);
+    await wsSql.exec(`drop topic if exists ${topics[0]};`);
+    await wsSql.exec(`drop database if exists ${db};`);
+    await wsSql.close();
+    wsConnectorPool_1.WebSocketConnectionPool.instance().destroyed();
+});

package/lib/test/client/wsConnector.failover.test.d.ts

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

package/lib/test/client/wsConnector.failover.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"wsConnector.failover.test.d.ts","sourceRoot":"","sources":["../../../test/client/wsConnector.failover.test.ts"],"names":[],"mappings":""}