add hook useMemoSingleThreaded
This commit is contained in:
parent
3ed7f157df
commit
a26c2ab3be
38
apps/photos/src/hooks/useMemoSingleThreaded.tsx
Normal file
38
apps/photos/src/hooks/useMemoSingleThreaded.tsx
Normal file
|
@ -0,0 +1,38 @@
|
|||
import { useEffect, useRef, useState } from 'react';
|
||||
|
||||
export default function useMemoSingleThreaded<T>(
|
||||
fn: () => T | Promise<T>,
|
||||
deps: any[]
|
||||
): T {
|
||||
const [result, setResult] = useState<T>(null);
|
||||
const updateInProgress = useRef(false);
|
||||
const updateRequired = useRef(false);
|
||||
useEffect(() => {
|
||||
const main = async () => {
|
||||
if (updateInProgress.current) {
|
||||
updateRequired.current = true;
|
||||
return;
|
||||
}
|
||||
updateInProgress.current = true;
|
||||
const result = fn();
|
||||
if (isPromise(result)) {
|
||||
const resultValue = await result;
|
||||
setResult(resultValue);
|
||||
} else {
|
||||
setResult(result);
|
||||
}
|
||||
updateInProgress.current = false;
|
||||
if (updateRequired.current) {
|
||||
updateRequired.current = false;
|
||||
setTimeout(main, 0);
|
||||
}
|
||||
};
|
||||
main();
|
||||
}, deps);
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
function isPromise<T>(obj: T | Promise<T>): obj is Promise<T> {
|
||||
return obj && typeof (obj as any).then === 'function';
|
||||
}
|
Loading…
Reference in a new issue