add refetch

DSGT-DLP · May 1, 2024 · a0f4272 · a0f4272
1 parent 0724b98
commit a0f4272
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/frontend/src/pages/train/[train_space_id].tsx b/frontend/src/pages/train/[train_space_id].tsx
@@ -152,7 +152,7 @@ const mapTrainResultsDataToCharts = (
 
 const TrainSpace = () => {
   const { train_space_id } = useRouter().query;
-  const { data, isLoading } = useGetTrainResultsDataQuery({
+  const { data, isLoading, refetch, error } = useGetTrainResultsDataQuery({
     trainspaceId: train_space_id,
   });
 
@@ -163,6 +163,11 @@ const TrainSpace = () => {
       router.replace({ pathname: "/login" });
     }
   }, [user, router.isReady]);
+
+  if (error) {
+    setTimeout(() => refetch(), 3000);
+  }
+
   if (!isSignedIn(user) || !data || isLoading) {
     return <></>;
   }