mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-14 08:51:26 +08:00
22 lines
730 B
TypeScript
22 lines
730 B
TypeScript
import * as React from 'react';
|
|
|
|
export default function usePatchElement(): [
|
|
React.ReactElement[],
|
|
(element: React.ReactElement) => Function,
|
|
] {
|
|
const [elements, setElements] = React.useState<React.ReactElement[]>([]);
|
|
|
|
const patchElement = React.useCallback((element: React.ReactElement) => {
|
|
// append a new element to elements (and create a new ref)
|
|
setElements((originElements) => [...originElements, element]);
|
|
|
|
// return a function that removes the new element out of elements (and create a new ref)
|
|
// it works a little like useEffect
|
|
return () => {
|
|
setElements((originElements) => originElements.filter((ele) => ele !== element));
|
|
};
|
|
}, []);
|
|
|
|
return [elements, patchElement];
|
|
}
|