ant-design/components/watermark/useRafDebounce.ts
2024-10-02 10:43:35 +08:00

27 lines
575 B
TypeScript

import React from 'react';
import useEvent from 'rc-util/lib/hooks/useEvent';
import raf from 'rc-util/lib/raf';
/**
* 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;
});
};
}