• 欢迎光临~

React Hooks 手写节流useThrottle

开发技术 开发技术 2022-06-20 次浏览

定义

import { useCallback, useEffect, useRef } from "react"

export interface ThrottleRefType {
    fn: Function,
    timer?: NodeJS.Timeout
}

export type ThrottlePropsType = [Function, number, Array<any>]


const useThrottle = (...[fn, throttle, deps]: ThrottlePropsType) => {

    const { current } = useRef<ThrottleRefType>({ fn })

    useEffect(() => {
        current.fn = fn
    }, [fn, current])

    return useCallback(
        function (this: any, ...args: any[]) {
            if (!current.timer) {
                current.timer = setTimeout(() => {
                    current.fn.apply(this, Array.from(args))
                    delete current.timer
                }, throttle)
            }
        },
        // eslint-disable-next-line
        [...deps, current, throttle]
    )
}

export default useThrottle

使用

    const throttleFn = useThrottle(() => { console.log('run') },timeout, deps)
程序员灯塔
转载请注明原文链接:React Hooks 手写节流useThrottle
喜欢 (0)