feat(global-stats): get median TTFB from header

filecoin-saturn · Oct 2, 2023 · acf39a9 · acf39a9
1 parent 9b18d73
commit acf39a9
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/src/contexts/NodesContext.tsx b/src/contexts/NodesContext.tsx
@@ -11,11 +11,13 @@ import { EntityType } from "./AppContext";
 type NodesContextType = {
   nodes: Node[] | [];
   setNodes: (nodes: Node[] | []) => void;
+  globalStats: { medianTTFB: number };
 };
 
 const initialValues = {
   nodes: [],
   setNodes: () => {},
+  globalStats: { medianTTFB: 0 },
 };
 
 const NodesContext = createContext<NodesContextType>(initialValues);
@@ -28,6 +30,7 @@ export const NodesContextProvider = ({
   const [nodes, setNodes] = useState<Node[] | []>([]);
   const [tempNodes, setTempNodes] = useState<Node[] | []>([]);
   const [isLoading, setIsLoading] = useState<boolean>();
+  const [globalStats, setGlobalStats] = useState<{ medianTTFB: number }>({ medianTTFB: 0 });
 
   useEffect(() => {
     let animationFrameId: number;
@@ -39,6 +42,9 @@ export const NodesContextProvider = ({
       const decoder = new TextDecoder();
       const reader = response.body?.getReader();
       const nodesMap = new Map<string, Node>();
+      const medianTTFB = Number(response.headers.get("x-saturn-median-ttfb")) || 0;
+
+      setGlobalStats({ medianTTFB });
 
       const onChunk = ({
         done,
@@ -99,6 +105,7 @@ export const NodesContextProvider = ({
   }, [isLoading]);
 
   const value = {
+    globalStats,
     nodes,
     setNodes,
   };

diff --git a/src/hooks/useNodes.tsx b/src/hooks/useNodes.tsx
@@ -52,7 +52,7 @@ export type Node = {
 export type Nodes = Node[];
 
 export const useNodes = () => {
-  const { nodes } = useContext(NodesContext);
+  const { nodes, globalStats } = useContext(NodesContext);
 
   const getNodeByID = (queryNodeId: string) => {
     return nodes.find((node) => node.id === queryNodeId);
@@ -78,6 +78,7 @@ export const useNodes = () => {
 
   return {
     nodes,
+    globalStats,
     getNodeByID,
     getNodesByLocationId,
     getNodesByCountryId,

diff --git a/src/hooks/useStats.tsx b/src/hooks/useStats.tsx
@@ -98,6 +98,7 @@ const computeStats = (
 export const useStats = () => {
   const {
     nodes: nodesMap,
+    globalStats,
     getNodesByContinentId,
     getNodesByCountryId,
     getNodesByLocationId,
@@ -111,8 +112,8 @@ export const useStats = () => {
 
   const getGlobalStats = () => {
     if (!nodes) return;
-    const globalStats = computeStats(nodes, undefined);
-    return globalStats;
+    const computedGlobalStats = computeStats(nodes, undefined);
+    return { ...computedGlobalStats, ...globalStats };
   };
 
   const getStatsByContinentId = (continentId: string) => {