24 lines
737 B
JavaScript
24 lines
737 B
JavaScript
|
import * as React from 'react';
|
||
|
|
/**
|
||
|
|
* Same as React.useState but `setState` accept `ignoreDestroy` param to not to setState after destroyed.
|
||
|
|
* We do not make this auto is to avoid real memory leak.
|
||
|
|
* Developer should confirm it's safe to ignore themselves.
|
||
|
|
*/
|
||
|
|
const useSafeState = defaultValue => {
|
||
|
|
const destroyRef = React.useRef(false);
|
||
|
|
const [value, setValue] = React.useState(defaultValue);
|
||
|
|
React.useEffect(() => {
|
||
|
|
destroyRef.current = false;
|
||
|
|
return () => {
|
||
|
|
destroyRef.current = true;
|
||
|
|
};
|
||
|
|
}, []);
|
||
|
|
function safeSetState(updater, ignoreDestroy) {
|
||
|
|
if (ignoreDestroy && destroyRef.current) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
setValue(updater);
|
||
|
|
}
|
||
|
|
return [value, safeSetState];
|
||
|
|
};
|
||
|
|
export default useSafeState;
|