logos-blockchain-testing/testing-framework/deployers/k8s/src/lifecycle/block_feed.rs

use testing_framework_core::scenario::{BlockFeed, BlockFeedTask, NodeClients, spawn_block_feed};
use tracing::{debug, info};

use crate::deployer::K8sRunnerError;

pub async fn spawn_block_feed_with(
    node_clients: &NodeClients,
) -> Result<(BlockFeed, BlockFeedTask), K8sRunnerError> {
    debug!(
        nodes = node_clients.node_clients().len(),
        "selecting node client for block feed"
    );

    let block_source_client = node_clients
        .node_clients()
        .into_iter()
        .next()
        .or_else(|| node_clients.any_client())
        .ok_or(K8sRunnerError::BlockFeedMissing)?;

    info!("starting block feed");
    spawn_block_feed(block_source_client)
        .await
        .map_err(|source| K8sRunnerError::BlockFeed { source })
}