feat: schedule strategy icons

Author: asuzuki-jumptrading

src/api/atoms.ts

@@ -22,6 +22,7 @@ import {
   VoteDistance,
   BlockEngineUpdate,
   VoteBalance,
+  ScheduleStrategy,
 } from "./types";
 import { rafAtom } from "../atomUtils";
 
@@ -39,6 +40,10 @@ export const startupTimeAtom = atom<
 
 export const tilesAtom = atom<Tile[] | undefined>(undefined);
 
+export const scheduleStrategyAtom = atom<ScheduleStrategy | undefined>(
+  undefined,
+);
+
 export const identityBalanceAtom = atom<IdentityBalance | undefined>(undefined);
 
 export const voteBalanceAtom = atom<VoteBalance | undefined>(undefined);

src/api/entities.ts

@@ -46,6 +46,9 @@ export const identityKeySchema = z.string();
 
 export const startupTimeNanosSchema = z.coerce.bigint();
 
+export const scheduleStrategySchema = z.enum(["perf", "balanced", "revenue"]);
+export const ScheduleStrategyEnum = scheduleStrategySchema.enum;
+
 export const tileTypeSchema = z.enum([
   "sock",
   "net",
@@ -306,6 +309,10 @@ export const summarySchema = z.discriminatedUnion("key", [
     key: z.literal("startup_time_nanos"),
     value: startupTimeNanosSchema,
   }),
+  summaryTopicSchema.extend({
+    key: z.literal("schedule_strategy"),
+    value: scheduleStrategySchema,
+  }),
   summaryTopicSchema.extend({
     key: z.literal("tiles"),
     value: tileSchema.array(),

src/api/types.ts

@@ -38,6 +38,7 @@ import {
   voteStateSchema,
   slotTransactionsSchema,
   voteBalanceSchema,
+  scheduleStrategySchema,
 } from "./entities";
 
 export type Version = z.infer<typeof versionSchema>;
@@ -52,6 +53,8 @@ export type StartupTimeNanos = z.infer<typeof startupTimeNanosSchema>;
 
 export type Tile = z.infer<typeof tileSchema>;
 
+export type ScheduleStrategy = z.infer<typeof scheduleStrategySchema>;
+
 export type IdentityBalance = z.infer<typeof identityBalanceSchema>;
 
 export type VoteBalance = z.infer<typeof voteBalanceSchema>;

src/api/useSetAtomWsData.ts

@@ -19,6 +19,7 @@ import {
   voteDistanceAtom,
   voteStateAtom,
   voteBalanceAtom,
+  scheduleStrategyAtom,
 } from "./atoms";
 import {
   blockEngineSchema,
@@ -70,6 +71,7 @@ export function useSetAtomWsData() {
 
   const setIdentityBalance = useSetAtom(identityBalanceAtom);
   const setVoteBalance = useSetAtom(voteBalanceAtom);
+  const setScheduleStrategy = useSetAtom(scheduleStrategyAtom);
 
   const [startupTime, setStartupTime] = useAtom(startupTimeAtom);
 
@@ -196,6 +198,10 @@ export function useSetAtomWsData() {
             setTiles(value);
             break;
           }
+          case "schedule_strategy": {
+            setScheduleStrategy(value);
+            break;
+          }
           case "identity_balance": {
             setIdentityBalance(value);
             break;

src/features/Header/Cluster.tsx

@@ -4,6 +4,7 @@ import {
   clusterAtom,
   versionAtom,
   commitHashAtom,
+  scheduleStrategyAtom,
 } from "../../api/atoms";
 import { Text, Tooltip } from "@radix-ui/themes";
 import styles from "./cluster.module.css";
@@ -16,6 +17,7 @@ import { getClusterColor } from "./util";
 import { useMedia } from "react-use";
 import { BlockEngineUpdate } from "../../api/types";
 import { connectedColor, connectingColor, failureColor } from "../../colors";
+import { ScheduleStrategyEnum } from "../../api/entities";
 
 export default function Cluster() {
   const cluster = useAtomValue(clusterAtom);
@@ -62,6 +64,8 @@ export default function Cluster() {
       </Tooltip>
 
       <JitoIcon />
+
+      <StrategyIcon />
     </div>
   );
 }
@@ -128,3 +132,27 @@ function JitoIcon() {
     </Tooltip>
   );
 }
+
+function StrategyIcon() {
+  const scheduleStrategy = useAtomValue(scheduleStrategyAtom);
+
+  if (scheduleStrategy === ScheduleStrategyEnum.balanced) {
+    return (
+      <Tooltip content="Transaction scheduler strategy: balanced">
+        <div>⚖️</div>
+      </Tooltip>
+    );
+  } else if (scheduleStrategy === ScheduleStrategyEnum.perf) {
+    return (
+      <Tooltip content="Transaction scheduler strategy: performance">
+        <div style={{ margin: "0 -2px" }}>⚡</div>
+      </Tooltip>
+    );
+  } else if (scheduleStrategy === ScheduleStrategyEnum.revenue) {
+    return (
+      <Tooltip content="Transaction scheduler strategy: revenue">
+        <div>📊</div>
+      </Tooltip>
+    );
+  }
+}