39 lines
668 B
TypeScript
39 lines
668 B
TypeScript
|
import { useMemo } from 'react'
|
||
|
|
|
||
|
|
const throttle = <Args extends unknown[]>(
|
||
|
|
fn: (...args: Args) => void,
|
||
|
|
cooldown: number
|
||
|
|
) => {
|
||
|
|
let lastArgs: Args | undefined
|
||
|
|
|
||
|
|
const run = () => {
|
||
|
|
if (lastArgs) {
|
||
|
|
fn(...lastArgs)
|
||
|
|
lastArgs = undefined
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const throttled = (...args: Args) => {
|
||
|
|
const isOnCooldown = !!lastArgs
|
||
|
|
|
||
|
|
lastArgs = args
|
||
|
|
|
||
|
|
if (isOnCooldown) {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
window.setTimeout(run, cooldown)
|
||
|
|
}
|
||
|
|
|
||
|
|
return throttled
|
||
|
|
}
|
||
|
|
|
||
|
|
const useThrottle = <Args extends unknown[]>(
|
||
|
|
cb: (...args: Args) => void,
|
||
|
|
cooldown: number
|
||
|
|
) => {
|
||
|
|
return useMemo(() => throttle(cb, cooldown), [cb, cooldown])
|
||
|
|
}
|
||
|
|
|
||
|
|
export { useThrottle }
|