ant-design/components/watermark/useRafDebounce.ts

27 lines
560 B
TypeScript
Raw Normal View History

import React from 'react';
import raf from 'rc-util/lib/raf';
import { useEvent } from 'rc-util';
/**
* Callback will only execute last one for each raf
*/
export default function useRafDebounce(callback: VoidFunction) {
const executeRef = React.useRef(false);
const rafRef = React.useRef<number>();
const wrapperCallback = useEvent(callback);
return () => {
if (executeRef.current) {
return;
}
executeRef.current = true;
wrapperCallback();
rafRef.current = raf(() => {
executeRef.current = false;
});
};
}