@tdengine/websocket 3.2.3 → 3.3.0

package/lib/test/helpers/utils.js

@@ -0,0 +1,341 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.testNon3360 = exports.testEnterprise = exports.testCloud = exports.tagMeta = exports.jsonMeta = exports.tableMeta = void 0;
+exports.getInsertBind = getInsertBind;
+exports.insertStable = insertStable;
+exports.insertNTable = insertNTable;
+exports.createBaseSTable = createBaseSTable;
+exports.createBaseSTableJSON = createBaseSTableJSON;
+exports.createBaseTable = createBaseTable;
+exports.createSTable = createSTable;
+exports.createSTableJSON = createSTableJSON;
+exports.createTable = createTable;
+exports.expectStableData = expectStableData;
+exports.hexToBytes = hexToBytes;
+exports.compareUint8Arrays = compareUint8Arrays;
+exports.Sleep = Sleep;
+exports.testUsername = testUsername;
+exports.testPassword = testPassword;
+exports.testCloudUrl = testCloudUrl;
+exports.testCloudToken = testCloudToken;
+exports.testCloudWsUrl = testCloudWsUrl;
+exports.hasTestCloudEnv = hasTestCloudEnv;
+const log_1 = __importDefault(require("@src/common/log"));
+function getInsertBind(valuesLen, tagsLen, db, stable) {
+    let sql = `insert into ? using ${db}.${stable} tags ( ?`;
+    for (let i = 1; i < tagsLen; i++) {
+        sql += ", ?";
+    }
+    sql += ") values( ?";
+    for (let i = 1; i < valuesLen; i++) {
+        sql += ", ?";
+    }
+    sql += ")";
+    log_1.default.debug(`Insert bind sql: ${sql}`);
+    return sql;
+}
+function insertStable(values, tags, stable, table = "empty") {
+    let childTable = table == "empty" ? stable + "_s_01" : table;
+    let sql = `insert into ${childTable} using ${stable} tags (`;
+    tags.forEach((tag) => {
+        if (typeof tag == "string") {
+            if (tag == "NULL") {
+                sql += tag + ",";
+            }
+            else {
+                sql += `\'${tag}\',`;
+            }
+        }
+        else {
+            sql += tag;
+            sql += ",";
+        }
+    });
+    sql = sql.slice(0, sql.length - 1);
+    sql += ")";
+    sql += "values";
+    values.forEach((value) => {
+        sql += "(";
+        value.forEach((v) => {
+            if (typeof v == "string") {
+                sql += `\'${v}\',`;
+            }
+            else {
+                sql += v;
+                sql += ",";
+            }
+        });
+        sql = sql.slice(0, sql.length - 1);
+        sql += ")";
+    });
+    return sql;
+}
+function insertNTable(values, table) {
+    let sql = `insert into ${table} values `;
+    values.forEach((value) => {
+        sql += "(";
+        value.forEach((v) => {
+            if (typeof v == "string") {
+                if (v == "NULL") {
+                    sql += v + ",";
+                }
+                else {
+                    sql += `\'${v}\',`;
+                }
+            }
+            else {
+                sql += v;
+                sql += ",";
+            }
+        });
+        sql = sql.slice(0, sql.length - 1);
+        sql += ")";
+    });
+    return sql;
+}
+exports.tableMeta = [
+    {
+        name: "ts",
+        type: "TIMESTAMP",
+        length: 8,
+    },
+    {
+        name: "i1",
+        type: "TINYINT",
+        length: 1,
+    },
+    {
+        name: "i2",
+        type: "SMALLINT",
+        length: 2,
+    },
+    {
+        name: "i4",
+        type: "INT",
+        length: 4,
+    },
+    {
+        name: "i8",
+        type: "BIGINT",
+        length: 8,
+    },
+    {
+        name: "u1",
+        type: "TINYINT UNSIGNED",
+        length: 1,
+    },
+    {
+        name: "u2",
+        type: "SMALLINT UNSIGNED",
+        length: 2,
+    },
+    {
+        name: "u4",
+        type: "INT UNSIGNED",
+        length: 4,
+    },
+    {
+        name: "u8",
+        type: "BIGINT UNSIGNED",
+        length: 8,
+    },
+    {
+        name: "f4",
+        type: "FLOAT",
+        length: 4,
+    },
+    {
+        name: "d8",
+        type: "DOUBLE",
+        length: 8,
+    },
+    {
+        name: "bnr",
+        type: "VARCHAR",
+        length: 200,
+    },
+    {
+        name: "nchr",
+        type: "NCHAR",
+        length: 200,
+    },
+    {
+        name: "b",
+        type: "BOOL",
+        length: 1,
+    },
+    {
+        name: "nilcol",
+        type: "INT",
+        length: 4,
+    },
+    {
+        name: "geo",
+        type: "GEOMETRY",
+        length: 512,
+    },
+    {
+        name: "vbinary",
+        type: "VARBINARY",
+        length: 32,
+    },
+];
+exports.jsonMeta = [
+    {
+        name: "json_tag",
+        type: "JSON",
+        length: 4095,
+    },
+];
+exports.tagMeta = [
+    {
+        name: "tb",
+        type: "BOOL",
+        length: 1,
+    },
+    {
+        name: "ti1",
+        type: "TINYINT",
+        length: 1,
+    },
+    {
+        name: "ti2",
+        type: "SMALLINT",
+        length: 2,
+    },
+    {
+        name: "ti4",
+        type: "INT",
+        length: 4,
+    },
+    {
+        name: "ti8",
+        type: "BIGINT",
+        length: 8,
+    },
+    {
+        name: "tu1",
+        type: "TINYINT UNSIGNED",
+        length: 1,
+    },
+    {
+        name: "tu2",
+        type: "SMALLINT UNSIGNED",
+        length: 2,
+    },
+    {
+        name: "tu4",
+        type: "INT UNSIGNED",
+        length: 4,
+    },
+    {
+        name: "tu8",
+        type: "BIGINT UNSIGNED",
+        length: 8,
+    },
+    {
+        name: "tf4",
+        type: "FLOAT",
+        length: 4,
+    },
+    {
+        name: "td8",
+        type: "DOUBLE",
+        length: 8,
+    },
+    {
+        name: "tbnr",
+        type: "VARCHAR",
+        length: 200,
+    },
+    {
+        name: "tnchr",
+        type: "NCHAR",
+        length: 200,
+    },
+];
+function createBaseSTable(stable) {
+    return (`create table if not exists ${stable}( ts timestamp,i1 tinyint,i2 smallint,i4 int,i8 bigint,u1 tinyint unsigned,u2 smallint unsigned,u4 int unsigned,u8 bigint unsigned,f4 float,d8 double,bnr binary(200),nchr nchar(200),b bool,nilcol int, g1 geometry(512), c4 varbinary(100))` +
+        "tags( tb bool,ti1 tinyint,ti2 smallint,ti4 int,ti8 bigint,tu1 tinyint unsigned,tu2 smallint unsigned,tu4 int unsigned,tu8 bigint unsigned,tf4 float,td8 double,tbnr binary(200),tnchr nchar(200));");
+}
+function createBaseSTableJSON(stable) {
+    return (`create table if not exists ${stable}(ts timestamp,i1 tinyint,i2 smallint,i4 int,i8 bigint,u1 tinyint unsigned,u2 smallint unsigned,u4 int unsigned,u8 bigint unsigned,f4 float,d8 double,bnr binary(200),nchr nchar(200),b bool,nilcol int)` +
+        "tags(json_tag json);");
+}
+function createBaseTable(table) {
+    return `create table if not exists ${table}(ts timestamp,i1 tinyint,i2 smallint,i4 int,i8 bigint,u1 tinyint unsigned,u2 smallint unsigned,u4 int unsigned,u8 bigint unsigned,f4 float,d8 double,bnr binary(200),nchr nchar(200),b bool,nilcol int)`;
+}
+function createSTable(stable) {
+    return (`create table if not exists ${stable}( ts timestamp,i1 tinyint,i2 smallint,i4 int,i8 bigint,u1 tinyint unsigned,u2 smallint unsigned,u4 int unsigned,u8 bigint unsigned,f4 float,d8 double,bnr binary(200),nchr nchar(200),b bool,nilcol int, geo geometry(512), vbinary varbinary(32))` +
+        "tags( tb bool,ti1 tinyint,ti2 smallint,ti4 int,ti8 bigint,tu1 tinyint unsigned,tu2 smallint unsigned,tu4 int unsigned,tu8 bigint unsigned,tf4 float,td8 double,tbnr binary(200),tnchr nchar(200));");
+}
+function createSTableJSON(stable) {
+    return (`create table if not exists ${stable}(ts timestamp,i1 tinyint,i2 smallint,i4 int,i8 bigint,u1 tinyint unsigned,u2 smallint unsigned,u4 int unsigned,u8 bigint unsigned,f4 float,d8 double,bnr binary(200),nchr nchar(200),b bool,nilcol int, geo geometry(512), vbinary varbinary(32))` +
+        "tags(json_tag json);");
+}
+function createTable(table) {
+    return `create table if not exists ${table}(ts timestamp,i1 tinyint,i2 smallint,i4 int,i8 bigint,u1 tinyint unsigned,u2 smallint unsigned,u4 int unsigned,u8 bigint unsigned,f4 float,d8 double,bnr binary(200),nchr nchar(200),b bool,nilcol int, geo geometry(512), vbinary varbinary(32))`;
+}
+function expectStableData(rows, tags) {
+    let resArr = [];
+    rows.forEach((row, index, rows) => {
+        resArr.push(row.concat(tags));
+    });
+    return resArr;
+}
+function hexToBytes(hex) {
+    let byteLen = hex.length / 2;
+    let a = new Uint8Array(byteLen);
+    for (let i = 0, count = 0; i < hex.length; i += 2, count++) {
+        let item = parseInt(hex.slice(i, i + 2), 16);
+        a[count] = item;
+    }
+    return a.buffer;
+}
+function compareUint8Arrays(arr1, arr2) {
+    if (arr1.length !== arr2.length) {
+        log_1.default.debug(`${arr1.length} !== ${arr2.length}`);
+        return false;
+    }
+    for (let i = 0; i < arr1.length; i++) {
+        if (arr1[i] !== arr2[i]) {
+            log_1.default.debug(`${arr1[i]} !== ${arr2[i]}`);
+            return false;
+        }
+    }
+    return true;
+}
+function Sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+function testUsername() {
+    return process.env.TDENGINE_TEST_USERNAME || "root";
+}
+function testPassword() {
+    return process.env.TDENGINE_TEST_PASSWORD || "taosdata";
+}
+function testCloudUrl() {
+    const cloudUrl = process.env.TDENGINE_CLOUD_URL?.trim();
+    return cloudUrl && cloudUrl.length > 0 ? cloudUrl : undefined;
+}
+function testCloudToken() {
+    const cloudToken = process.env.TDENGINE_CLOUD_TOKEN?.trim();
+    return cloudToken && cloudToken.length > 0 ? cloudToken : undefined;
+}
+function testCloudWsUrl() {
+    const cloudUrl = testCloudUrl();
+    const cloudToken = testCloudToken();
+    if (!cloudUrl || !cloudToken) {
+        return undefined;
+    }
+    return `wss://${cloudUrl}?token=${cloudToken}`;
+}
+function hasTestCloudEnv() {
+    return !!testCloudWsUrl();
+}
+exports.testCloud = hasTestCloudEnv() ? test : test.skip;
+exports.testEnterprise = process.env.TEST_ENTERPRISE?.toLowerCase() === "true" ? test : test.skip;
+exports.testNon3360 = process.env.TEST_3360?.toLowerCase() === "true" ? test.skip : test;

