37 lines
1017 B
JavaScript
37 lines
1017 B
JavaScript
|
import raf from "rc-util/es/raf";
|
||
|
import * as React from 'react';
|
||
|
import { useRef } from 'react';
|
||
|
export default function useFrameState(defaultValue) {
|
||
|
const [value, setValue] = React.useState(defaultValue);
|
||
|
const frameRef = useRef(null);
|
||
|
const batchRef = useRef([]);
|
||
|
const destroyRef = useRef(false);
|
||
|
React.useEffect(() => {
|
||
|
destroyRef.current = false;
|
||
|
return () => {
|
||
|
destroyRef.current = true;
|
||
|
raf.cancel(frameRef.current);
|
||
|
frameRef.current = null;
|
||
|
};
|
||
|
}, []);
|
||
|
function setFrameValue(updater) {
|
||
|
if (destroyRef.current) {
|
||
|
return;
|
||
|
}
|
||
|
if (frameRef.current === null) {
|
||
|
batchRef.current = [];
|
||
|
frameRef.current = raf(() => {
|
||
|
frameRef.current = null;
|
||
|
setValue(prevValue => {
|
||
|
let current = prevValue;
|
||
|
batchRef.current.forEach(func => {
|
||
|
current = func(current);
|
||
|
});
|
||
|
return current;
|
||
|
});
|
||
|
});
|
||
|
}
|
||
|
batchRef.current.push(updater);
|
||
|
}
|
||
|
return [value, setFrameValue];
|
||
|
}
|