40 lines
1,008 B
TypeScript
40 lines
1,008 B
TypeScript
import { useEffect, useRef, useCallback } from "react";
|
|
|
|
export default function useSetTimeout() {
|
|
const didUnmountRef = useRef(false);
|
|
const scheduledTimersRef = useRef(new Set<number>());
|
|
|
|
useEffect(() => {
|
|
didUnmountRef.current = false;
|
|
|
|
return () => {
|
|
didUnmountRef.current = true;
|
|
clearAll();
|
|
};
|
|
}, []);
|
|
|
|
function clearAll() {
|
|
scheduledTimersRef.current.forEach((timer) => clearTimeout(timer));
|
|
scheduledTimersRef.current.clear();
|
|
}
|
|
|
|
return useCallback((callback: () => void, delay: number) => {
|
|
if (didUnmountRef.current) {
|
|
throw new Error("Can't schedule a timeout on an unmounted component.");
|
|
}
|
|
|
|
const timer = Number(setTimeout(handleCallback, delay));
|
|
|
|
scheduledTimersRef.current.add(timer);
|
|
|
|
function handleCallback() {
|
|
scheduledTimersRef.current.delete(timer);
|
|
callback();
|
|
}
|
|
|
|
return function cancelTimeout() {
|
|
clearTimeout(timer);
|
|
scheduledTimersRef.current.delete(timer);
|
|
};
|
|
}, []);
|
|
}
|