javascript 反应路由器挂钩

Posted 2021-05-03

import { useContext, useEffect } from 'react';
import { __RouterContext } from 'react-router';
import useForceUpdate from 'use-force-update';

const useRouter = () => {
	// Re Rendering
	const forceUpdate = useForceUpdate();
	// Getting Router Context
	const routerContext = useContext(__RouterContext);

	useEffect(() => routerContext.history.listen(forceUpdate), [routerContext]);

	return routerContext;
};

export default useRouter;