fix: useUpdateEffect run on the wrong time

This commit is contained in:
砖家 2019-08-12 23:39:59 +08:00 committed by GitHub
parent 2ccde2f7fa
commit 1d5cd10fe4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -3,14 +3,13 @@ import { useEffect, useRef } from 'react';
const useUpdateEffect: typeof useEffect = (effect, deps) => {
const isInitialMount = useRef(true);
useEffect(
isInitialMount.current
? () => {
isInitialMount.current = false;
}
: effect,
deps
);
useEffect(() => {
if (isInitialMount.current) {
isInitialMount.current = false;
} else {
effect();
}
}, deps);
};
export default useUpdateEffect;