fix: 兼容无法通过 ref 获取到 dom 的情况

This commit is contained in:
lihao.ylh 2021-08-10 14:14:14 +08:00
parent 11bbbd02d3
commit 44d93de9e9

View File

@ -29,6 +29,11 @@ export function reactFindDOMNodes(elem: ReactInstance | null): Array<Element | T
}
const elements: Array<Element | Text> = [];
const fiberNode = (elem as any)[FIBER_KEY];
elementsFromFiber(fiberNode.child, elements);
return elements.length > 0 ? elements : [findDOMNode(elem)];
elementsFromFiber(fiberNode?.child, elements);
if (elements.length > 0) return elements;
try {
return [findDOMNode(elem)];
} catch (e) {
return null;
}
}