Fix chain_spec for BlockService

sigp · Dec 19, 2024 · 838eed6 · 838eed6
1 parent 79367de
commit 838eed6
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/validator_client/validator_services/src/block_service.rs b/validator_client/validator_services/src/block_service.rs
@@ -51,7 +51,7 @@ pub struct BlockServiceBuilder<S, T> {
     beacon_nodes: Option<Arc<BeaconNodeFallback<T>>>,
     proposer_nodes: Option<Arc<BeaconNodeFallback<T>>>,
     executor: Option<TaskExecutor>,
-    chain_spec: Option<ChainSpec>,
+    chain_spec: Option<Arc<ChainSpec>>,
     graffiti: Option<Graffiti>,
     graffiti_file: Option<GraffitiFile>,
 }
@@ -95,6 +95,11 @@ impl<S: ValidatorStore, T: SlotClock + 'static> BlockServiceBuilder<S, T> {
         self
     }
 
+    pub fn chain_spec(mut self, chain_spec: Arc<ChainSpec>) -> Self {
+        self.chain_spec = Some(chain_spec);
+        self
+    }
+
     pub fn graffiti(mut self, graffiti: Option<Graffiti>) -> Self {
         self.graffiti = graffiti;
         self
@@ -189,7 +194,7 @@ pub struct Inner<S, T> {
     pub beacon_nodes: Arc<BeaconNodeFallback<T>>,
     pub proposer_nodes: Option<Arc<BeaconNodeFallback<T>>>,
     executor: TaskExecutor,
-    chain_spec: ChainSpec,
+    chain_spec: Arc<ChainSpec>,
     graffiti: Option<Graffiti>,
     graffiti_file: Option<GraffitiFile>,
 }