chore: use server time for shreds

Author: asuzuki-jumptrading

src/api/atoms.ts

@@ -29,6 +29,7 @@ import type {
   GossipPeersSize,
   GossipPeersRowsUpdate,
   GossipPeersCellUpdate,
+  ServerTimeNanos,
 } from "./types";
 import { rafAtom } from "../atomUtils";
 
@@ -61,6 +62,7 @@ export const optimisticallyConfirmedSlotAtom = atom<
 >(undefined);
 
 export const completedSlotAtom = atom<CompletedSlot | undefined>(undefined);
+export const serverTimeNanosAtom = atom<ServerTimeNanos | undefined>(undefined);
 
 export const estimatedSlotAtom = atom<EstimatedSlot | undefined>(undefined);
 

src/api/entities.ts

@@ -121,6 +121,8 @@ export const catchUpHistorySchema = z.object({
   turbine: z.number().array(),
 });
 
+export const serverTimeNanosSchema = z.coerce.number();
+
 export const estimatedSlotSchema = z.number();
 export const resetSlotSchema = z.number().nullable();
 export const storageSlotSchema = z.number().nullable();
@@ -565,6 +567,10 @@ export const summarySchema = z.discriminatedUnion("key", [
     key: z.literal("catch_up_history"),
     value: catchUpHistorySchema,
   }),
+  summaryTopicSchema.extend({
+    key: z.literal("server_time_nanos"),
+    value: serverTimeNanosSchema,
+  }),
 ]);
 
 export const epochNewSchema = z.object({

src/api/types.ts

@@ -57,6 +57,7 @@ import type {
   gossipStorageStatsSchema,
   gossipMessageStatsSchema,
   schedulerCountsSchema,
+  serverTimeNanosSchema,
 } from "./entities";
 
 export type Client = z.infer<typeof clientSchema>;
@@ -94,6 +95,8 @@ export type OptimisticallyConfirmedSlot = z.infer<
 export type CompletedSlot = z.infer<typeof completedSlotSchema>;
 export type CatchUpHistory = z.infer<typeof catchUpHistorySchema>;
 
+export type ServerTimeNanos = z.infer<typeof serverTimeNanosSchema>;
+
 // export type SlotCompleted = z.infer<typeof slotCompletedSchema>;
 
 export type EstimatedSlot = z.infer<typeof estimatedSlotSchema>;

src/api/useSetAtomWsData.ts

@@ -27,6 +27,7 @@ import {
   gossipPeersSizeAtom,
   gossipPeersRowsUpdateAtom,
   gossipPeersCellUpdateAtom,
+  serverTimeNanosAtom,
 } from "./atoms";
 import {
   blockEngineSchema,
@@ -196,6 +197,7 @@ export function useSetAtomWsData() {
   const setBlockEngine = useSetAtom(blockEngineAtom);
 
   const setCompletedSlot = useSetAtom(completedSlotAtom);
+  const setServerTimeNanos = useSetAtom(serverTimeNanosAtom);
 
   const addSkippedClusterSlots = useSetAtom(addSkippedClusterSlotsAtom);
   const deleteSkippedClusterSlot = useSetAtom(deleteSkippedClusterSlotAtom);
@@ -383,6 +385,10 @@ export function useSetAtomWsData() {
             addRepairSlots(value.repair);
             break;
           }
+          case "server_time_nanos": {
+            setServerTimeNanos(value);
+            break;
+          }
           case "root_slot":
           case "optimistically_confirmed_slot":
           case "estimated_slot":

src/features/Overview/ShredsProgression/shredsProgressionPlugin.ts

@@ -22,7 +22,8 @@ import { skippedClusterSlotsAtom } from "../../../atoms";
 import { clamp } from "lodash";
 import { ShredEvent } from "../../../api/entities";
 import { getSlotGroupLabelId, getSlotLabelId } from "./utils";
-import { slotsPerLeader } from "../../../consts";
+import { nsPerMs, slotsPerLeader } from "../../../consts";
+import { serverTimeNanosAtom } from "../../../api/atoms";
 
 const store = getDefaultStore();
 export const shredsXScaleKey = "shredsXScaleKey";
@@ -55,10 +56,16 @@ export function shredsProgressionPlugin(
           const minCompletedSlot = store.get(atoms.minCompletedSlot);
           const skippedSlotsCluster = store.get(skippedClusterSlotsAtom);
           const rangeAfterStartup = store.get(atoms.rangeAfterStartup);
+          const serverTimeNanos = store.get(serverTimeNanosAtom);
 
           const maxX = u.scales[shredsXScaleKey].max;
 
-          if (!liveShreds || !slotRange || maxX == null) {
+          if (
+            !liveShreds ||
+            !slotRange ||
+            maxX == null ||
+            serverTimeNanos == null
+          ) {
             return;
           }
 
@@ -74,7 +81,7 @@ export function shredsProgressionPlugin(
           }
 
           // Offset to convert shred event delta to chart x value
-          const delayedNow = Date.now() - delayMs;
+          const delayedNow = Math.trunc(serverTimeNanos / nsPerMs) - delayMs;
 
           const tsXValueOffset = delayedNow - liveShreds.referenceTs;