fix rpc nodes count after limiting peers msg

Author: jherrera-jump

src/api/entities.ts

@@ -413,6 +413,10 @@ export const peerRemoveSchema = z.object({
   identity_pubkey: z.string(),
 });
 
+const rpcCountUpdateSchema = z.object({
+  rpc_count: z.number(),
+});
+
 const peersUpdateSchema = z.object({
   add: z.array(peerUpdateSchema).optional(),
   update: z.array(peerUpdateSchema).optional(),
@@ -424,6 +428,10 @@ export const peersSchema = z.discriminatedUnion("key", [
     key: z.literal("update"),
     value: peersUpdateSchema,
   }),
+  peersTopicSchema.extend({
+    key: z.literal("rpc"),
+    value: rpcCountUpdateSchema,
+  }),
 ]);
 
 const tsTileTimersSchema = z.object({

src/api/useSetAtomWsData.ts

@@ -39,6 +39,7 @@ import {
   deleteSlotStatusBoundsAtom,
   deleteSlotResponseBoundsAtom,
   skipRateAtom,
+  rpcCountAtom,
 } from "../atoms";
 import {
   EstimatedSlotDuration,
@@ -128,6 +129,7 @@ export function useSetAtomWsData() {
   const addPeers = useSetAtom(addPeersAtom);
   const updatePeers = useSetAtom(updatePeersAtom);
   const removePeers = useSetAtom(removePeersAtom);
+  const setRpcCount = useSetAtom(rpcCountAtom)
 
   const setBlockEngine = useSetAtom(blockEngineAtom);
 
@@ -247,10 +249,19 @@ export function useSetAtomWsData() {
             break;
         }
       } else if (topic === "peers") {
-        const { value } = peersSchema.parse(msg);
-        addPeers(value.add);
-        updatePeers(value.update);
-        removePeers(value.remove);
+        const { key, value } = peersSchema.parse(msg);
+        switch (key) {
+          case "update": {
+            addPeers(value.add);
+            updatePeers(value.update);
+            removePeers(value.remove);
+            break;
+          }
+          case "rpc": {
+            setRpcCount(value.rpc_count);
+            break;
+          }
+        }
       } else if (topic === "slot") {
         const { key, value } = slotSchema.parse(msg);
         switch (key) {

src/atoms.ts

@@ -366,14 +366,21 @@ export const removePeersAtom = atom(null, (get, set, peers?: PeerRemove[]) => {
   }, removePeerDelay);
 });
 
+const _rpcCountAtom = atom<number>(0);
+
+export const rpcCountAtom = atom(
+  (get) => get(_rpcCountAtom),
+  (get, set, count: number) => {
+    set(_rpcCountAtom, count);
+  }
+);
+
 export const peerStatsAtom = atom((get) => {
   const peers = get(peersAtom);
+  const rpcCount = get(rpcCountAtom);
   if (!peers) return;
 
   const activePeers = Object.values(peers).filter((p) => !p.removed);
-  const rpc = activePeers.filter(
-    (p) => p.vote.every((v) => !v.activated_stake) && !!p.gossip,
-  );
   const validators = activePeers.filter((p) =>
     p.vote.some((v) => v.activated_stake),
   );
@@ -395,7 +402,7 @@ export const peerStatsAtom = atom((get) => {
   );
 
   return {
-    rpcCount: rpc.length,
+    rpcCount: rpcCount,
     validatorCount: validators.length,
     activeStake,
     delinquentStake,