chore(engine): enable engine debug streams in new implementation (#10282)

This commit is contained in:
Roman Krasiuk
2024-08-12 22:06:39 -07:00
committed by GitHub
parent dfcd00fc6a
commit 8a802dab60
4 changed files with 32 additions and 12 deletions

View File

@ -30,7 +30,6 @@ reth-tasks.workspace = true
# async
futures.workspace = true
pin-project.workspace = true
tokio-stream.workspace = true
# misc
thiserror.workspace = true
@ -46,3 +45,4 @@ reth-primitives.workspace = true
reth-prune-types.workspace = true
tokio = { workspace = true, features = ["sync"] }
tokio-stream.workspace = true

View File

@ -30,13 +30,15 @@ use std::{
sync::Arc,
task::{Context, Poll},
};
use tokio_stream::wrappers::UnboundedReceiverStream;
/// Alias for consensus engine stream.
type EngineMessageStream<T> = Pin<Box<dyn Stream<Item = BeaconEngineMessage<T>> + Send + Sync>>;
/// Alias for chain orchestrator.
type EngineServiceType<DB, Client, T> = ChainOrchestrator<
EngineHandler<
EngineApiRequestHandler<EngineApiRequest<T>>,
UnboundedReceiverStream<BeaconEngineMessage<T>>,
EngineMessageStream<T>,
BasicBlockDownloader<Client>,
>,
PipelineSync<DB>,
@ -70,7 +72,7 @@ where
executor_factory: E,
chain_spec: Arc<ChainSpec>,
client: Client,
incoming_requests: UnboundedReceiverStream<BeaconEngineMessage<T>>,
incoming_requests: EngineMessageStream<T>,
pipeline: Pipeline<DB>,
pipeline_task_spawner: Box<dyn TaskSpawner>,
provider: ProviderFactory<DB>,
@ -149,6 +151,7 @@ mod tests {
use reth_tasks::TokioTaskExecutor;
use std::sync::Arc;
use tokio::sync::{mpsc::unbounded_channel, watch};
use tokio_stream::wrappers::UnboundedReceiverStream;
#[test]
fn eth_chain_orchestrator_build() {
@ -185,7 +188,7 @@ mod tests {
executor_factory,
chain_spec,
client,
incoming_requests,
Box::pin(incoming_requests),
pipeline,
pipeline_task_spawner,
provider_factory,