dolphindb 0.0.1 → 0.0.5

package/index.js

@@ -1,6 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.DDB = exports.DdbFunction = exports.DdbPair = exports.DdbVectorAny = exports.DdbVectorDouble = exports.DdbVectorString = exports.DdbVectorInt = exports.DdbDouble = exports.DdbLong = exports.DdbString = exports.DdbInt = exports.DdbBool = exports.DdbVoid = exports.DdbObj = exports.nulls = exports.DdbFunctionType = exports.ddb_sizes = exports.DdbType = exports.DdbForm = void 0;
+exports.DDB = exports.timestamp2str = exports.date2str = exports.DdbFunction = exports.DdbPair = exports.DdbVectorAny = exports.DdbVectorDouble = exports.DdbVectorString = exports.DdbVectorInt = exports.DdbDouble = exports.DdbLong = exports.DdbString = exports.DdbInt = exports.DdbBool = exports.DdbVoid = exports.DdbObj = exports.timezone_offset = exports.nulls = exports.DdbFunctionType = exports.ddb_sizes = exports.DdbType = exports.DdbForm = void 0;
+const tslib_1 = require("tslib");
+const dayjs_1 = (0, tslib_1.__importDefault)(require("dayjs"));
 const ws_1 = require("ws");
 const utils_js_1 = require("xshell/utils.js");
 var DdbForm;
@@ -109,6 +111,8 @@ exports.nulls = {
     int32: -2147483648,
     int64: -0x8000000000000000n, // -922_3372_0368_5477_5808
 };
+exports.timezone_offset = 1000 * 60 * new Date().getTimezoneOffset();
+/** 可以表示所有 DolphinDB 数据库中的数据类型 */
 class DdbObj {
     constructor(data) {
         /** 是否为小端 (little endian) */
@@ -367,16 +371,86 @@ class DdbObj {
     static parse_vector(buf, le, type) {
         const dv = new DataView(buf.buffer, buf.byteOffset);
         const rows = dv.getUint32(0, le);
-        const i_items_start = 8;
-        const [len_items, value] = this.parse_vector_items(buf.subarray(i_items_start), le, type, rows);
+        let i_items_start = 8;
+        if (type < 64 || type >= 128) {
+            const [len_items, value] = this.parse_vector_items(buf.subarray(i_items_start), le, type, rows);
+            return new this({
+                le,
+                form: DdbForm.vector,
+                type,
+                length: i_items_start + len_items,
+                cols: 1,
+                rows,
+                value,
+            });
+        }
+        // array vector
+        // av = array(INT[], 0, 3)
+        // append!(av, [1..4])
+        // append!(av, [1..70000])
+        // av
+        // <Buffer 44 01  type = array vector, form = vector
+        // 02 00 00 00 74 11 01 00 rows = 2, cols = 70004 (0x011174)
+        // block 0
+        // 01 00 block.rows = 1
+        // 04 block.unit = 4
+        // 00 reserved
+        // 04 00 00 00 block.lengths = [4]
+        // 01 00 00 00 02 00 00 00 03 00 00 00 04 00 00 00 block.data
+        // block 1
+        // 01 00 block.rows = 1
+        // 04 block.unit = 4
+        // 00 reserved
+        // 70 11 01 00 block.lengths = [70000 (0x00011170)]
+        // 01 00 00 00 02 00 00 00 ... 279992 more bytes> block.data
+        const cols = dv.getUint32(4, le);
+        let blocks = [];
+        let i_block_start = i_items_start;
+        // 解析一个 block
+        for (let i_row = 0; i_row < rows;) {
+            /** 对应 array vector 中元素个数 */
+            const rows = dv.getUint16(i_block_start, le);
+            /** 每个 length 占用的字节数 */
+            const unit = dv.getUint8(i_block_start + 2);
+            /** array vector 每个元素的子元素长度 */
+            let lengths;
+            const i_lengths_start = i_block_start + 4;
+            const i_data_start = i_lengths_start + rows * unit;
+            const lengths_buf = buf.slice(i_lengths_start, i_data_start);
+            switch (unit) {
+                case 1:
+                    lengths = lengths_buf;
+                    break;
+                case 2:
+                    lengths = new Uint16Array(lengths_buf.buffer);
+                    break;
+                case 4:
+                    lengths = new Uint32Array(lengths_buf.buffer);
+                    break;
+                default:
+                    throw new Error(`array vector 存在非法 unit = ${unit}`);
+            }
+            let total_length = 0;
+            for (let i = 0; i < lengths.length; i++)
+                total_length += lengths[i];
+            const [len_items, data] = this.parse_vector_items(buf.subarray(i_data_start), le, type - 64, total_length);
+            blocks.push({
+                unit,
+                rows,
+                lengths,
+                data
+            });
+            i_block_start = i_data_start + len_items;
+            i_row += rows;
+        }
         return new this({
             le,
             form: DdbForm.vector,
             type,
-            length: i_items_start + len_items,
-            cols: 1,
+            length: i_block_start,
+            cols,
             rows,
-            value,
+            value: blocks
         });
     }
     /** 有可能没有字节对齐，不能直接使用原有 message 的 arraybuffer, 统一复制出来，让原有 arraybuffer 被回收掉比较好 */
@@ -457,13 +531,13 @@ class DdbObj {
                 }
                 const value_start = 8 + base_length;
                 const value_end = value_start + length * 4;
-                const value = new Uint32Array(buf.buffer.slice(buf.byteOffset + value_start, buf.byteOffset + value_end));
+                const data = new Uint32Array(buf.buffer.slice(buf.byteOffset + value_start, buf.byteOffset + value_end));
                 return [
                     value_end,
                     {
                         base_id,
                         base,
-                        value
+                        data
                     }
                 ];
             }
@@ -588,11 +662,24 @@ class DdbObj {
                             return [Float64Array.from(this.value)];
                         case DdbType.duration:
                             return [value];
+                        case DdbType.handle:
+                            throw new Error('DolphinDB Server 不支持 handle 的反序列化');
                         default:
                             throw new Error(`${DdbType[type]} 暂时不支持序列化`);
                     }
                 case DdbForm.vector:
                 case DdbForm.pair:
+                    // pack array vector
+                    if (form === DdbForm.vector && 64 <= type && type < 128)
+                        return [
+                            Uint32Array.of(this.rows, this.cols),
+                            ...this.value.map(block => ([
+                                Uint16Array.of(block.rows),
+                                Uint8Array.of(block.unit, 0),
+                                block.lengths,
+                                block.data
+                            ])).flat()
+                        ];
                     return [
                         Uint32Array.of(this.rows, 1),
                         ...DdbObj.pack_vector_body(value, type, this.rows)
@@ -708,6 +795,14 @@ class DdbObj {
                     bufs[i] = value[i].pack();
                 return bufs;
             }
+            case DdbType.symbol_extended: {
+                const { base_id, base, data } = value;
+                return [
+                    Uint32Array.of(base_id, base.length),
+                    ...this.pack_vector_body(base, DdbType.string, base.length),
+                    data
+                ];
+            }
             default:
                 throw new Error(`vector ${DdbType[type]} 暂不支持序列化`);
         }
@@ -721,6 +816,8 @@ class DdbObj {
                         return `functiondef<${DdbFunctionType[this.value.type]}>`;
                     return tname;
                 case DdbForm.vector:
+                    if (64 <= this.type && this.type < 128)
+                        return `${DdbType[this.type - 64]}[][${this.rows}]`;
                     return `${tname}[${this.rows}]`;
                 case DdbForm.pair:
                     return `pair<${tname}>`;
@@ -747,6 +844,15 @@ class DdbObj {
         })();
         return `${type.blue}(${this.name ? `${(0, utils_js_1.inspect)(this.name)}, ` : ''}${data})`;
     }
+    to_dict() {
+        if (this.form !== DdbForm.dict)
+            throw new Error('this.form 不是 dict, 不能转换为 Object');
+        const [{ value: keys }, { value: values }] = this.value;
+        let obj = {};
+        for (let i = 0; i < this.rows; i++)
+            obj[keys[i]] = values[i].value;
+        return obj;
+    }
 }
 exports.DdbObj = DdbObj;
 DdbObj.dec = new TextDecoder('utf-8');
