From 1a6c31414e403dfa79603fc1b0ede7119fdd1fea Mon Sep 17 00:00:00 2001
From: Atridad Lahiji <me@atri.dad>
Date: Sun, 15 Oct 2023 00:44:21 -0300
Subject: [PATCH] Fixed react query retry logic

---
 app/(client)/room/[id]/VoteUI.tsx | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/app/(client)/room/[id]/VoteUI.tsx b/app/(client)/room/[id]/VoteUI.tsx
index 5182ea7..1498939 100644
--- a/app/(client)/room/[id]/VoteUI.tsx
+++ b/app/(client)/room/[id]/VoteUI.tsx
@@ -39,9 +39,14 @@ const VoteUI = () => {
 
   const queryClient = useQueryClient();
 
-  const { data: roomFromDb } = useQuery({
+  const {
+    data: roomFromDb,
+    isLoading: roomFromDbLoading,
+    isFetching: roomFromDbFetching,
+  } = useQuery({
     queryKey: ["room"],
     queryFn: getRoomHandler,
+    retry: false,
   });
 
   const { data: votesFromDb } = useQuery({
@@ -312,11 +317,11 @@ const VoteUI = () => {
   // UI
   // =================================
   // Room is loading
-  if (roomFromDb === undefined) {
+  if (roomFromDbLoading || roomFromDbFetching) {
     return <LoadingIndicator />;
     // Room has been loaded
-  } else if (roomFromDb) {
-    return (
+  } else {
+    return roomFromDb ? (
       <div className="flex flex-col gap-4 text-center justify-center items-center">
         <div className="text-2xl">{roomFromDb.roomName}</div>
         <div className="flex flex-row flex-wrap text-center justify-center items-center gap-1 text-md">
@@ -542,10 +547,9 @@ const VoteUI = () => {
             </>
           )}
       </div>
+    ) : (
+      <NoRoomUI />
     );
-    // Room does not exist
-  } else {
-    return <NoRoomUI />;
   }
 };