fix(metrics): set sync metrics when blocks are added (#10799)

This commit is contained in:
Dan Cline
2024-09-10 07:43:03 -04:00
committed by GitHub
parent ed4235609f
commit 72c53b88bc
3 changed files with 27 additions and 5 deletions

View File

@ -20,7 +20,7 @@ use reth_payload_builder::PayloadBuilderHandle;
use reth_payload_validator::ExecutionPayloadValidator;
use reth_provider::{providers::BlockchainProvider2, ProviderFactory};
use reth_prune::Pruner;
use reth_stages_api::Pipeline;
use reth_stages_api::{MetricEventsSender, Pipeline};
use reth_tasks::TaskSpawner;
use std::{
marker::PhantomData,
@ -77,10 +77,12 @@ where
payload_builder: PayloadBuilderHandle<N::Engine>,
tree_config: TreeConfig,
invalid_block_hook: Box<dyn InvalidBlockHook>,
sync_metrics_tx: MetricEventsSender,
) -> Self {
let downloader = BasicBlockDownloader::new(client, consensus.clone());
let persistence_handle = PersistenceHandle::spawn_service(provider, pruner);
let persistence_handle =
PersistenceHandle::spawn_service(provider, pruner, sync_metrics_tx);
let payload_validator = ExecutionPayloadValidator::new(chain_spec);
let canonical_in_memory_state = blockchain_db.canonical_in_memory_state();
@ -179,6 +181,7 @@ mod tests {
let pruner =
Pruner::<_, ProviderFactory<_>>::new(provider_factory.clone(), vec![], 0, 0, None, rx);
let (sync_metrics_tx, _sync_metrics_rx) = unbounded_channel();
let (tx, _rx) = unbounded_channel();
let _eth_service = EngineService::new(
consensus,
@ -194,6 +197,7 @@ mod tests {
PayloadBuilderHandle::new(tx),
TreeConfig::default(),
Box::new(NoopInvalidBlockHook::default()),
sync_metrics_tx,
);
}
}