@@ -896,6 +1002,20 @@ class DdbFunction extends DdbObj {
     }
 }
 exports.DdbFunction = DdbFunction;
+function date2str(date) {
+    return date === exports.nulls.int32 ?
+        ''
+        :
+            (0, dayjs_1.default)(exports.timezone_offset + 1000 * 3600 * 24 * date).format('YYYY.MM.DD');
+}
+exports.date2str = date2str;
+function timestamp2str(timestamp) {
+    return timestamp === exports.nulls.int64 ?
+        ''
+        :
+            (0, dayjs_1.default)(exports.timezone_offset + Number(timestamp)).format('YYYY.MM.DD HH:mm:ss.SSS');
+}
+exports.timestamp2str = timestamp2str;
 class DDB {
     constructor(ws_url) {
         this.port = 8848;
@@ -972,7 +1092,7 @@ class DDB {
     disconnect() {
         this.ws?.close(1000);
     }
-    /** rpc through websocket (function command)
+    /** rpc through websocket (function/script/variable command)
         - type: API 类型: 'script' | 'function' | 'variable'
         - options:
             - urgent?: 决定 `行为标识` 那一行字符串的取值（只适用于 script 和 function）
@@ -1047,7 +1167,7 @@ class DDB {
         - func: 函数名
         - args?: `[ ]` 调用参数 (传入的原生 string 和 boolean 会被自动转换为 DdbObj<string> 和 DdbObj<boolean>)
         - options?: 调用选项
-            - urgent?: 紧急 flag，使用 urgent worker 处理，防止被其它作业阻塞
+            - urgent?: 紧急 flag。使用 urgent worker 执行，防止被其它作业阻塞
             - node?: 设置结点 alias 时发送到集群中对应的结点执行 (使用 DolphinDB 中的 rpc 方法)
             - nodes?: 设置多个结点 alias 时发送到集群中对应的多个结点执行 (使用 DolphinDB 中的 pnodeRun 方法)
             - func_type?: 设置 node 参数时必传，需指定函数类型，其它情况下不传
@@ -1100,7 +1220,7 @@ class DDB {
         /** session id */
         const sid = parts[0];
         if (sid !== this.sid) {
-            console.log(`sid 从 ${this.sid} 变为 ${sid}`);
+            console.log(`session.id: ${this.sid} -> ${sid}`);
             this.sid = sid;
         }
         /** 返回对象的数量 */