Fix reselecting another surrounding document

elastic · Nov 4, 2021 · a438726 · a438726
1 parent 37571a4
commit a438726
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/src/plugins/discover/public/application/apps/context/context_app.tsx b/src/plugins/discover/public/application/apps/context/context_app.tsx
@@ -65,7 +65,7 @@ export const ContextApp = ({ indexPattern, anchorId }: ContextAppProps) => {
    */
   useEffect(() => {
     if (!prevAppState.current || fetchedState.anchor._id !== anchorId) {
-      fetchAllRows();
+      fetchAllRows(!!prevAppState.current);
     } else if (prevAppState.current.predecessorCount !== appState.predecessorCount) {
       fetchSurroundingRows(SurrDocType.PREDECESSORS);
     } else if (prevAppState.current.successorCount !== appState.successorCount) {

diff --git a/src/plugins/discover/public/application/apps/context/utils/use_context_app_fetch.tsx b/src/plugins/discover/public/application/apps/context/utils/use_context_app_fetch.tsx
@@ -161,8 +161,22 @@ export function useContextAppFetch({
   );
 
   const fetchAllRows = useCallback(
-    () => fetchAnchorRow().then((anchor) => anchor && fetchContextRows(anchor)),
-    [fetchAnchorRow, fetchContextRows]
+    (reset: boolean = false) => {
+      if (reset) {
+        const resetState = {
+          ...fetchedState,
+          predecessors: [],
+          successors: [],
+          anchorStatus: { value: LoadingStatus.LOADING },
+          predecessorsStatus: { value: LoadingStatus.LOADING },
+          successorsStatus: { value: LoadingStatus.LOADING },
+        };
+        setFetchedState(resetState);
+      }
+
+      fetchAnchorRow().then((anchor) => anchor && fetchContextRows(anchor));
+    },
+    [fetchAnchorRow, fetchContextRows, setFetchedState, fetchedState]
   );
 
   return {