Skip to content

Commit 2fa9b14

Browse files
committed
fix: openIds
1 parent 102fbdc commit 2fa9b14

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

lib/HeTree.tsx

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ export function useHeTree<T extends Record<string, any>>(
8787
throw new Error("Either renderNodeBox or renderNode is required.");
8888
}
8989
const rtl = props.direction === 'rtl'
90-
const openIdsStr = useMemo(() => props.openIds ? [...props.openIds].sort().toString() : '', [props.openIds])
90+
const openIdsStr = useMemo(() => props.openIds ? [...props.openIds].sort().toString() : undefined, [props.openIds])
9191
const openIdSet = useMemo(() => new Set(props.openIds), [openIdsStr])
9292
const checkedIdsStr = useMemo(() => props.checkedIds ? [...props.checkedIds].sort().toString() : '', [props.checkedIds])
9393
const checkedIdSet = useMemo(() => new Set(props.checkedIds), [checkedIdsStr])
@@ -101,6 +101,7 @@ export function useHeTree<T extends Record<string, any>>(
101101
const rootIds: Id[] = []
102102
const rootNodes: T[] = []
103103
const rootStats: Stat<T>[] = []
104+
const allIds: Id[] = []
104105
//
105106
function* simpleWalk() {
106107
if (props.dataType === 'flat') {
@@ -116,6 +117,7 @@ export function useHeTree<T extends Record<string, any>>(
116117
let count = 0
117118
for (const [node, info] of simpleWalk()) {
118119
const id: Id = node[ID] ?? count
120+
allIds.push(id)
119121
let pid = node[PID] as Id
120122
if (props.dataType === 'tree') {
121123
pid = info.parent?.[ID] ?? null
@@ -186,6 +188,8 @@ export function useHeTree<T extends Record<string, any>>(
186188
return {
187189
// root
188190
rootIds, rootNodes, rootStats,
191+
//
192+
allIds,
189193
// methods
190194
getStat,
191195
}

0 commit comments

Comments
 (0)