用React 手写 倒计时功能--自定义hooks (逻辑复用)

 平时工作偶尔会把一些逻辑相通的功能作逻辑复用,这里记录下来 希望对大家实际开发有些帮助

 需求 : 点击一个按钮 按钮用 过n秒回复


import { useState, useEffect, useRef } from 'react'
// 第一参数为你定时的数字
// 第二参数为你定时结束时想做的事
// 返回的第一参数为可变值
// 返回第二参数为触发函数
export function useTimer (Num, callBack = () => {}) {
  const [num, setNum] = useState(Num)
  const ref = useRef(null)
  // 调用这个方法,开始倒数
  const start = () => {
    setNum(Num) // 重新赋值
  // 定时器
    ref.current = setInterval(() => {
      setNum((num) => num - 1)
    }, 1000)
  }
  // 倒数为0 关闭定时器
  useEffect(
    () => {
      if (num === 0) {
        return () => {
          clearInterval(ref.current) // 关闭定时器
          callBack() // 自己想做的事
        }
      }
    },
    [num]
  )
 
  // 解决当正在计数的组件开始倒数 删除组件导致无法取消定时器
  useEffect(() => {
    return () => {
      clearInterval(ref.current)
    }
  }, [])
  // 返回可变值 与 触发函数
  return {
    nu

你可能感兴趣的:(react.js,javascript,前端)