24 lines
693 B
TypeScript
24 lines
693 B
TypeScript
import { useCallback, useMemo } from '../lib/teact/teact';
|
|
|
|
import type { Scheduler } from '../util/schedulers';
|
|
|
|
import { throttle, throttleWith } from '../util/schedulers';
|
|
|
|
export default function useThrottledCallback<T extends AnyToVoidFunction>(
|
|
fn: T,
|
|
deps: any[],
|
|
msOrSchedulerFn: number | Scheduler,
|
|
noFirst = false,
|
|
) {
|
|
// eslint-disable-next-line react-hooks-static-deps/exhaustive-deps
|
|
const fnMemo = useCallback(fn, deps);
|
|
|
|
return useMemo(() => {
|
|
if (typeof msOrSchedulerFn === 'number') {
|
|
return throttle(fnMemo, msOrSchedulerFn, !noFirst);
|
|
} else {
|
|
return throttleWith(msOrSchedulerFn, fnMemo);
|
|
}
|
|
}, [fnMemo, msOrSchedulerFn, noFirst]);
|
|
}
|