fix: add additional ref check on clean up

This commit is contained in:
James 2019-06-17 11:07:02 +01:00
parent b3ba702ef8
commit d18d2d8683

View File

@ -3,27 +3,28 @@ import { RefObject, useEffect, useState } from 'react';
const useScrolling = (ref: RefObject<HTMLElement>): boolean => {
const [scrolling, setScrolling] = useState<boolean>(false);
useEffect(
() => {
if (ref.current) {
let scrollingTimeout;
useEffect(() => {
if (ref.current) {
let scrollingTimeout;
const handleScrollEnd = () => {
setScrolling(false);
};
const handleScrollEnd = () => {
setScrolling(false);
};
const handleScroll = () => {
setScrolling(true);
clearTimeout(scrollingTimeout);
scrollingTimeout = setTimeout(() => handleScrollEnd(), 150);
};
const handleScroll = () => {
setScrolling(true);
clearTimeout(scrollingTimeout);
scrollingTimeout = setTimeout(() => handleScrollEnd(), 150);
};
ref.current.addEventListener('scroll', handleScroll, false);
return () => ref.current.removeEventListener('scroll', handleScroll, false);
}
},
[ref.current],
);
ref.current.addEventListener('scroll', handleScroll, false);
return () => {
if (ref.current) {
ref.current.removeEventListener('scroll', handleScroll, false);
}
};
}
}, [ref.current]);
return scrolling;
};