apollo_network_benchmark: add RoundRobin mode to enum and round_duration_seconds arg

Author: sirandreww-starkware

crates/apollo_network_benchmark/src/bin/broadcast_network_stress_test_node/stress_test_node.rs

@@ -98,7 +98,7 @@ impl BroadcastNetworkStressTestNode {
     /// Determines if this node should broadcast messages based on the mode
     pub fn should_broadcast(&self) -> bool {
         match self.args.user.mode {
-            Mode::AllBroadcast => true,
+            Mode::AllBroadcast | Mode::RoundRobin => true,
             Mode::OneBroadcast => {
                 let broadcaster_id = Self::get_broadcaster_id(&self.args);
                 self.args.runner.id == broadcaster_id

crates/apollo_network_benchmark/src/node_args.rs

@@ -11,6 +11,9 @@ pub enum Mode {
     /// Only the node specified by --broadcaster broadcasts messages
     #[value(name = "one")]
     OneBroadcast,
+    /// Nodes take turns broadcasting in round-robin fashion
+    #[value(name = "rr")]
+    RoundRobin,
 }
 
 #[derive(Debug, Clone, ValueEnum, PartialEq, Eq, Serialize, Deserialize)]
@@ -83,6 +86,10 @@ pub struct UserArgs {
     #[arg(long, env, required_if_eq("mode", "one"))]
     pub broadcaster: Option<u64>,
 
+    /// Duration each node broadcasts before switching (in seconds) - for RoundRobin mode
+    #[arg(long, env, default_value = "3")]
+    pub round_duration_seconds: u64,
+
     /// Size of StressTestMessage
     #[arg(long, env, default_value = "1024")]
     pub message_size_bytes: usize,