useTimeout.ts 945 字节
Newer Older
V
vben 已提交
1
import { ref, watch } from 'vue';
V
Vben 已提交
2
import { tryOnUnmounted } from '@vueuse/core';
V
vben 已提交
3 4
import { isFunction } from '/@/utils/is';

5
export function useTimeoutFn(handle: Fn<any>, wait: number, native = false) {
V
vben 已提交
6 7 8 9 10
  if (!isFunction(handle)) {
    throw new Error('handle is not Function!');
  }

  const { readyRef, stop, start } = useTimeoutRef(wait);
11 12 13 14 15 16 17 18 19 20 21
  if (native) {
    handle();
  } else {
    watch(
      readyRef,
      (maturity) => {
        maturity && handle();
      },
      { immediate: false }
    );
  }
V
vben 已提交
22 23 24 25 26 27
  return { readyRef, stop, start };
}

export function useTimeoutRef(wait: number) {
  const readyRef = ref(false);

V
vben 已提交
28
  let timer: TimeoutHandle;
V
vben 已提交
29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
  function stop(): void {
    readyRef.value = false;
    timer && window.clearTimeout(timer);
  }
  function start(): void {
    stop();
    timer = setTimeout(() => {
      readyRef.value = true;
    }, wait);
  }

  start();

  tryOnUnmounted(stop);

  return { readyRef, stop, start };
}