Address review comments.

sigp · Jan 30, 2025 · 5f70c41 · 5f70c41
1 parent 8cb49c7
commit 5f70c41
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 11 deletions.
diff --git a/beacon_node/lighthouse_network/src/rpc/protocol.rs b/beacon_node/lighthouse_network/src/rpc/protocol.rs
@@ -386,7 +386,7 @@ impl SupportedProtocol {
             ProtocolId::new(Self::BlocksByRootV1, Encoding::SSZSnappy),
             ProtocolId::new(Self::PingV1, Encoding::SSZSnappy),
         ];
-        if fork_context.fork_exists(ForkName::Fulu) {
+        if fork_context.spec.is_peer_das_scheduled() {
             supported.extend_from_slice(&[
                 // V3 variants have higher preference for protocol negotation
                 ProtocolId::new(Self::MetaDataV3, Encoding::SSZSnappy),
@@ -405,7 +405,7 @@ impl SupportedProtocol {
                 ProtocolId::new(SupportedProtocol::BlobsByRangeV1, Encoding::SSZSnappy),
             ]);
         }
-        if fork_context.fork_exists(ForkName::Fulu) {
+        if fork_context.spec.is_peer_das_scheduled() {
             supported.extend_from_slice(&[
                 ProtocolId::new(SupportedProtocol::DataColumnsByRootV1, Encoding::SSZSnappy),
                 ProtocolId::new(SupportedProtocol::DataColumnsByRangeV1, Encoding::SSZSnappy),

diff --git a/beacon_node/network/src/service.rs b/beacon_node/network/src/service.rs
@@ -25,7 +25,6 @@ use lighthouse_network::{
     MessageId, NetworkEvent, NetworkGlobals, PeerId,
 };
 use slog::{crit, debug, error, info, o, trace, warn};
-use std::borrow::Cow;
 use std::collections::BTreeSet;
 use std::{collections::HashSet, pin::Pin, sync::Arc, time::Duration};
 use store::HotColdDB;
@@ -34,7 +33,7 @@ use task_executor::ShutdownReason;
 use tokio::sync::mpsc;
 use tokio::time::Sleep;
 use types::{
-    ChainSpec, DataColumnSubnetId, EthSpec, ForkContext, ForkName, Slot, SubnetId,
+    ChainSpec, DataColumnSubnetId, EthSpec, ForkContext, Slot, SubnetId,
     SyncCommitteeSubscription, SyncSubnetId, Unsigned, ValidatorSubscription,
 };
 
@@ -752,7 +751,7 @@ impl<T: BeaconChainTypes> NetworkService<T> {
                     }
                 }
 
-                if self.fork_context.fork_exists(ForkName::Fulu) {
+                if self.fork_context.spec.is_peer_das_scheduled() {
                     self.subscribe_to_peer_das_topics(&mut subscribed_topics);
                 }
 
@@ -813,13 +812,11 @@ impl<T: BeaconChainTypes> NetworkService<T> {
     /// `network.subscribe_new_fork_topics()`.
     fn subscribe_to_peer_das_topics(&mut self, subscribed_topics: &mut Vec<GossipTopic>) {
         let column_subnets_to_subscribe = if self.subscribe_all_data_column_subnets {
-            Cow::Owned(
-                (0..self.fork_context.spec.data_column_sidecar_subnet_count)
-                    .map(DataColumnSubnetId::new)
-                    .collect(),
-            )
+            &(0..self.fork_context.spec.data_column_sidecar_subnet_count)
+                .map(DataColumnSubnetId::new)
+                .collect()
         } else {
-            Cow::Borrowed(&self.network_globals.sampling_subnets)
+            &self.network_globals.sampling_subnets
         };
 
         for column_subnet in column_subnets_to_subscribe.iter() {