Allow refetching of proxy latencies

Emyrk · Emyrk · commit 771d41fa2f3a · 2023-06-01T08:53:51.000-04:00
diff --git a/site/src/contexts/ProxyContext.tsx b/site/src/contexts/ProxyContext.tsx
@@ -101,7 +101,7 @@ export const ProxyProvider: FC<PropsWithChildren> = ({ children }) => {
 
   // Every time we get a new proxiesResponse, update the latency check
   // to each workspace proxy.
-  const proxyLatencies = useProxyLatency(proxiesResp)
+  const { proxyLatencies } = useProxyLatency(proxiesResp)
 
   // updateProxy is a helper function that when called will
   // update the proxy being used.
diff --git a/site/src/contexts/useProxyLatency.ts b/site/src/contexts/useProxyLatency.ts
@@ -1,5 +1,5 @@
 import { Region, RegionsResponse } from "api/typesGenerated"
-import { useEffect, useReducer } from "react"
+import { useEffect, useReducer, useState } from "react"
 import PerformanceObserver from "@fastly/performance-observer-polyfill"
 import axios from "axios"
 import { generateRandomString } from "utils/random"
@@ -25,20 +25,29 @@ const proxyLatenciesReducer = (
   action: ProxyLatencyAction,
 ): Record<string, ProxyLatencyReport> => {
   // Just overwrite any existing latency.
-  return {
-    ...state,
-    [action.proxyID]: action.report,
-  }
+  state[action.proxyID] = action.report
+  return state
 }
 
 export const useProxyLatency = (
   proxies?: RegionsResponse,
-): Record<string, ProxyLatencyReport> => {
+): {
+  // Refetch can be called to refetch the proxy latencies.
+  // Until the new values are loaded, the old values will still be used.
+  refetch: () => void
+  proxyLatencies: Record<string, ProxyLatencyReport>
+} => {
   const [proxyLatencies, dispatchProxyLatencies] = useReducer(
     proxyLatenciesReducer,
     {},
   )
 
+  // This fetchNumber is used to trigger a refetch of the proxy latencies.
+  const [fetchNumber, setFetchNumber] = useState(0)
+  const refetch = (): void => {
+    setFetchNumber(fetchNumber + 1)
+  }
+
   // Only run latency updates when the proxies change.
   useEffect(() => {
     if (!proxies) {
@@ -148,7 +157,10 @@ export const useProxyLatency = (
         // via the performance observer. So we can disconnect the observer.
         observer.disconnect()
       })
-  }, [proxies])
+  }, [proxies, fetchNumber])
 
-  return proxyLatencies
+  return {
+    proxyLatencies,
+    refetch,
+  }
 }
