dolphindb 0.0.10 → 0.0.13

package/index.js

@@ -1,9 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-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;
+exports.DDB = exports.str2nanotimestamp = exports.nanotimestamp2str = exports.timestamp2str = exports.datetime2str = exports.date2str = exports.DdbFunction = exports.DdbPair = exports.DdbVectorAny = exports.DdbVectorDouble = exports.DdbVectorString = exports.DdbVectorInt = exports.DdbNanoTimeStamp = exports.DdbTimeStamp = exports.DdbDateTime = exports.DdbDouble = exports.DdbLong = exports.DdbString = exports.DdbInt = exports.DdbChar = 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 ws_1 = require("ws");
 const dayjs_1 = tslib_1.__importDefault(require("dayjs"));
+const customParseFormat_js_1 = tslib_1.__importDefault(require("dayjs/plugin/customParseFormat.js"));
+dayjs_1.default.extend(customParseFormat_js_1.default);
 const utils_js_1 = require("xshell/utils.js");
 var DdbForm;
 (function (DdbForm) {
@@ -150,7 +152,7 @@ class DdbObj {
                     le,
                     form,
                     type,
-                    length: 2 + length,
+                    length: i_data + length,
                     value,
                 });
             }
@@ -158,7 +160,7 @@ class DdbObj {
             case DdbForm.pair:
             case DdbForm.set: {
                 let vector = this.parse_vector(buf_data, le, type);
-                vector.length += 2;
+                vector.length += i_data;
                 vector.form = form;
                 return vector;
             }
@@ -189,17 +191,17 @@ class DdbObj {
                 let i_start = i_items_start + len_items;
                 for (let i = 0; i < cols; i++) {
                     const type = buf_data[i_start];
-                    const i_vector_head = i_start + 2;
-                    let col = this.parse_vector(buf_data.subarray(i_vector_head), le, type);
+                    let col = this.parse_vector(buf_data.subarray(i_start + 2), le, type);
+                    col.length += 2;
                     col.name = colnames[i];
                     value[i] = col;
-                    i_start = i_vector_head + col.length;
+                    i_start += col.length;
                 }
                 return new this({
                     le,
                     form,
                     type,
-                    length: i_start,
+                    length: i_data + i_start,
                     name,
                     rows,
                     cols,
@@ -222,7 +224,7 @@ class DdbObj {
                     le,
                     form,
                     type,
-                    length: 2 + keys.length + values.length,
+                    length: i_data + keys.length + values.length,
                     rows: keys.rows,
                     cols: 2,
                     value: [
@@ -255,7 +257,7 @@ class DdbObj {
                     le,
                     form,
                     type,
-                    length: 11 + len_items,
+                    length: i_data + 11 + len_items,
                     rows,
                     cols,
                     datatype,
@@ -267,7 +269,7 @@ class DdbObj {
                     le,
                     form,
                     type,
-                    length: buf_data.length,
+                    length: i_data + buf_data.length,
                     value: buf_data
                 });
         }
@@ -808,7 +810,7 @@ class DdbObj {
                 throw new Error(`vector ${DdbType[type]} 暂不支持序列化`);
         }
     }
-    [utils_js_1.inspect.custom]() {
+    [utils_js_1.inspect.custom](depth, options, _inspect) {
         const type = (() => {
             const tname = DdbType[this.type];
             switch (this.form) {
@@ -823,7 +825,7 @@ class DdbObj {
                 case DdbForm.pair:
                     return `pair<${tname}>`;
                 case DdbForm.set:
-                    return `set<${tname}>`;
+                    return `set<${tname}>[${this.rows}]`;
                 case DdbForm.table:
                     return `table[${this.rows} rows][${this.cols} cols]`;
                 case DdbForm.dict:
@@ -836,14 +838,37 @@ class DdbObj {
         })();
         const data = (() => {
             if (this.form === DdbForm.pair)
-                return `${(0, utils_js_1.inspect)(this.value[0])}, ${(0, utils_js_1.inspect)(this.value[1])}`;
+                return `${(0, utils_js_1.inspect)(this.value[0], options)}, ${(0, utils_js_1.inspect)(this.value[1], options)}`;
             if (this.form === DdbForm.scalar && this.type === DdbType.functiondef)
-                return (0, utils_js_1.inspect)(this.value.name);
+                return (0, utils_js_1.inspect)(this.value.name, options);
             if (this.value instanceof Uint8Array)
-                return (0, utils_js_1.inspect)((0, utils_js_1.typed_array_to_buffer)(this.value));
-            return (0, utils_js_1.inspect)(this.value);
+                return (0, utils_js_1.inspect)((0, utils_js_1.typed_array_to_buffer)(this.value), options);
+            return (0, utils_js_1.inspect)(this.value, options);
         })();
-        return `${type.blue}(${this.name ? `${(0, utils_js_1.inspect)(this.name)}, ` : ''}${data})`;
+        return `${options.colors ? type.blue : type}(${this.name ? `${(0, utils_js_1.inspect)(this.name, options)}, ` : ''}${data})`;
+    }
+    to_rows() {
+        if (this.form !== DdbForm.table)
+            throw new Error('this.form is not DdbForm.table, cannot call to_rows');
+        let rows = new Array(this.rows);
+        for (let i = 0; i < this.rows; i++) {
+            let row = {};
+            for (let j = 0; j < this.cols; j++) {
+                const { type, name, value } = this.value[j]; // column
+                switch (type) {
+                    case DdbType.bool:
+                        row[name] = Boolean(value[i]);
+                        break;
+                    case DdbType.ipaddr:
+                        row[name] = value.subarray(16 * i, 16 * (i + 1));
+                        break;
+                    default:
+                        row[name] = value[i];
+                }
+            }
+            rows[i] = row;
+        }
+        return rows;
     }
     to_dict() {
         if (this.form !== DdbForm.dict)
@@ -883,6 +908,17 @@ class DdbBool extends DdbObj {
     }
 }
 exports.DdbBool = DdbBool;
+class DdbChar extends DdbObj {
+    constructor(value) {
+        super({
+            form: DdbForm.scalar,
+            type: DdbType.char,
+            length: 1,
+            value,
+        });
+    }
+}
+exports.DdbChar = DdbChar;
 class DdbInt extends DdbObj {
     constructor(value) {
         super({
@@ -927,6 +963,39 @@ class DdbDouble extends DdbObj {
     }
 }
 exports.DdbDouble = DdbDouble;
+class DdbDateTime extends DdbObj {
+    constructor(value) {
+        super({
+            form: DdbForm.scalar,
+            type: DdbType.datetime,
+            length: 4,
+            value
+        });
+    }
+}
+exports.DdbDateTime = DdbDateTime;
+class DdbTimeStamp extends DdbObj {
+    constructor(value) {
+        super({
+            form: DdbForm.scalar,
+            type: DdbType.timestamp,
+            length: 8,
+            value
+        });
+    }
+}
+exports.DdbTimeStamp = DdbTimeStamp;
+class DdbNanoTimeStamp extends DdbObj {
+    constructor(value) {
+        super({
+            form: DdbForm.scalar,
+            type: DdbType.nanotimestamp,
+            length: 8,
+            value
+        });
+    }
+}
+exports.DdbNanoTimeStamp = DdbNanoTimeStamp;
 class DdbVectorInt extends DdbObj {
     constructor(value, name) {
         super({
@@ -1007,13 +1076,20 @@ class DdbFunction extends DdbObj {
     }
 }
 exports.DdbFunction = DdbFunction;
-function date2str(date) {
+function date2str(date, format = 'YYYY.MM.DD') {
     return date === exports.nulls.int32 ?
         ''
         :
-            (0, dayjs_1.default)(exports.timezone_offset + 1000 * 3600 * 24 * date).format('YYYY.MM.DD');
+            (0, dayjs_1.default)(exports.timezone_offset + 1000 * 3600 * 24 * date).format(format);
 }
 exports.date2str = date2str;
+function datetime2str(datetime, format = 'YYYY.MM.DD HH:mm:ss') {
+    return datetime === exports.nulls.int32 ?
+        ''
+        :
+            (0, dayjs_1.default)(exports.timezone_offset + 1000 * datetime).format(format);
+}
+exports.datetime2str = datetime2str;
 function timestamp2str(timestamp) {
     return timestamp === exports.nulls.int64 ?
         ''
@@ -1021,6 +1097,29 @@ function timestamp2str(timestamp) {
             (0, dayjs_1.default)(exports.timezone_offset + Number(timestamp)).format('YYYY.MM.DD HH:mm:ss.SSS');
 }
 exports.timestamp2str = timestamp2str;
+function nanotimestamp2str(nanotimestamp, format = 'YYYY.MM.DD HH:mm:ss') {
+    return nanotimestamp === exports.nulls.int64 ?
+        ''
+        :
+            (0, dayjs_1.default)(exports.timezone_offset + (Number(nanotimestamp) / 1000000)).format(format) + `.${nanotimestamp % 1000000000n}`;
+}
+exports.nanotimestamp2str = nanotimestamp2str;
+/** parse nano timestamp string to bigint value
+    - str: nano timestamp string, 如果为空字符串会返回对应的空值 (nulls.int64)
+    - format?: 对应传入字符串的到秒之前 (包括) 的格式串，默认是 `YYYY.MM.DD HH:mm:ss`
+        https://day.js.org/docs/en/parse/string-format#list-of-all-available-parsing-tokens
+*/
+function str2nanotimestamp(str, format = 'YYYY.MM.DD HH:mm:ss') {
+    if (!str)
+        return exports.nulls.int64;
+    if (str.length - format.length !== 9)
+        throw new Error('nanotimestamp 字符串长度减格式串长度应该等于 9');
+    return (BigInt(-exports.timezone_offset +
+        (0, dayjs_1.default)(str.slice(0, format.length), format).valueOf()) * 1000000n
+        +
+            BigInt(str.slice(format.length)));
+}
+exports.str2nanotimestamp = str2nanotimestamp;
 class DDB {
     /**
         Initialize an instance of DolphinDB Client using the WebSocket URL