fix: 스크롤 복원 버그

go-cham · Aug 3, 2023 · e23fbb5 · e23fbb5
1 parent fbf9da0
commit e23fbb5
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 15 deletions.
diff --git a/src/components/user/SelectMyPostType.tsx b/src/components/user/SelectMyPostType.tsx
@@ -1,7 +1,4 @@
-import { useSetAtom } from 'jotai';
-
 import { PostType } from '@/pages/user/UserPage';
-import { scrollRestorationAtom } from '@/states/scroll-restoration';
 
 interface SelectMyPostTypeProps {
   postType: PostType;
@@ -14,8 +11,6 @@ export default function SelectMyPostType({
   switchPostType,
   postingCount,
 }: SelectMyPostTypeProps) {
-  const setScrollRestoration = useSetAtom(scrollRestorationAtom);
-
   return (
     <div className="mx-[2.5rem] mb-[2.1rem] flex justify-between space-x-[1.2rem] overflow-hidden rounded-[2.35rem] bg-background-subBg-100">
       {['내 게시글', '참여한 게시글'].map((label) => {

diff --git a/src/hooks/useScrollRestoration.tsx b/src/hooks/useScrollRestoration.tsx
@@ -11,6 +11,13 @@ export default function useScrollRestoration<U extends HTMLElement>(
   );
   const ref = useRef<U | null>(null);
 
+  useEffect(() => {
+    setScrollRestoration((prevScrollRestoration) => ({
+      ...prevScrollRestoration,
+      [key]: 0,
+    }));
+  }, []);
+
   useEffect(() => {
     if (!ref.current) return;
 
@@ -19,16 +26,10 @@ export default function useScrollRestoration<U extends HTMLElement>(
 
       const height = ref.current.scrollTop;
 
-      if (!scrollRestoration) {
-        setScrollRestoration({
-          [key]: height,
-        });
-      } else {
-        setScrollRestoration((prevScrollRestoration) => ({
-          ...prevScrollRestoration,
-          [key]: height,
-        }));
-      }
+      setScrollRestoration((prevScrollRestoration) => ({
+        ...prevScrollRestoration,
+        [key]: height,
+      }));
     };
 
     ref.current.addEventListener('scroll', handleScroll);