jleony/nanoreth
1
0
Fork 0
mirror of https://github.com/hl-archive-node/nanoreth.git synced 2025-12-06 10:59:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

1
crates/net/downloaders/Cargo.toml
View File

@ -14,6 +14,7 @@ reth-interfaces.workspace = true
reth-primitives.workspace = true
reth-tasks.workspace = true
reth-provider.workspace = true
reth-config.workspace = true
# async
futures.workspace = true

16
crates/net/downloaders/src/bodies/bodies.rs
View File

@ -2,6 +2,7 @@ use super::queue::BodiesRequestQueue;
use crate::{bodies::task::TaskDownloader, metrics::BodyDownloaderMetrics};
use futures::Stream;
use futures_util::StreamExt;
use reth_config::BodiesConfig;
use reth_interfaces::{
consensus::Consensus,
p2p::{
@ -492,6 +493,21 @@ pub struct BodiesDownloaderBuilder {
pub concurrent_requests_range: RangeInclusive<usize>,
}
impl BodiesDownloaderBuilder {
/// Creates a new [BodiesDownloaderBuilder] with configurations based on the provided
/// [BodiesConfig].
pub fn new(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,
)
}
}
impl Default for BodiesDownloaderBuilder {
fn default() -> Self {
Self {

14
crates/net/downloaders/src/headers/reverse_headers.rs
View File

@ -5,6 +5,7 @@ use crate::metrics::HeaderDownloaderMetrics;
use futures::{stream::Stream, FutureExt};
use futures_util::{stream::FuturesUnordered, StreamExt};
use rayon::prelude::*;
use reth_config::config::HeadersConfig;
use reth_interfaces::{
consensus::Consensus,
p2p::{
@ -1087,6 +1088,19 @@ pub struct ReverseHeadersDownloaderBuilder {
max_buffered_responses: usize,
}
impl ReverseHeadersDownloaderBuilder {
/// Creates a new [ReverseHeadersDownloaderBuilder] with configurations based on the provided
/// [HeadersConfig].
pub fn new(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)
}
}
impl Default for ReverseHeadersDownloaderBuilder {
fn default() -> Self {
Self {