17 lines
360 B
TypeScript
17 lines
360 B
TypeScript
|
import { useRef, useLayoutEffect, useCallback } from "react";
|
||
|
|
|
||
|
|
export function useCallbackRef<T extends (...args: any[]) => any>(
|
||
|
|
callback: T,
|
||
|
|
): T {
|
||
|
|
const callbackRef = useRef(callback);
|
||
|
|
|
||
|
|
useLayoutEffect(() => {
|
||
|
|
callbackRef.current = callback;
|
||
|
|
});
|
||
|
|
|
||
|
|
return useCallback(
|
||
|
|
((...args) => callbackRef.current?.(...args)) as T,
|
||
|
|
[],
|
||
|
|
) as T;
|
||
|
|
}
|