React Hook监听页面大小

import React, { useState, useEffect } from 'react'

// 函数要以use开头
function useWindowSize() {
    const getWindowSize = () => ({
        innerHeight: window.innerHeight,
        innerWidth: window.innerWidth,
    });

    const [windowSize, setWindowSize] = useState(getWindowSize());

    const handleResize = () => {
        setWindowSize(getWindowSize());
    };

    useEffect(() => {
    	// 监听
        window.addEventListener('resize', handleResize);
        
        // 销毁
        return () => window.removeEventListener('resize', handleResize)
    }, []);

    return windowSize;
}