react-use
Version:
Collection of React Hooks
30 lines (29 loc) • 1.02 kB
JavaScript
import { useMemo, useRef } from 'react';
import useEffectOnce from './useEffectOnce';
var useUnmountPromise = function () {
var refUnmounted = useRef(false);
useEffectOnce(function () { return function () {
refUnmounted.current = true;
}; });
var wrapper = useMemo(function () {
var race = function (promise, onError) {
var newPromise = new Promise(function (resolve, reject) {
promise.then(function (result) {
if (!refUnmounted.current)
resolve(result);
}, function (error) {
if (!refUnmounted.current)
reject(error);
else if (onError)
onError(error);
else
console.error('useUnmountPromise', error);
});
});
return newPromise;
};
return race;
}, []);
return wrapper;
};
export default useUnmountPromise;