wsResponse.ts

import { MessageResp, readVarchar } from "../common/taosResult";

export class WSVersionResponse {
    version: string;
    code: number;
    message: string;
    action: string;
    totalTime: number;

    constructor(resp: MessageResp) {
        this.version = resp.msg.version;
        this.code = resp.msg.code;
        this.message = resp.msg.message;
        this.action = resp.msg.action;
        this.totalTime = resp.totalTime;
    }
}

export class WSQueryResponse {
    code?: number;
    message?: string;
    action?: string;
    req_id?: number;
    timing?: bigint | null;
    totalTime: number;
    id?: bigint;
    is_update?: boolean;
    affected_rows?: number | null;
    fields_count?: number | null;
    fields_names?: Array<string> | null;
    fields_types?: Array<number> | null;
    fields_lengths?: Array<number> | null;
    fields_precisions?: Array<bigint> | null;
    fields_scales?: Array<bigint> | null;
    precision?: number;

    constructor(resp: MessageResp) {
        this.totalTime = resp.totalTime;
        this.initMsg(resp.msg);
    }

    private initMsg(msg: any) {
        this.code = msg.code;
        this.message = msg.message;
        this.action = msg.action;
        this.req_id = msg.req_id;
        this.timing = BigInt(msg.timing);
        if (msg.id) {
            this.id = BigInt(msg.id);
        } else {
            this.id = BigInt(0);
        }

        this.is_update = msg.is_update;
        this.affected_rows = msg.affected_rows;
        this.fields_count = msg.fields_count;
        this.fields_names = msg.fields_names;
        this.fields_types = msg.fields_types;
        this.fields_lengths = msg.fields_lengths;
        this.precision = msg.precision;
        this.fields_precisions = msg.fields_precisions
            ? msg.fields_precisions.map((p: number) => BigInt(p))
            : [];
        this.fields_scales = msg.fields_scales
            ? msg.fields_scales.map((s: number) => BigInt(s))
            : [];
    }
}

export class WSFetchBlockResponse {
    data: DataView | undefined;
    action: bigint;
    timing: bigint;
    reqId: bigint;
    code: number;
    blockLen: number;
    message: string | undefined;
    resultId: bigint | undefined;
    finished: number | undefined;
    metaType: number | undefined;
    textDecoder: TextDecoder;

    constructor(msg: ArrayBuffer) {
        let dataView = new DataView(msg);
        this.action = dataView.getBigUint64(8, true);
        this.timing = dataView.getBigUint64(18, true);
        this.reqId = dataView.getBigUint64(26, true);
        this.code = dataView.getUint32(34, true);
        this.textDecoder = new TextDecoder();
        this.blockLen = 0;
        if (this.code != 0) {
            let len = dataView.getUint32(38, true);
            this.message = readVarchar(msg, 42, len, this.textDecoder);
            return;
        }
        this.resultId = dataView.getBigUint64(42, true);
        let offset = 50;
        if (this.action == BigInt(8)) {
            this.metaType = dataView.getUint16(50, true);
            offset += 2;
        } else {
            this.finished = dataView.getUint8(50);
            if (this.finished == 1) {
                return;
            }
            offset += 1;
        }

        this.blockLen = dataView.getUint32(offset, true);
        if (this.blockLen > 0) {
            this.data = new DataView(msg, offset + 4);
        }
    }
}