feat: create new methods with Config arguments for existing builders (#5837)

Co-authored-by: Dan Cline <6798349+Rjected@users.noreply.github.com>
This commit is contained in:
Thomas Coratger
2023-12-20 22:55:13 +01:00
committed by GitHub
parent ad3b893c23
commit c066aa60af
10 changed files with 39 additions and 37 deletions

View File

@ -12,7 +12,6 @@ repository.workspace = true
reth-network.workspace = true
reth-net-nat.workspace = true
reth-discv4.workspace = true
reth-downloaders.workspace = true
reth-primitives.workspace = true
# io

View File

@ -1,9 +1,5 @@
//! Configuration files.
use reth_discv4::Discv4Config;
use reth_downloaders::{
bodies::bodies::BodiesDownloaderBuilder,
headers::reverse_headers::ReverseHeadersDownloaderBuilder,
};
use reth_network::{NetworkConfigBuilder, PeersConfig, SessionsConfig};
use reth_primitives::PruneModes;
use secp256k1::SecretKey;
@ -110,17 +106,6 @@ impl Default for HeadersConfig {
}
}
impl From<HeadersConfig> for ReverseHeadersDownloaderBuilder {
fn from(config: HeadersConfig) -> Self {
ReverseHeadersDownloaderBuilder::default()
.request_limit(config.downloader_request_limit)
.min_concurrent_requests(config.downloader_min_concurrent_requests)
.max_concurrent_requests(config.downloader_max_concurrent_requests)
.max_buffered_responses(config.downloader_max_buffered_responses)
.stream_batch_size(config.commit_threshold as usize)
}
}
/// Total difficulty stage configuration
#[derive(Debug, Clone, Copy, Deserialize, PartialEq, Serialize)]
#[serde(default)]
@ -175,19 +160,6 @@ impl Default for BodiesConfig {
}
}
impl From<BodiesConfig> for BodiesDownloaderBuilder {
fn from(config: BodiesConfig) -> Self {
BodiesDownloaderBuilder::default()
.with_stream_batch_size(config.downloader_stream_batch_size)
.with_request_limit(config.downloader_request_limit)
.with_max_buffered_blocks_size_bytes(config.downloader_max_buffered_blocks_size_bytes)
.with_concurrent_requests_range(
config.downloader_min_concurrent_requests..=
config.downloader_max_concurrent_requests,
)
}
}
/// Sender recovery stage configuration.
#[derive(Debug, Clone, Copy, Deserialize, PartialEq, Eq, Serialize)]
#[serde(default)]

View File

@ -10,4 +10,4 @@
#![cfg_attr(docsrs, feature(doc_cfg, doc_auto_cfg))]
pub mod config;
pub use config::{Config, PruneConfig};
pub use config::{BodiesConfig, Config, PruneConfig};