react-environment-setup/template/umi/src/hooks/index.ts

react-environment-setup

import { useRef } from 'react'

/**
 * 节流
 * @param fn 函数
 * @param delay 节流时间
 * @returns Function
 */
export function useThrottle(fn: Function, delay: number): () => void {
  const { current } = useRef<any>({})

  return function (...args) {
    if (!current.timer) {
      current.timer = setTimeout(() => {
        clearTimeout(current.timer)
        Reflect.deleteProperty(current, 'timer')
      }, delay)

      fn(...args)
    }
  }
}

/**
 * 防抖
 * @param fn 函数
 * @param delay 防抖时间
 * @returns Function
 */
export function useDebounce(fn: Function, delay: number): () => void {
  const { current } = useRef<any>({})

  return function (...args) {
    if (current.timer) {
      clearTimeout(current.timer)
    }

    current.timer = setTimeout(() => {
      fn(...args)
    }, delay)
  }
}