antd
Version:
An enterprise-class UI design language and React components implementation
23 lines (22 loc) • 824 B
JavaScript
"use client";
import { useRef } from 'react';
import useLayoutEffect from "rc-util/es/hooks/useLayoutEffect";
import useForceUpdate from '../../_util/hooks/useForceUpdate';
import useResponsiveObserver from '../../_util/responsiveObserver';
function useBreakpoint() {
let refreshOnChange = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : true;
const screensRef = useRef({});
const forceUpdate = useForceUpdate();
const responsiveObserver = useResponsiveObserver();
useLayoutEffect(() => {
const token = responsiveObserver.subscribe(supportScreens => {
screensRef.current = supportScreens;
if (refreshOnChange) {
forceUpdate();
}
});
return () => responsiveObserver.unsubscribe(token);
}, []);
return screensRef.current;
}
export default useBreakpoint;