Send if registrations vector is empty

Commit-Boost · Jan 20, 2025 · 45d7e49 · 45d7e49
1 parent 06d0ba0
commit 45d7e49
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 7 deletions.
diff --git a/crates/common/src/config/pbs.rs b/crates/common/src/config/pbs.rs
@@ -52,7 +52,7 @@ pub struct RelayConfig {
     pub frequency_get_header_ms: Option<u64>,
     /// Maximum number of validators to send to relays in one registration
     /// request
-    #[serde(deserialize_with = "empty_string_as_none")]
+    #[serde(deserialize_with = "empty_string_as_none", default)]
     pub validator_registration_batch_size: Option<usize>,
 }
 

diff --git a/crates/pbs/src/mev_boost/register_validator.rs b/crates/pbs/src/mev_boost/register_validator.rs
@@ -35,20 +35,34 @@ pub async fn register_validator<S: BuilderApiState>(
     let relays = state.all_relays().to_vec();
     let mut handles = Vec::with_capacity(relays.len());
     for relay in relays.clone() {
-        for batch in registrations
-            .chunks(relay.config.validator_registration_batch_size.unwrap_or(registrations.len()))
-            .map(|chunk| chunk.to_vec())
-            .collect::<Vec<Vec<ValidatorRegistration>>>()
-        {
+        if registrations.is_empty() {
             handles.push(tokio::spawn(
                 send_register_validator_with_timeout(
-                    batch,
+                    vec![],
                     relay.clone(),
                     send_headers.clone(),
                     state.pbs_config().timeout_register_validator_ms,
                 )
                 .in_current_span(),
             ));
+        } else {
+            for batch in registrations
+                .chunks(
+                    relay.config.validator_registration_batch_size.unwrap_or(registrations.len()),
+                )
+                .map(|chunk| chunk.to_vec())
+                .collect::<Vec<Vec<ValidatorRegistration>>>()
+            {
+                handles.push(tokio::spawn(
+                    send_register_validator_with_timeout(
+                        batch,
+                        relay.clone(),
+                        send_headers.clone(),
+                        state.pbs_config().timeout_register_validator_ms,
+                    )
+                    .in_current_span(),
+                ));
+            }
         }
     }