✨ feature: quickFind api add set and remove

Author: Simon he

README.md

@@ -2,7 +2,7 @@
 - deepCompare  // 比较2个对象的差异返回不同的属性和具体不同的值
 - deepMerge    // Object.assign的深度拷贝版本，返回合并后传入的第一个对象
 - asyncPool   // limit：控制异步并发执行的数量，tasks：异步任务数组
-- quickFind   // quickFind(array: any[], key: any),返回一个新的实例,在实例中find方法可以根据key查找对应的项,查找效率O(1)
+- quickFind   // quickFind(array: any[], key: any),返回一个新的实例,在实例中find方法可以根据key查找对应的项,查找效率O(1),set更新或新增项,delete删除项
 - quickFilter // quickFilter(array: any[], key: string | number | Array<string | number>, value: string | number | RegExp), 快速模糊查找key名字的项,支持正则匹配
 
 ## 使用方法

src/quickFind.ts

@@ -1,22 +1,75 @@
-export function quickFind(array: any[], key: any) {
+export function quickFind(array: any[], id: any) {
   const indexMap = new Map()
   array.forEach((item, i) => {
-    indexMap.set(item[key], i)
+    indexMap.set(item[id], i)
   })
-  return new QuickFind(array, indexMap)
+  return new QuickFind(array, indexMap, id)
 }
 
 class QuickFind {
-  constructor(public array: any[], public indexMap: Map<any, number>) {
+  constructor(public array: any[], public indexMap: Map<any, number>, public id: any) {
     this.array = array
     this.indexMap = indexMap
+    this.id = id
   }
 
-  find(key: any) {
-    const index = this.indexMap.get(key)
+  find(id: any) {
+    const index = this.indexMap.get(id)
     if (index === undefined)
       return undefined
     return this.array[index]
   }
+
+  _update(id: any, key: any, value: any) {
+    if (key === undefined) {
+      const index = this.indexMap.get(id)
+      if (index === undefined)
+        throw new Error('当前id不存在')
+
+      if (value[this.id] !== id)
+        throw new Error('不可修改唯一id')
+
+      this.array[index] = value
+    }
+    else {
+      const target = this.find(id)
+      if (target === undefined)
+        return
+      target[key] = value
+    }
+    return this.array
+  }
+
+  delete(id: any) {
+    const index = this.indexMap.get(id)
+    if (index === undefined)
+      return
+    this.array.splice(index, 1)
+    this.indexMap.delete(id)
+    return this.array
+  }
+
+  set(id: any, key: any, value?: any) {
+    const index = this.indexMap.get(id)
+    if (value === undefined) {
+      if (key === undefined)
+        return
+      value = key
+      key = undefined
+    }
+    if (index !== undefined) {
+      return this._update(id, key, value)
+    }
+    else {
+      if (value[this.id] === undefined)
+        throw new Error('新增的数据必须包含唯一id')
+      if (value[this.id] !== id)
+        throw new Error('新增的数据id必须与当前id一致')
+
+      this.indexMap.set(id, this.array.length)
+      this.array.push(value)
+      return this.array
+    }
+  }
 }
 

test/basic.test.ts

@@ -164,13 +164,75 @@ describe('Test 4', () => {
         id: 2,
       },
     ]
-    expect(quickFind(arr, 'id').find(2)).toMatchInlineSnapshot(`
+    const finder = quickFind(arr, 'id')
+    expect(finder.find(2)).toMatchInlineSnapshot(`
       {
         "age": 29,
         "id": 2,
         "name": "simon2",
       }
     `)
+    expect(finder.set(2, 'name', 'new value~~')).toMatchInlineSnapshot(`
+      [
+        {
+          "age": 18,
+          "id": 0,
+          "name": "simon",
+        },
+        {
+          "age": 19,
+          "id": 1,
+          "name": "simon1",
+        },
+        {
+          "age": 29,
+          "id": 2,
+          "name": "new value~~",
+        },
+      ]
+    `)
+    expect(finder.set(3, { id: 3, name: 'new value' })).toMatchInlineSnapshot(`
+      [
+        {
+          "age": 18,
+          "id": 0,
+          "name": "simon",
+        },
+        {
+          "age": 19,
+          "id": 1,
+          "name": "simon1",
+        },
+        {
+          "age": 29,
+          "id": 2,
+          "name": "new value~~",
+        },
+        {
+          "id": 3,
+          "name": "new value",
+        },
+      ]
+    `)
+    expect(finder.delete(3)).toMatchInlineSnapshot(`
+      [
+        {
+          "age": 18,
+          "id": 0,
+          "name": "simon",
+        },
+        {
+          "age": 19,
+          "id": 1,
+          "name": "simon1",
+        },
+        {
+          "age": 29,
+          "id": 2,
+          "name": "new value~~",
+        },
+      ]
+    `)
   })
 })