React实现类似Vue的路由监听Hook

React实现类似Vue的路由监听Hook

• 监听路由变化；
• React Hook封装，返回回调函数，新旧路由为函数参数；

代码

import { useEffect, useRef } from 'react';
import { useHistory, useLocation } from 'react-router-dom';

/**
 * 监听路由变化
 * @param callback
 */
const useRouteChange = (callback: (prevLocation: string, newLocation: string) => void) => {
  const location = useLocation();
  const history = useHistory();
  const prevLocation = useRef(location.pathname);

  useEffect(() => {
    const unListen = history.listen(newLocation => {
      callback(prevLocation.current, newLocation.pathname);
      prevLocation.current = newLocation.pathname;
    });

    return () => {
      unListen();
    };
  }, [history, callback]);
};

export default useRouteChange;

使用

  const routeChange = useCallback((prevLocation, newLocation) => {
    console.log(`RouteChange：From ${prevLocation} To ${newLocation}`);
  }, []);

  useRouteChange(routeChange);