useMount

作用

在组件初始化挂载时执行回调函数

原理

当useEffect的第二个参数为空数组时，useEffect只会在组件挂载时执行一次，因此可以在此时执行回调函数，并只执行一次

源码

import { useEffect } from 'react'
import { isFunction } from '../utils'
import isDev from '../utils/isDev'

const useMount = (fn: () => void) => {
  if (isDev) {
    if (!isFunction(fn)) {
      console.error(`useMount: parameter \`fn\` expected to be a function, but got "${typeof fn}".`)
    }
  }

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

export default useMount