status-im
/
nimbus-eth2
mirror of https://github.com/status-im/nimbus-eth2.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

add: subscribeAllSubnets feature

This commit is contained in:
Agnish Ghosh 2024-07-15 13:50:41 +05:30
parent b33900bd35
commit e034f30b5e
No known key found for this signature in database
GPG Key ID: 7BDDA05D1B25E9F8
2 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
beacon_chain/nimbus_beacon_node.nim
View File

@ -1130,10 +1130,11 @@ proc addCapellaMessageHandlers(
node.addAltairMessageHandlers(forkDigest, slot) node.addAltairMessageHandlers(forkDigest, slot)
node.network.subscribe(getBlsToExecutionChangeTopic(forkDigest), basicParams) node.network.subscribe(getBlsToExecutionChangeTopic(forkDigest), basicParams)
proc fetchCustodySubnetCount* (node: BeaconNode, res: var uint64)= proc fetchCustodySubnetCount* (node: BeaconNode): uint64=
res = CUSTODY_REQUIREMENT var res = CUSTODY_REQUIREMENT.uint64
if node.config.subscribeAllSubnets: if node.config.subscribeAllSubnets:
res = DATA_COLUMN_SIDECAR_SUBNET_COUNT res = DATA_COLUMN_SIDECAR_SUBNET_COUNT.uint64
res
proc fetchCustodyColumnCountFromRemotePeer*(node: BeaconNode, pid: PeerId): uint64 = proc fetchCustodyColumnCountFromRemotePeer*(node: BeaconNode, pid: PeerId): uint64 =
# Fetches the custody column count from a remote peer # Fetches the custody column count from a remote peer
@ -1164,7 +1165,8 @@ proc fetchCustodyColumnCountFromRemotePeer*(node: BeaconNode, pid: PeerId): uint
proc addDenebMessageHandlers( proc addDenebMessageHandlers(
node: BeaconNode, forkDigest: ForkDigest, slot: Slot) = node: BeaconNode, forkDigest: ForkDigest, slot: Slot) =
node.addCapellaMessageHandlers(forkDigest, slot) node.addCapellaMessageHandlers(forkDigest, slot)
for topic in dataColumnSidecarTopics(forkDigest): let targetSubnets = node.fetchCustodySubnetCount()
for topic in dataColumnSidecarTopics(forkDigest, targetSubnets):
node.network.subscribe(topic, basicParams) node.network.subscribe(topic, basicParams)
proc addElectraMessageHandlers( proc addElectraMessageHandlers(
@ -1188,7 +1190,8 @@ proc removeCapellaMessageHandlers(node: BeaconNode, forkDigest: ForkDigest) =
proc removeDenebMessageHandlers(node: BeaconNode, forkDigest: ForkDigest) = proc removeDenebMessageHandlers(node: BeaconNode, forkDigest: ForkDigest) =
node.removeCapellaMessageHandlers(forkDigest) node.removeCapellaMessageHandlers(forkDigest)
for topic in dataColumnSidecarTopics(forkDigest): let targetSubnets = node.fetchCustodySubnetCount()
for topic in dataColumnSidecarTopics(forkDigest, targetSubnets):
node.network.unsubscribe(topic) node.network.unsubscribe(topic)
proc removeElectraMessageHandlers(node: BeaconNode, forkDigest: ForkDigest) = proc removeElectraMessageHandlers(node: BeaconNode, forkDigest: ForkDigest) =

4
beacon_chain/spec/network.nim
View File

@ -244,6 +244,6 @@ func getDataColumnSidecarTopic*(forkDigest: ForkDigest,
func compute_subnet_for_data_column_sidecar*(column_index: ColumnIndex): uint64 = func compute_subnet_for_data_column_sidecar*(column_index: ColumnIndex): uint64 =
uint64(column_index mod DATA_COLUMN_SIDECAR_SUBNET_COUNT) uint64(column_index mod DATA_COLUMN_SIDECAR_SUBNET_COUNT)
iterator dataColumnSidecarTopics*(forkDigest: ForkDigest): string = iterator dataColumnSidecarTopics*(forkDigest: ForkDigest, targetSubnetCount: uint64): string =
for subnet_id in 0'u64..<DATA_COLUMN_SIDECAR_SUBNET_COUNT: for subnet_id in 0'u64..<targetSubnetCount:
yield getDataColumnSidecarTopic(forkDigest, subnet_id) yield getDataColumnSidecarTopic(forkDigest, subnet_id)