baja-lite 1.0.0

package/es/set-ex.js

@@ -0,0 +1,332 @@
+export class SetEx extends Set {
+    /**
+     * @param key 识别是否存在的对象的属性名
+     * @param onExist 当存在时作何操作? oldData/newData 哪个将添加到set,由replaceItemWhenExits决定,默认是oldData生效
+     * @param onNotExist 当不存在时作何操作?
+     * @param replaceWhenExits 当存在时是否覆盖？
+     * @param values 初始数组
+     */
+    constructor(key, onExist, replaceWhenExits = false, values, onNotExist) {
+        super();
+        if (typeof key === 'object') {
+            this.whenOnExist = key.onExist;
+            this.uniqueKey = key.key;
+            this.replaceItemWhenExits = key.replaceWhenExits === true;
+            this.whenOnNotExist = key.onNotExist;
+            if (key.values) {
+                this.addAll(...key.values);
+            }
+        }
+        else {
+            this.whenOnExist = onExist;
+            this.uniqueKey = key;
+            this.replaceItemWhenExits = replaceWhenExits;
+            this.whenOnNotExist = onNotExist;
+            if (values) {
+                this.addAll(...values);
+            }
+        }
+    }
+    /**
+     *
+     * 添加返回
+     * @param {T} value
+     * @returns {this} 当前对象
+     */
+    add(value) {
+        let flag = false;
+        this.forEach((item) => {
+            if (item[this.uniqueKey] === value[this.uniqueKey]) {
+                flag = true;
+                if (this.whenOnExist) {
+                    this.whenOnExist(item, value);
+                }
+                if (this.replaceItemWhenExits === true) {
+                    super.delete(item);
+                    flag = false;
+                }
+                return false;
+            }
+        });
+        if (flag === false) {
+            super.add(value);
+            if (this.whenOnNotExist) {
+                this.whenOnNotExist(value);
+            }
+        }
+        return this;
+    }
+    /**
+     * 批量添加
+     * @param values
+     * @returns 当前对象
+     */
+    addAll(...values) {
+        for (const value of values) {
+            this.add(value);
+        }
+        return this;
+    }
+    /**
+     *
+     * 添加
+     * @param {T} value
+     * @returns {T} 添加成功的对象:可能是新加入集合的，也可能是原本存在的
+     */
+    add2(value) {
+        let flag = false;
+        let tmp = value;
+        this.forEach((item) => {
+            if (item[this.uniqueKey] === value[this.uniqueKey]) {
+                flag = true;
+                if (this.whenOnExist) {
+                    this.whenOnExist(item, value);
+                }
+                if (this.replaceItemWhenExits === true) {
+                    super.delete(item);
+                    flag = false;
+                }
+                else {
+                    tmp = item;
+                }
+                return false;
+            }
+        });
+        if (flag === false) {
+            super.add(value);
+            if (this.whenOnNotExist) {
+                this.whenOnNotExist(value);
+            }
+        }
+        return tmp;
+    }
+    /**
+     *
+     * 添加并返回添加成功的对象:可能是新加入集合的，也可能是原本存在的
+     * @param {T} values
+     * @returns {T}
+     */
+    addAll2(values) {
+        const result = [];
+        for (const value of values) {
+            result.push(this.add2(value));
+        }
+        return result;
+    }
+    /**
+     * 用key找到匹配的第一个对象
+     * @param {*} value 这是对象的关键属性,而非对象
+     * @returns {(T | null)}
+     */
+    find(value) {
+        for (const item of this) {
+            if (item[this.uniqueKey] === value) {
+                return item;
+            }
+        }
+        return null;
+    }
+    /**
+     * 用key找到匹配的所有对象
+     * @param {*} value 这是对象的关键属性,而非对象
+     * @returns {T[]}
+     */
+    findAll(value) {
+        const res = new Array();
+        this.forEach((item) => {
+            if (item[this.uniqueKey] === value) {
+                res.push(item);
+            }
+        });
+        return res;
+    }
+    /**
+     *
+     * 用函数回调找到匹配的第一个对象
+     * @param {(item: T) => boolean} fn
+     * @returns {T[]}
+     */
+    filter(fn) {
+        for (const item of this) {
+            if (fn(item) === true) {
+                return item;
+            }
+        }
+        return null;
+    }
+    /**
+     *
+     * 用函数回调找到匹配的所有对象
+     * @param {(item: T) => boolean} fn
+     * @returns {T[]}
+     */
+    filterAll(fn) {
+        const res = new Array();
+        this.forEach((item) => {
+            if (fn(item) === true) {
+                res.push(item);
+            }
+        });
+        return res;
+    }
+    /**
+     *
+     * 是否存在key对应的对象
+     * @param {*} value 这是对象的关键属性,而非对象
+     * @returns {boolean}
+     */
+    has(value) {
+        for (const item of this) {
+            if (item[this.uniqueKey] === value) {
+                return true;
+            }
+        }
+        return false;
+    }
+    /**
+     * 转为数组
+     * @param param0
+     * @returns
+     */
+    toArray({ sort, each, filter, map } = {}) {
+        let list = Array.from(this);
+        if (filter) {
+            list = list.filter(filter);
+        }
+        if (sort) {
+            list.sort(sort);
+        }
+        if (each) {
+            list.forEach(each);
+        }
+        if (map) {
+            list = list.map(map);
+        }
+        return list;
+    }
+    /**
+     * 转为JSON对象
+     * @param key
+     * @param value
+     * @param param2
+     * @returns
+     */
+    toJSON(key, value, { sort, each, filter, map } = {}) {
+        return Object.fromEntries(this.toArray({ sort, each, filter, map }).map(i => [i[key], value ? i[value] : i]));
+    }
+    /**
+     *
+     * 删除key对应的对象
+     * @param {*} value 这是对象的关键属性,而非对象
+     * @returns {boolean}
+     */
+    delete(value) {
+        for (const item of this) {
+            if (item[this.uniqueKey] === value) {
+                super.delete(item);
+                return true;
+            }
+        }
+        return false;
+    }
+    /**
+     *
+     * 重置
+     * @param {keyof T} key
+     * @param {(oldData: T, newData: T) => void} [onExist]
+     * @param {boolean} [replaceWhenExits=false]
+     */
+    reset({ key, onExist, onNotExist, replaceWhenExits }) {
+        if (onExist !== undefined) {
+            this.whenOnExist = onExist;
+        }
+        if (onNotExist !== undefined) {
+            this.whenOnNotExist = onNotExist;
+        }
+        if (key) {
+            this.uniqueKey = key;
+        }
+        if (replaceWhenExits !== undefined) {
+            this.replaceItemWhenExits = replaceWhenExits;
+        }
+        this.clear();
+        return this;
+    }
+    /**
+     *
+     * @param param0 转为JSON对象，value可能是数组
+     * @returns
+     */
+    toJSONArray({ sort, each, filter, map } = {}) {
+        const result = {};
+        const list = this.toArray({ sort, each, filter, map });
+        for (const item of list) {
+            for (const key in item) {
+                if (!result[key]) {
+                    result[key] = [];
+                }
+                result[key].push(item[key]);
+            }
+        }
+        return result;
+    }
+    /**
+     * 转为hot-table支持的数组
+     * @param param0
+     * @param keys
+     * @returns
+     */
+    toDataGrid({ sort, each, filter, map } = {}, ...keys) {
+        if (this.size === 0) {
+            return [];
+        }
+        if (keys.length === 0) {
+            keys = Object.keys(this.values().next().value);
+        }
+        const result = [keys];
+        const list = this.toArray({ sort, each, filter, map });
+        for (const item of list) {
+            const one = [];
+            for (const key of keys) {
+                one.push(item[key]);
+            }
+            result.push(one);
+        }
+        return result;
+    }
+    /**
+     * 转为饼图支持的数组
+     * @param param0
+     * @param keys
+     * @returns
+     */
+    toPieGrid({ sort, each, filter, map } = {}, ...keys) {
+        if (this.size === 0) {
+            return {};
+        }
+        if (keys.length === 0) {
+            keys = Object.keys(this.values().next().value);
+        }
+        const result = {};
+        const list = this.toArray({ sort, each, filter, map });
+        for (const item of list) {
+            const name = item[this.uniqueKey];
+            for (const key in item) {
+                if (!result[key]) {
+                    result[key] = [];
+                }
+                result[key].push({ name, value: item[key] });
+            }
+        }
+        return result;
+    }
+    set onExist(onExist) {
+        this.whenOnExist = onExist;
+    }
+    set key(key) {
+        this.uniqueKey = key;
+    }
+    set replaceWhenExits(replaceWhenExits) {
+        this.replaceItemWhenExits = replaceWhenExits;
+    }
+}