sim-rs: track vote bundle sizes

Author: SupernaviX

data/simulation/config.d.ts

@@ -83,9 +83,7 @@ export interface Config {
   "vote-generation-cpu-time-ms-per-ib": number;
   "vote-validation-cpu-time-ms": number;
   "vote-threshold": bigint;
-  /** Only supported by Haskell simulation. */
   "vote-bundle-size-bytes-constant": bigint;
-  /** Only supported by Haskell simulation. */
   "vote-bundle-size-bytes-per-eb": bigint;
   /** Only supported by Haskell simulation. */
   "vote-diffusion-strategy": DiffusionStrategy;

data/simulation/config.schema.json

@@ -281,13 +281,11 @@
     },
     "vote-bundle-size-bytes-constant": {
       "additionalProperties": false,
-      "description": "Only supported by Haskell simulation.",
       "properties": {},
       "type": "number"
     },
     "vote-bundle-size-bytes-per-eb": {
       "additionalProperties": false,
-      "description": "Only supported by Haskell simulation.",
       "properties": {},
       "type": "number"
     },

sim-rs/sim-cli/src/events.rs

@@ -307,7 +307,7 @@ impl EventMonitor {
                     eb_messages.received += 1;
                 }
                 Event::VoteLotteryWon { .. } => {}
-                Event::VotesGenerated { id, votes } => {
+                Event::VotesGenerated { id, votes, .. } => {
                     for (eb, count) in votes.0 {
                         total_votes += count as u64;
                         *votes_per_bundle.entry(id.clone()).or_default() += count as f64;

sim-rs/sim-core/src/config.rs

@@ -98,8 +98,8 @@ pub struct RawParameters {
     pub vote_generation_cpu_time_ms_per_ib: f64,
     pub vote_validation_cpu_time_ms: f64,
     pub vote_threshold: u64,
-    // pub vote_size_bytes_constant: u64,
-    // pub vote_size_bytes_per_node: u64,
+    pub vote_bundle_size_bytes_constant: u64,
+    pub vote_bundle_size_bytes_per_eb: u64,
 
     // Certificate configuration
     pub cert_generation_cpu_time_ms_constant: f64,
@@ -311,6 +311,8 @@ pub(crate) struct BlockSizeConfig {
     pub ib_header: u64,
     eb_constant: u64,
     eb_per_ib: u64,
+    vote_constant: u64,
+    vote_per_eb: u64,
 }
 
 impl BlockSizeConfig {
@@ -322,6 +324,8 @@ impl BlockSizeConfig {
             ib_header: params.ib_head_size_bytes,
             eb_constant: params.eb_size_bytes_constant,
             eb_per_ib: params.eb_size_bytes_per_ib,
+            vote_constant: params.vote_bundle_size_bytes_constant,
+            vote_per_eb: params.vote_bundle_size_bytes_per_eb,
         }
     }
 
@@ -332,6 +336,10 @@ impl BlockSizeConfig {
     pub fn eb(&self, ibs: usize) -> u64 {
         self.eb_constant + self.eb_per_ib * ibs as u64
     }
+
+    pub fn vote_bundle(&self, ebs: usize) -> u64 {
+        self.vote_constant + self.vote_per_eb * ebs as u64
+    }
 }
 
 #[derive(Debug, Clone)]

sim-rs/sim-core/src/events.rs

@@ -163,6 +163,7 @@ pub enum Event {
     VotesGenerated {
         #[serde(flatten)]
         id: VoteBundleId<Node>,
+        bytes: u64,
         votes: Votes<Node>,
     },
     NoVote {
@@ -389,6 +390,7 @@ impl EventTracker {
     pub fn track_votes_generated(&self, votes: &VoteBundle) {
         self.send(Event::VotesGenerated {
             id: self.to_vote_bundle(votes.id),
+            bytes: votes.bytes,
             votes: Votes(
                 votes
                     .ebs

sim-rs/sim-core/src/model.rs

@@ -189,6 +189,7 @@ impl<Node: Display + Serialize> Serialize for VoteBundleId<Node> {
 #[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
 pub struct VoteBundle {
     pub id: VoteBundleId,
+    pub bytes: u64,
     pub ebs: BTreeMap<EndorserBlockId, usize>,
 }
 

sim-rs/sim-core/src/sim/node.rs

@@ -622,13 +622,13 @@ impl Node {
         }
         // For every VRF lottery you won, you can vote for every EB
         let votes_allowed = vote_count * ebs.len();
-        let eb_counts = ebs.into_iter().map(|eb| (eb, votes_allowed)).collect();
         let votes = VoteBundle {
             id: VoteBundleId {
                 slot,
                 producer: self.id,
             },
-            ebs: eb_counts,
+            bytes: self.sim_config.sizes.vote_bundle(ebs.len()),
+            ebs: ebs.into_iter().map(|eb| (eb, votes_allowed)).collect(),
         };
         if !votes.ebs.is_empty() {
             self.schedule_cpu_task(CpuTask::VoteBundleGenerated(votes));

ui/src/app/api/messages/batch/utils.ts

@@ -202,7 +202,7 @@ export const processMessage = (
       message.producer.toString(),
       "votesGenerated",
     );
-    // TODO: track vote bundle size
+    intermediate.bytes.set(`votes-${message.id}`, message.bytes);
   } else if (message.type === EMessageType.VotesSent) {
     incrementNodeAggregationData(
       aggregatedData,

ui/src/components/Sim/types.ts

@@ -191,6 +191,7 @@ export interface IVotesGenerated {
   id: string;
   slot: number;
   producer: string;
+  bytes: number;
   votes: any[] // @todo
 }