mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-04-26 18:07:58 +00:00
Pierre Krieger
3d75cc2001
* Companion PR for adding max notification sizes * Add TODO * Missing ; * => * Common limit * "Update Substrate" Co-authored-by: parity-processbot <>
97 lines
3.4 KiB
Rust
97 lines
3.4 KiB
Rust
// Copyright 2021 Parity Technologies (UK) Ltd.
|
|
// This file is part of Polkadot.
|
|
|
|
// Polkadot is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// Polkadot is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with Polkadot. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
//! All peersets and protocols used for parachains.
|
|
|
|
use sc_network::config::{NonDefaultSetConfig, SetConfig};
|
|
use std::borrow::Cow;
|
|
use strum::{EnumIter, IntoEnumIterator};
|
|
|
|
/// The peer-sets and thus the protocols which are used for the network.
|
|
#[derive(Debug, Clone, Copy, PartialEq, EnumIter)]
|
|
pub enum PeerSet {
|
|
/// The validation peer-set is responsible for all messages related to candidate validation and communication among validators.
|
|
Validation,
|
|
/// The collation peer-set is used for validator<>collator communication.
|
|
Collation,
|
|
}
|
|
|
|
impl PeerSet {
|
|
/// Get `sc_network` peer set configurations for each peerset.
|
|
///
|
|
/// Those should be used in the network configuration to register the protocols with the
|
|
/// network service.
|
|
pub fn get_info(self) -> NonDefaultSetConfig {
|
|
let protocol = self.into_protocol_name();
|
|
// TODO: lower this limit after https://github.com/paritytech/polkadot/issues/2283 is
|
|
// done and collations use request-response protocols
|
|
let max_notification_size = 16 * 1024 * 1024;
|
|
|
|
match self {
|
|
PeerSet::Validation => NonDefaultSetConfig {
|
|
notifications_protocol: protocol,
|
|
max_notification_size,
|
|
set_config: sc_network::config::SetConfig {
|
|
in_peers: 25,
|
|
out_peers: 0,
|
|
reserved_nodes: Vec::new(),
|
|
non_reserved_mode: sc_network::config::NonReservedPeerMode::Accept,
|
|
},
|
|
},
|
|
PeerSet::Collation => NonDefaultSetConfig {
|
|
notifications_protocol: protocol,
|
|
max_notification_size,
|
|
set_config: SetConfig {
|
|
in_peers: 25,
|
|
out_peers: 0,
|
|
reserved_nodes: Vec::new(),
|
|
non_reserved_mode: sc_network::config::NonReservedPeerMode::Accept,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
/// Get the protocol name associated with each peer set as static str.
|
|
pub const fn get_protocol_name_static(self) -> &'static str {
|
|
match self {
|
|
PeerSet::Validation => "/polkadot/validation/1",
|
|
PeerSet::Collation => "/polkadot/collation/1",
|
|
}
|
|
}
|
|
|
|
/// Convert a peer set into a protocol name as understood by Substrate.
|
|
pub fn into_protocol_name(self) -> Cow<'static, str> {
|
|
self.get_protocol_name_static().into()
|
|
}
|
|
|
|
/// Try parsing a protocol name into a peer set.
|
|
pub fn try_from_protocol_name(name: &Cow<'static, str>) -> Option<PeerSet> {
|
|
match name {
|
|
n if n == &PeerSet::Validation.into_protocol_name() => Some(PeerSet::Validation),
|
|
n if n == &PeerSet::Collation.into_protocol_name() => Some(PeerSet::Collation),
|
|
_ => None,
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Get `NonDefaultSetConfig`s for all available peer sets.
|
|
///
|
|
/// Should be used during network configuration (added to [`NetworkConfiguration::extra_sets`])
|
|
/// or shortly after startup to register the protocols with the network service.
|
|
pub fn peer_sets_info() -> Vec<sc_network::config::NonDefaultSetConfig> {
|
|
PeerSet::iter().map(PeerSet::get_info).collect()
|
|
}
|