Merge branch 'feat/#196' of https://github.com/boostcampwm-2022/web08…

…-ChoBab into feat/#196
boostcampwm-2022 · Dec 13, 2022 · 94d4a03 · 94d4a03
2 parents a477a4c + d31240a
commit 94d4a03
Showing 1 changed file with 17 additions and 6 deletions.
diff --git a/client/src/components/RestaurantVoteButton/index.tsx b/client/src/components/RestaurantVoteButton/index.tsx
@@ -56,12 +56,12 @@ function RestaurantVoteButton({
       return;
     }
 
-    socket.on('userVoteRestaurantIdList', (result: VoteRestaurantListType) => {
+    const handleUserVoteRestaurantIdList = (result: VoteRestaurantListType) => {
       votedRestaurantListRef.current = result.data.voteRestaurantIdList;
       setIsVoted(votedRestaurantListRef.current.includes(restaurantId));
-    });
+    }
 
-    socket.on('cancelVoteRestaurantResult', (result: VoteResultType) => {
+    const handleCancelVoteRestaurantResult = (result: VoteResultType) => {
       if (restaurantId !== result.data?.restaurantId) {
         return;
       }
@@ -76,9 +76,9 @@ function RestaurantVoteButton({
         return;
       }
       setIsVoted(false);
-    });
+    }
 
-    socket.on('voteRestaurantResult', (result: VoteResultType) => {
+    const handleVoteRestaurantResult = (result: VoteResultType) => {
       if (restaurantId !== result.data?.restaurantId) {
         return;
       }
@@ -93,9 +93,20 @@ function RestaurantVoteButton({
         return;
       }
       setIsVoted(true);
-    });
+    }
+
+    socket.on('userVoteRestaurantIdList', handleUserVoteRestaurantIdList);
+    socket.on('cancelVoteRestaurantResult', handleCancelVoteRestaurantResult);
+    socket.on('voteRestaurantResult', handleVoteRestaurantResult);
 
     setVotedRestaurantList();
+
+    // eslint-disable-next-line consistent-return
+    return () => {
+      socket.removeListener('userVoteRestaurantIdList', handleUserVoteRestaurantIdList);
+      socket.removeListener('voteRestaurantResult', handleVoteRestaurantResult);
+      socket.removeListener('cancelVoteRestaurantResult', handleCancelVoteRestaurantResult);
+    }
   }, []);
 
   const voteRestaurant = () => {