fix(site): proxy menu not showing latency (coder#7785)

BrunoQuaresma · web-flow · commit 9dbbe82cf8fe · 2023-06-01T20:03:27.000-03:00
diff --git a/site/src/components/Navbar/NavbarView.tsx b/site/src/components/Navbar/NavbarView.tsx
@@ -192,8 +192,11 @@ const ProxyMenu: FC<{ proxyContextValue: ProxyContextValue }> = ({
   const refreshLatencies = proxyContextValue.refetchProxyLatencies
   const closeMenu = () => setIsOpen(false)
   const navigate = useNavigate()
+  const latencies = proxyContextValue.proxyLatencies
+  const isLoadingLatencies = Object.keys(latencies).length === 0
+  const isLoading = proxyContextValue.isLoading || isLoadingLatencies
 
-  if (!proxyContextValue.isFetched) {
+  if (isLoading) {
     return (
       <Skeleton
         width="160px"
@@ -230,9 +233,7 @@ const ProxyMenu: FC<{ proxyContextValue: ProxyContextValue }> = ({
             {selectedProxy.display_name}
             <ProxyStatusLatency
               proxy={selectedProxy}
-              latency={
-                proxyContextValue.proxyLatencies?.[selectedProxy.id]?.latencyMS
-              }
+              latency={latencies?.[selectedProxy.id]?.latencyMS}
             />
           </Box>
         ) : (
@@ -278,9 +279,7 @@ const ProxyMenu: FC<{ proxyContextValue: ProxyContextValue }> = ({
               {proxy.display_name}
               <ProxyStatusLatency
                 proxy={proxy}
-                latency={
-                  proxyContextValue.proxyLatencies?.[proxy.id]?.latencyMS
-                }
+                latency={latencies?.[proxy.id]?.latencyMS}
               />
             </Box>
           </MenuItem>
diff --git a/site/src/contexts/useProxyLatency.ts b/site/src/contexts/useProxyLatency.ts
@@ -24,9 +24,10 @@ const proxyLatenciesReducer = (
   state: Record<string, ProxyLatencyReport>,
   action: ProxyLatencyAction,
 ): Record<string, ProxyLatencyReport> => {
-  // Just overwrite any existing latency.
-  state[action.proxyID] = action.report
-  return state
+  return {
+    ...state,
+    [action.proxyID]: action.report,
+  }
 }
 
 export const useProxyLatency = (