package/lib/test/helpers/wsFailoverProxy.d.ts

@@ -0,0 +1,109 @@
+type WsProxyDirection = "client_to_upstream" | "upstream_to_client";
+type WsProxyEventType = "proxy_started" | "proxy_stopped" | "client_connected" | "upstream_connected" | "message" | "tunnel_closed" | "error";
+interface WsProxyEventBase {
+    type: WsProxyEventType;
+    timestamp: number;
+    connectionId?: number;
+}
+export interface WsProxyStartedEvent extends WsProxyEventBase {
+    type: "proxy_started";
+    host: string;
+    port: number;
+    url: string;
+    reason?: string;
+}
+export interface WsProxyStoppedEvent extends WsProxyEventBase {
+    type: "proxy_stopped";
+    reason?: string;
+}
+export interface WsProxyClientConnectedEvent extends WsProxyEventBase {
+    type: "client_connected";
+    connectionId: number;
+    path: string;
+    remoteAddress: string;
+}
+export interface WsProxyUpstreamConnectedEvent extends WsProxyEventBase {
+    type: "upstream_connected";
+    connectionId: number;
+    upstreamUrl: string;
+}
+export interface WsProxyMessageEvent extends WsProxyEventBase {
+    type: "message";
+    connectionId: number;
+    direction: WsProxyDirection;
+    isBinary: boolean;
+    byteLength: number;
+    rawData: Buffer | string;
+    messageSeq: number;
+}
+export interface WsProxyTunnelClosedEvent extends WsProxyEventBase {
+    type: "tunnel_closed";
+    connectionId: number;
+    reason: string;
+    source: "client" | "upstream" | "proxy";
+}
+export interface WsProxyErrorEvent extends WsProxyEventBase {
+    type: "error";
+    error: Error;
+    stage: string;
+    sourceEventType?: WsProxyEventType;
+    details?: string;
+}
+export type WsProxyEvent = WsProxyStartedEvent | WsProxyStoppedEvent | WsProxyClientConnectedEvent | WsProxyUpstreamConnectedEvent | WsProxyMessageEvent | WsProxyTunnelClosedEvent | WsProxyErrorEvent;
+export interface WsProxyControl {
+    restart(opts?: {
+        downtimeMs?: number;
+        reason?: string;
+    }): Promise<void>;
+    stop(reason?: string): Promise<void>;
+    isRunning(): boolean;
+}
+export type WsProxyEventHandler = (event: WsProxyEvent, control: WsProxyControl) => void | Promise<void>;
+export interface WsFailoverProxyOptions {
+    host: string;
+    port: number;
+    onEvent?: WsProxyEventHandler;
+}
+export declare class WsFailoverProxy {
+    private readonly _listenHost;
+    private readonly _requestedPort;
+    private _lockedPort;
+    private readonly _onEvent?;
+    private readonly _control;
+    private _httpServer;
+    private _wsServer;
+    private _running;
+    private _nextConnectionId;
+    private _lifecycleChain;
+    private readonly _tunnels;
+    private readonly _eventLog;
+    constructor(options: WsFailoverProxyOptions);
+    static create(options: WsFailoverProxyOptions): Promise<WsFailoverProxy>;
+    start(reason?: string): Promise<void>;
+    stop(reason?: string): Promise<void>;
+    restart(opts?: {
+        downtimeMs?: number;
+        reason?: string;
+    }): Promise<void>;
+    isRunning(): boolean;
+    getPort(): number;
+    getHost(): string;
+    getUrl(): string;
+    getEventLog(): WsProxyEvent[];
+    private enqueueLifecycle;
+    private startInternal;
+    private stopInternal;
+    private handleClientRequest;
+    private connectUpstream;
+    private handleClientMessage;
+    private handleUpstreamMessage;
+    private flushPendingFrames;
+    private closeTunnel;
+    private isConnected;
+    private emitMessageEvent;
+    private emitError;
+    private emitEvent;
+    private safeCloseConnection;
+}
+export {};
+//# sourceMappingURL=wsFailoverProxy.d.ts.map

