You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
1.0 KiB
34 lines
1.0 KiB
3 years ago
|
import { useEffect } from 'react';
|
||
|
|
import {
|
||
|
|
disconnectGasFeeEstimatePoller,
|
||
|
|
getGasFeeEstimatesAndStartPolling,
|
||
|
|
} from '../store/actions';
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Provides a reusable hook that can be used for safely updating the polling
|
||
|
|
* data in the gas fee controller. It makes a request to get estimates and
|
||
|
|
* begin polling, keeping track of the poll token for the lifetime of the hook.
|
||
|
|
* It then disconnects polling upon unmount. If the hook is unmounted while waiting
|
||
|
|
* for `getGasFeeEstimatesAndStartPolling` to resolve, the `active` flag ensures
|
||
|
|
* that a call to disconnect happens after promise resolution.
|
||
|
|
*/
|
||
|
|
export function useSafeGasEstimatePolling() {
|
||
|
|
useEffect(() => {
|
||
|
|
let active = true;
|
||
|
|
let pollToken;
|
||
|
|
getGasFeeEstimatesAndStartPolling().then((newPollToken) => {
|
||
|
|
if (active) {
|
||
|
|
pollToken = newPollToken;
|
||
|
|
} else {
|
||
|
|
disconnectGasFeeEstimatePoller(newPollToken);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
return () => {
|
||
|
|
active = false;
|
||
|
|
if (pollToken) {
|
||
|
|
disconnectGasFeeEstimatePoller(pollToken);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}, []);
|
||
|
|
}
|