github/reth
1
1
Fork 0
mirror of https://github.com/paradigmxyz/reth.git synced 2026-04-30 03:01:58 -04:00
Code Issues Packages Projects Releases Wiki Activity

refactor: move write_peers_to_file to NetworkManager impl (#9134)

Browse Source 
Co-authored-by: Matthias Seitz <matthias.seitz@outlook.de>
This commit is contained in:
Querty
2024-07-03 16:05:44 +02:00
committed by GitHub
parent d41aac380b
commit c5167a4784
6 changed files with 39 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
crates/node/builder/Cargo.toml
View File

@@ -72,5 +72,8 @@ confy.workspace = true
rayon.workspace = true
backon.workspace = true
# tracing
tracing.workspace = true
[dev-dependencies]
tempfile.workspace = true

15
crates/node/builder/src/builder/mod.rs
View File

@@ -30,7 +30,6 @@ use reth_node_core::{
dirs::{ChainPath, DataDirPath, MaybePlatformPath},
node_config::NodeConfig,
primitives::Head,
utils::write_peers_to_file,
};
use reth_primitives::revm_primitives::EnvKzgSettings;
use reth_provider::{providers::BlockchainProvider, ChainSpecProvider};
@@ -39,6 +38,7 @@ use reth_transaction_pool::{PoolConfig, TransactionPool};
use secp256k1::SecretKey;
pub use states::*;
use std::sync::Arc;
use tracing::{info, trace, warn};
mod states;
@@ -509,7 +509,18 @@ impl<Node: FullNodeTypes> BuilderContext<Node> {
"p2p network task",
|shutdown| {
network.run_until_graceful_shutdown(shutdown, |network| {
write_peers_to_file(&network, known_peers_file)
if let Some(peers_file) = known_peers_file {
let num_known_peers = network.num_known_peers();
trace!(target: "reth::cli", peers_file=?peers_file, num_peers=%num_known_peers, "Saving current peers");
match network.write_peers_to_file(peers_file.as_path()) {
Ok(_) => {
info!(target: "reth::cli", peers_file=?peers_file, "Wrote network peers to file");
}
Err(err) => {
warn!(target: "reth::cli", %err, "Failed to write network peers to file");
}
}
}
})
},
);