useState改装版

308 阅读1分钟
import {useEffect, useState, useRef} from "react";
 
 
function useCallbackState (od) {
    const cbRef = useRef();
    const [data, setData] = useState(od);
 
    useEffect(() => {
        cbRef.current && cbRef.current(data);
    }, [data]);
 
    return [data, function (d, callback) {
        cbRef.current = callback;
        setData(d);
    }];
}
 
export {useCallbackState};
const [data,setData] = useCallbackState({});
 
setData({}, function (data) {
    console.log("哈哈哈哈哈", data);
})