35 lines
1.2 KiB
JavaScript
35 lines
1.2 KiB
JavaScript
|
"use strict";
|
||
|
|
||
|
var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard").default;
|
||
|
var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
|
||
|
Object.defineProperty(exports, "__esModule", {
|
||
|
value: true
|
||
|
});
|
||
|
exports.default = useSafeState;
|
||
|
var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
|
||
|
var React = _interopRequireWildcard(require("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.
|
||
|
*/
|
||
|
function useSafeState(defaultValue) {
|
||
|
var destroyRef = React.useRef(false);
|
||
|
var _React$useState = React.useState(defaultValue),
|
||
|
_React$useState2 = (0, _slicedToArray2.default)(_React$useState, 2),
|
||
|
value = _React$useState2[0],
|
||
|
setValue = _React$useState2[1];
|
||
|
React.useEffect(function () {
|
||
|
destroyRef.current = false;
|
||
|
return function () {
|
||
|
destroyRef.current = true;
|
||
|
};
|
||
|
}, []);
|
||
|
function safeSetState(updater, ignoreDestroy) {
|
||
|
if (ignoreDestroy && destroyRef.current) {
|
||
|
return;
|
||
|
}
|
||
|
setValue(updater);
|
||
|
}
|
||
|
return [value, safeSetState];
|
||
|
}
|