feat: walkFlatDataGenerator stat add id, pid

Author: phphe

lib/HeTree.tsx

@@ -851,7 +851,7 @@ const flatDataDefaultOptions = {
   parentIdKey: 'parent_id',
 }
 export type WalkFlatDataYield<T> = [T, {
-  parent: T | null, parents: T[], index: number, treeIndex: number, skipChildren: VoidFunction, exitWalk: VoidFunction
+  parent: T | null, parents: T[], index: number, treeIndex: number, id: Id, pid: Id, skipChildren: VoidFunction, exitWalk: VoidFunction
 }]
 export function* walkFlatDataGenerator<T extends Record<Id, any>>(flatData: T[], options0?: Partial<typeof flatDataDefaultOptions>): Generator<WalkFlatDataYield<T>> {
   const options = { ...flatDataDefaultOptions, ...options0 }
@@ -879,6 +879,8 @@ export function* walkFlatDataGenerator<T extends Record<Id, any>>(flatData: T[],
       parent,
       parents: parent ? [...stats[pid]!.parents, parent] : [],
       index,
+      id,
+      pid,
       treeIndex,
       skipChildren,
       exitWalk,

src/test/flatData.test.ts

@@ -124,13 +124,30 @@ test("walkFlatDataGenerator: exitWalk", () => {
 });
 test("walkFlatData", () => {
   let data = createData();
-  walkFlatData(data, (node, { treeIndex, parent, index, parents }) => {
-    if (treeIndex === 0) {
-      expect(parent).toBe(null);
-      expect(JSON.stringify(parents)).toBe("[]");
-      expect(node.id).toBe(1);
+  walkFlatData(
+    data,
+    (node, { treeIndex, parent, index, parents, exitWalk }) => {
+      if (treeIndex === 0) {
+        expect(parent).toBe(null);
+        expect(JSON.stringify(parents)).toBe("[]");
+        expect(node.id).toBe(1);
+        exitWalk();
+      }
     }
-  });
+  );
+});
+test("walkFlatData: id, pid", () => {
+  let data = createData();
+  walkFlatData(
+    data,
+    (node, { treeIndex, parent, index, parents, exitWalk, id, pid }) => {
+      if (treeIndex === 2) {
+        expect(id).toBe(5);
+        expect(pid).toBe(2);
+        exitWalk();
+      }
+    }
+  );
 });
 test("walkFlatData: options", () => {
   let data = createData("key", "pid");