amis-rpc-design/node_modules/rc-tree/es/useUnmount.js
2023-10-07 19:42:30 +08:00

25 lines
698 B
JavaScript

import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
import * as React from 'react';
/**
* Trigger only when component unmount
*/
export default function useUnmount(triggerStart, triggerEnd) {
var _React$useState = React.useState(false),
_React$useState2 = _slicedToArray(_React$useState, 2),
firstMount = _React$useState2[0],
setFirstMount = _React$useState2[1];
React.useLayoutEffect(function () {
if (firstMount) {
triggerStart();
return function () {
triggerEnd();
};
}
}, [firstMount]);
React.useLayoutEffect(function () {
setFirstMount(true);
return function () {
setFirstMount(false);
};
}, []);
}