Drop start stop pattern from Network where possible

ChainSafe · May 30, 2023 · 961c559 · 961c559
1 parent 0e93c07
commit 961c559
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 18 deletions.
diff --git a/packages/beacon-node/src/network/core/networkCore.ts b/packages/beacon-node/src/network/core/networkCore.ts
@@ -213,15 +213,13 @@ export class NetworkCore implements INetworkCore {
     await reqResp.start();
 
     await gossip.start();
-    attnetsService.start();
-    syncnetsService.start();
 
     // Network spec decides version changes based on clock fork, not head fork
     const forkCurrentSlot = config.getForkName(clock.currentSlot);
     // Register only ReqResp protocols relevant to clock's fork
     reqResp.registerProtocolsAtFork(forkCurrentSlot);
 
-    await peerManager.start();
+    await peerManager.startDiscovery();
 
     // Bind discv5's ENR to local metadata
     discv5 = peerManager["discovery"]?.discv5;
@@ -255,14 +253,14 @@ export class NetworkCore implements INetworkCore {
 
     // Must goodbye and disconnect before stopping libp2p
     await this.peerManager.goodbyeAndDisconnectAllPeers();
-    await this.peerManager.stop();
+    await this.peerManager.close();
     await this.gossip.stop();
 
     await this.reqResp.stop();
     await this.reqResp.unregisterAllProtocols();
 
-    this.attnetsService.stop();
-    this.syncnetsService.stop();
+    this.attnetsService.close();
+    this.syncnetsService.close();
     await this.libp2p.stop();
 
     this.closed = true;

diff --git a/packages/beacon-node/src/network/peers/peerManager.ts b/packages/beacon-node/src/network/peers/peerManager.ts
@@ -175,10 +175,7 @@ export class PeerManager {
     if (metrics) {
       metrics.peers.addCollect(() => this.runPeerCountMetrics(metrics));
     }
-  }
 
-  async start(): Promise<void> {
-    await this.discovery?.start();
     this.libp2p.connectionManager.addEventListener(Libp2pEvent.peerConnect, this.onLibp2pPeerConnect);
     this.libp2p.connectionManager.addEventListener(Libp2pEvent.peerDisconnect, this.onLibp2pPeerDisconnect);
     this.networkEventBus.on(NetworkEvent.reqRespRequest, this.onRequest);
@@ -195,7 +192,11 @@ export class PeerManager {
     ];
   }
 
-  async stop(): Promise<void> {
+  async startDiscovery(): Promise<void> {
+    await this.discovery?.start();
+  }
+
+  async close(): Promise<void> {
     await this.discovery?.stop();
     this.libp2p.connectionManager.removeEventListener(Libp2pEvent.peerConnect, this.onLibp2pPeerConnect);
     this.libp2p.connectionManager.removeEventListener(Libp2pEvent.peerDisconnect, this.onLibp2pPeerDisconnect);

diff --git a/packages/beacon-node/src/network/subnets/attnetsService.ts b/packages/beacon-node/src/network/subnets/attnetsService.ts
@@ -93,14 +93,12 @@ export class AttnetsService implements IAttnetsService {
     if (metrics) {
       metrics.attnetsService.subscriptionsRandom.addCollect(() => this.onScrapeLodestarMetrics(metrics));
     }
-  }
 
-  start(): void {
     this.clock.on(ClockEvent.slot, this.onSlot);
     this.clock.on(ClockEvent.epoch, this.onEpoch);
   }
 
-  stop(): void {
+  close(): void {
     this.clock.off(ClockEvent.slot, this.onSlot);
     this.clock.off(ClockEvent.epoch, this.onEpoch);
   }

diff --git a/packages/beacon-node/src/network/subnets/interface.ts b/packages/beacon-node/src/network/subnets/interface.ts
@@ -12,8 +12,7 @@ export type CommitteeSubscription = {
 };
 
 export type SubnetsService = {
-  start(): void;
-  stop(): void;
+  close(): void;
   addCommitteeSubscriptions(subscriptions: CommitteeSubscription[]): void;
   getActiveSubnets(): RequestedSubnet[];
   subscribeSubnetsToNextFork(nextFork: ForkName): void;

diff --git a/packages/beacon-node/src/network/subnets/syncnetsService.ts b/packages/beacon-node/src/network/subnets/syncnetsService.ts
@@ -39,13 +39,11 @@ export class SyncnetsService implements SubnetsService {
     if (metrics) {
       metrics.syncnetsService.subscriptionsCommittee.addCollect(() => this.onScrapeLodestarMetrics(metrics));
     }
-  }
 
-  start(): void {
     this.clock.on(ClockEvent.epoch, this.onEpoch);
   }
 
-  stop(): void {
+  close(): void {
     this.clock.off(ClockEvent.epoch, this.onEpoch);
   }