package/lib/test/helpers/wsFailoverProxy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"wsFailoverProxy.d.ts","sourceRoot":"","sources":["../../../test/helpers/wsFailoverProxy.ts"],"names":[],"mappings":"AAYA,KAAK,gBAAgB,GAAG,oBAAoB,GAAG,oBAAoB,CAAC;AACpE,KAAK,gBAAgB,GACf,eAAe,GACf,eAAe,GACf,kBAAkB,GAClB,oBAAoB,GACpB,SAAS,GACT,eAAe,GACf,OAAO,CAAC;AAEd,UAAU,gBAAgB;IACtB,IAAI,EAAE,gBAAgB,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,mBAAoB,SAAQ,gBAAgB;IACzD,IAAI,EAAE,eAAe,CAAC;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,mBAAoB,SAAQ,gBAAgB;IACzD,IAAI,EAAE,eAAe,CAAC;IACtB,MAAM,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,2BAA4B,SAAQ,gBAAgB;IACjE,IAAI,EAAE,kBAAkB,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,aAAa,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,6BAA8B,SAAQ,gBAAgB;IACnE,IAAI,EAAE,oBAAoB,CAAC;IAC3B,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,mBAAoB,SAAQ,gBAAgB;IACzD,IAAI,EAAE,SAAS,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,gBAAgB,CAAC;IAC5B,QAAQ,EAAE,OAAO,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,GAAG,MAAM,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,wBAAyB,SAAQ,gBAAgB;IAC9D,IAAI,EAAE,eAAe,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,QAAQ,GAAG,UAAU,GAAG,OAAO,CAAC;CAC3C;AAED,MAAM,WAAW,iBAAkB,SAAQ,gBAAgB;IACvD,IAAI,EAAE,OAAO,CAAC;IACd,KAAK,EAAE,KAAK,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,eAAe,CAAC,EAAE,gBAAgB,CAAC;IACnC,OAAO,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,MAAM,YAAY,GAClB,mBAAmB,GACnB,mBAAmB,GACnB,2BAA2B,GAC3B,6BAA6B,GAC7B,mBAAmB,GACnB,wBAAwB,GACxB,iBAAiB,CAAC;AAExB,MAAM,WAAW,cAAc;IAC3B,OAAO,CAAC,IAAI,CAAC,EAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACxE,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACrC,SAAS,IAAI,OAAO,CAAC;CACxB;AAED,MAAM,MAAM,mBAAmB,GAAG,CAC9B,KAAK,EAAE,YAAY,EACnB,OAAO,EAAE,cAAc,KACtB,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAE1B,MAAM,WAAW,sBAAsB;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,mBAAmB,CAAC;CACjC;AAiBD,qBAAa,eAAe;IACxB,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAS;IACxC,OAAO,CAAC,WAAW,CAAuB;IAC1C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAsB;IAChD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAiB;IAE1C,OAAO,CAAC,WAAW,CAA4B;IAC/C,OAAO,CAAC,SAAS,CAAgC;IACjD,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,iBAAiB,CAAK;IAC9B,OAAO,CAAC,eAAe,CAAoC;IAC3D,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAuC;IAChE,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAsB;gBAEpC,OAAO,EAAE,sBAAsB;WA2B9B,MAAM,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,eAAe,CAAC;IAMxE,KAAK,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAMrC,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAMpC,OAAO,CAAC,IAAI,CAAC,EAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAY7E,SAAS,IAAI,OAAO;IAIpB,OAAO,IAAI,MAAM;IAOjB,OAAO,IAAI,MAAM;IAIjB,MAAM,IAAI,MAAM;IAIhB,WAAW,IAAI,YAAY,EAAE;YAIf,gBAAgB;YAKhB,aAAa;YA2Db,YAAY;IA4C1B,OAAO,CAAC,mBAAmB;IAuD3B,OAAO,CAAC,eAAe;IAoDvB,OAAO,CAAC,mBAAmB;IA+B3B,OAAO,CAAC,qBAAqB;IAsB7B,OAAO,CAAC,kBAAkB;IAiB1B,OAAO,CAAC,WAAW;IAyBnB,OAAO,CAAC,WAAW;IAInB,OAAO,CAAC,gBAAgB;IAqBxB,OAAO,CAAC,SAAS;IAmBjB,OAAO,CAAC,SAAS;IAiBjB,OAAO,CAAC,mBAAmB;CAiB9B"}