2024-01-17 06:17:49 -08:00
|
|
|
use crate::grpc_subscription::{
|
2024-01-25 01:29:14 -08:00
|
|
|
create_block_processing_task, create_slot_stream_task, from_grpc_block_update,
|
2024-01-17 06:17:49 -08:00
|
|
|
};
|
2024-01-15 11:22:59 -08:00
|
|
|
use anyhow::Context;
|
2024-01-21 04:56:40 -08:00
|
|
|
use futures::{Stream, StreamExt};
|
2023-12-22 05:42:20 -08:00
|
|
|
use geyser_grpc_connector::grpc_subscription_autoreconnect::{
|
2024-01-08 05:14:59 -08:00
|
|
|
create_geyser_reconnecting_stream, GeyserFilter, GrpcSourceConfig,
|
2023-12-22 05:42:20 -08:00
|
|
|
};
|
|
|
|
use geyser_grpc_connector::grpcmultiplex_fastestwins::{
|
|
|
|
create_multiplexed_stream, FromYellowstoneExtractor,
|
|
|
|
};
|
2024-01-08 05:21:05 -08:00
|
|
|
use log::{debug, info, trace, warn};
|
2023-12-22 05:42:20 -08:00
|
|
|
use solana_lite_rpc_core::structures::produced_block::ProducedBlock;
|
|
|
|
use solana_lite_rpc_core::structures::slot_notification::SlotNotification;
|
|
|
|
use solana_lite_rpc_core::AnyhowJoinHandle;
|
|
|
|
use solana_sdk::clock::Slot;
|
|
|
|
use solana_sdk::commitment_config::CommitmentConfig;
|
2024-01-21 04:56:40 -08:00
|
|
|
use std::collections::{BTreeSet, HashMap, HashSet};
|
2023-12-22 05:42:20 -08:00
|
|
|
use std::time::Duration;
|
|
|
|
use tokio::sync::broadcast::Receiver;
|
2024-01-21 04:56:40 -08:00
|
|
|
use tokio::sync::mpsc::UnboundedSender;
|
2023-12-22 05:42:20 -08:00
|
|
|
use yellowstone_grpc_proto::geyser::subscribe_update::UpdateOneof;
|
2024-01-17 06:17:49 -08:00
|
|
|
use yellowstone_grpc_proto::geyser::SubscribeUpdate;
|
2023-12-22 05:42:20 -08:00
|
|
|
|
|
|
|
struct BlockExtractor(CommitmentConfig);
|
|
|
|
|
|
|
|
impl FromYellowstoneExtractor for BlockExtractor {
|
|
|
|
type Target = ProducedBlock;
|
|
|
|
fn map_yellowstone_update(&self, update: SubscribeUpdate) -> Option<(Slot, Self::Target)> {
|
|
|
|
match update.update_oneof {
|
|
|
|
Some(UpdateOneof::Block(update_block_message)) => {
|
2024-01-25 01:29:14 -08:00
|
|
|
let block = from_grpc_block_update(update_block_message, self.0);
|
2023-12-22 05:42:20 -08:00
|
|
|
Some((block.slot, block))
|
|
|
|
}
|
|
|
|
_ => None,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-01-08 05:14:59 -08:00
|
|
|
struct BlockMetaHashExtractor(CommitmentConfig);
|
2024-01-08 03:11:48 -08:00
|
|
|
|
2024-01-08 05:14:59 -08:00
|
|
|
impl FromYellowstoneExtractor for BlockMetaHashExtractor {
|
2024-01-08 03:11:48 -08:00
|
|
|
type Target = String;
|
|
|
|
fn map_yellowstone_update(&self, update: SubscribeUpdate) -> Option<(u64, String)> {
|
|
|
|
match update.update_oneof {
|
|
|
|
Some(UpdateOneof::BlockMeta(block_meta)) => {
|
|
|
|
Some((block_meta.slot, block_meta.blockhash))
|
|
|
|
}
|
|
|
|
_ => None,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-01-22 23:17:48 -08:00
|
|
|
fn create_grpc_multiplex_processed_block_stream(
|
2024-01-21 04:56:40 -08:00
|
|
|
grpc_sources: &Vec<GrpcSourceConfig>,
|
2024-01-25 01:29:14 -08:00
|
|
|
processed_block_sender: UnboundedSender<ProducedBlock>,
|
2024-01-21 04:56:40 -08:00
|
|
|
) -> Vec<AnyhowJoinHandle> {
|
2024-01-22 01:54:39 -08:00
|
|
|
let commitment_config = CommitmentConfig::processed();
|
2024-01-21 04:56:40 -08:00
|
|
|
|
|
|
|
let mut tasks = Vec::new();
|
|
|
|
let mut streams = vec![];
|
|
|
|
for grpc_source in grpc_sources {
|
|
|
|
let (block_sender, block_reciever) = async_channel::unbounded();
|
|
|
|
tasks.push(create_block_processing_task(
|
|
|
|
grpc_source.grpc_addr.clone(),
|
|
|
|
grpc_source.grpc_x_token.clone(),
|
|
|
|
block_sender,
|
2024-01-22 01:54:39 -08:00
|
|
|
yellowstone_grpc_proto::geyser::CommitmentLevel::Processed,
|
2024-01-21 04:56:40 -08:00
|
|
|
));
|
|
|
|
streams.push(block_reciever)
|
|
|
|
}
|
|
|
|
let merging_streams: AnyhowJoinHandle = tokio::task::spawn(async move {
|
|
|
|
let mut slots_processed = BTreeSet::<u64>::new();
|
|
|
|
loop {
|
|
|
|
let block_message = futures::stream::select_all(streams.clone()).next().await;
|
|
|
|
const MAX_SIZE: usize = 1024;
|
|
|
|
if let Some(block) = block_message {
|
|
|
|
let slot = block.slot;
|
|
|
|
// check if the slot is in the map, if not check if the container is half full and the slot in question is older than the lowest value
|
|
|
|
// it means that the slot is too old to process
|
|
|
|
if !slots_processed.contains(&slot)
|
|
|
|
&& (slots_processed.len() < MAX_SIZE / 2
|
|
|
|
|| slot > slots_processed.first().cloned().unwrap_or_default())
|
|
|
|
{
|
2024-01-25 01:29:14 -08:00
|
|
|
processed_block_sender
|
|
|
|
.send(from_grpc_block_update(block, commitment_config))
|
2024-01-21 04:56:40 -08:00
|
|
|
.context("Issue to send confirmed block")?;
|
|
|
|
slots_processed.insert(slot);
|
|
|
|
if slots_processed.len() > MAX_SIZE {
|
|
|
|
slots_processed.pop_first();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
});
|
|
|
|
tasks.push(merging_streams);
|
|
|
|
tasks
|
|
|
|
}
|
|
|
|
|
|
|
|
fn create_grpc_multiplex_block_meta_stream(
|
|
|
|
grpc_sources: &Vec<GrpcSourceConfig>,
|
|
|
|
commitment_config: CommitmentConfig,
|
|
|
|
) -> impl Stream<Item = String> {
|
|
|
|
let mut streams = Vec::new();
|
|
|
|
for grpc_source in grpc_sources {
|
|
|
|
let stream = create_geyser_reconnecting_stream(
|
|
|
|
grpc_source.clone(),
|
|
|
|
GeyserFilter(commitment_config).blocks_meta(),
|
|
|
|
);
|
|
|
|
streams.push(stream);
|
|
|
|
}
|
|
|
|
create_multiplexed_stream(streams, BlockMetaHashExtractor(commitment_config))
|
|
|
|
}
|
|
|
|
|
2024-01-22 23:17:48 -08:00
|
|
|
/// connect to multiple grpc sources to consume processed (full) blocks and block meta for commitment level confirmed and finalized
|
|
|
|
/// will emit blocks for commitment level processed, confirmed and finalized OR only processed block never gets confirmed
|
2023-12-22 05:42:20 -08:00
|
|
|
pub fn create_grpc_multiplex_blocks_subscription(
|
|
|
|
grpc_sources: Vec<GrpcSourceConfig>,
|
|
|
|
) -> (Receiver<ProducedBlock>, AnyhowJoinHandle) {
|
|
|
|
info!("Setup grpc multiplexed blocks connection...");
|
|
|
|
if grpc_sources.is_empty() {
|
|
|
|
info!("- no grpc connection configured");
|
|
|
|
}
|
|
|
|
for grpc_source in &grpc_sources {
|
|
|
|
info!("- connection to {}", grpc_source);
|
|
|
|
}
|
|
|
|
|
2024-01-08 05:14:59 -08:00
|
|
|
// return value is the broadcast receiver
|
|
|
|
let (producedblock_sender, blocks_output_stream) =
|
|
|
|
tokio::sync::broadcast::channel::<ProducedBlock>(1000);
|
|
|
|
|
|
|
|
let jh_block_emitter_task = {
|
2024-01-08 03:11:48 -08:00
|
|
|
tokio::task::spawn(async move {
|
|
|
|
loop {
|
2024-01-21 04:56:40 -08:00
|
|
|
let (processed_block_sender, mut processed_block_reciever) =
|
2024-01-15 11:22:59 -08:00
|
|
|
tokio::sync::mpsc::unbounded_channel::<ProducedBlock>();
|
2024-01-21 04:56:40 -08:00
|
|
|
|
2024-01-22 23:17:48 -08:00
|
|
|
let processed_blocks_tasks = create_grpc_multiplex_processed_block_stream(
|
|
|
|
&grpc_sources,
|
|
|
|
processed_block_sender,
|
|
|
|
);
|
2024-01-21 04:56:40 -08:00
|
|
|
|
|
|
|
let confirmed_blockmeta_stream = create_grpc_multiplex_block_meta_stream(
|
|
|
|
&grpc_sources,
|
|
|
|
CommitmentConfig::confirmed(),
|
|
|
|
);
|
|
|
|
let finalized_blockmeta_stream = create_grpc_multiplex_block_meta_stream(
|
|
|
|
&grpc_sources,
|
|
|
|
CommitmentConfig::finalized(),
|
|
|
|
);
|
2024-01-12 05:21:54 -08:00
|
|
|
|
|
|
|
// by blockhash
|
2024-01-21 04:56:40 -08:00
|
|
|
let mut recent_processed_blocks = HashMap::<String, ProducedBlock>::new();
|
|
|
|
let mut confirmed_blockmeta_stream = std::pin::pin!(confirmed_blockmeta_stream);
|
2024-01-12 05:21:54 -08:00
|
|
|
let mut finalized_blockmeta_stream = std::pin::pin!(finalized_blockmeta_stream);
|
|
|
|
|
|
|
|
let mut last_finalized_slot: Slot = 0;
|
2024-01-22 23:17:48 -08:00
|
|
|
let mut cleanup_tick = tokio::time::interval(Duration::from_secs(5));
|
|
|
|
const MAX_ALLOWED_CLEANUP_WITHOUT_RECV: u32 = 12; // 12*5 = 60s without recving data
|
|
|
|
const CLEANUP_SLOTS_BEHIND_FINALIZED: u64 = 100;
|
|
|
|
let mut cleanup_without_recv_blocks: u32 = 0;
|
|
|
|
let mut cleanup_without_confirmed_recv_blocks_meta: u32 = 0;
|
|
|
|
let mut cleanup_without_finalized_recv_blocks_meta: u32 = 0;
|
2024-01-21 04:56:40 -08:00
|
|
|
let mut confirmed_block_not_yet_processed = HashSet::<String>::new();
|
|
|
|
|
|
|
|
// start logging errors when we recieve first finalized block
|
2024-01-22 23:17:48 -08:00
|
|
|
let mut startup_completed = false;
|
2024-01-12 05:21:54 -08:00
|
|
|
loop {
|
|
|
|
tokio::select! {
|
2024-01-21 04:56:40 -08:00
|
|
|
processed_block = processed_block_reciever.recv() => {
|
2024-01-12 05:21:54 -08:00
|
|
|
cleanup_without_recv_blocks = 0;
|
|
|
|
|
2024-01-21 04:56:40 -08:00
|
|
|
let processed_block = processed_block.expect("processed block from stream");
|
|
|
|
trace!("got processed block {} with blockhash {}",
|
|
|
|
processed_block.slot, processed_block.blockhash.clone());
|
|
|
|
if let Err(e) = producedblock_sender.send(processed_block.clone()) {
|
|
|
|
warn!("produced block channel has no receivers {e:?}");
|
2024-01-12 05:21:54 -08:00
|
|
|
}
|
2024-01-21 04:56:40 -08:00
|
|
|
if confirmed_block_not_yet_processed.remove(&processed_block.blockhash) {
|
|
|
|
if let Err(e) = producedblock_sender.send(processed_block.to_confirmed_block()) {
|
|
|
|
warn!("produced block channel has no receivers {e:?}");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
recent_processed_blocks.insert(processed_block.blockhash.clone(), processed_block);
|
|
|
|
},
|
|
|
|
meta_confirmed = confirmed_blockmeta_stream.next() => {
|
|
|
|
cleanup_without_confirmed_recv_blocks_meta = 0;
|
|
|
|
let blockhash = meta_confirmed.expect("confirmed block meta from stream");
|
|
|
|
if let Some(cached_processed_block) = recent_processed_blocks.get(&blockhash) {
|
|
|
|
let confirmed_block = cached_processed_block.to_confirmed_block();
|
|
|
|
debug!("got confirmed blockmeta {} with blockhash {}",
|
|
|
|
confirmed_block.slot, confirmed_block.blockhash.clone());
|
|
|
|
if let Err(e) = producedblock_sender.send(confirmed_block) {
|
2024-01-22 23:17:48 -08:00
|
|
|
warn!("Confirmed block channel has no receivers {e:?}");
|
2024-01-21 04:56:40 -08:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
confirmed_block_not_yet_processed.insert(blockhash.clone());
|
2024-01-22 01:54:39 -08:00
|
|
|
log::debug!("confirmed blocks not found : {}", confirmed_block_not_yet_processed.len());
|
2024-01-21 04:56:40 -08:00
|
|
|
}
|
2024-01-12 05:21:54 -08:00
|
|
|
},
|
|
|
|
meta_finalized = finalized_blockmeta_stream.next() => {
|
2024-01-21 04:56:40 -08:00
|
|
|
cleanup_without_finalized_recv_blocks_meta = 0;
|
2024-01-12 05:21:54 -08:00
|
|
|
let blockhash = meta_finalized.expect("finalized block meta from stream");
|
2024-01-21 04:56:40 -08:00
|
|
|
if let Some(cached_processed_block) = recent_processed_blocks.remove(&blockhash) {
|
|
|
|
let finalized_block = cached_processed_block.to_finalized_block();
|
2024-01-12 05:21:54 -08:00
|
|
|
last_finalized_slot = finalized_block.slot;
|
2024-01-22 23:17:48 -08:00
|
|
|
startup_completed = true;
|
2024-01-12 05:21:54 -08:00
|
|
|
debug!("got finalized blockmeta {} with blockhash {}",
|
|
|
|
finalized_block.slot, finalized_block.blockhash.clone());
|
|
|
|
if let Err(e) = producedblock_sender.send(finalized_block) {
|
|
|
|
warn!("Finalized block channel has no receivers {e:?}");
|
|
|
|
}
|
2024-01-22 23:17:48 -08:00
|
|
|
} else if startup_completed {
|
2024-01-15 11:22:59 -08:00
|
|
|
// this warning is ok for first few blocks when we start lrpc
|
Production (#313)
* fix: panic on geyser close, multiplex bug
https://github.com/blockworks-foundation/geyser-grpc-connector/issues/3
* update Cargo.lock
* reverting cargo.lock
* Fix issues with grpc and postgres
* Solving merge issues
* Fixing cargo fmt
* Increase finish quic timeout (#280) (#281)
* integrate geyser slot subscription (#283)
* Increase finish quic timeout (#280)
* Moving geyser slot subscription from stream to channels (#282)
* Moving geyser slot subscription from stream to channels
* Closing all the slot subscription tasks incase of restart
* Making slot channel unbounded (bug)
* remove block_debug_listen
caused a panic - need more time to investigate
2024-01-17T20:31:42.913 app[683d392fd45368] ams [info] thread 'tokio-runtime-worker' panicked at cluster-endpoints/src/grpc_inspect.rs:59:21:
2024-01-17T20:31:42.913 app[683d392fd45368] ams [info] Error receiving block: Closed
2024-01-17T20:31:42.922 app[683d392fd45368] ams [info] 2024-01-17T20:31:42.912597Z ERROR lite_rpc: Services quit unexpectedly Err(cluster endpoint failure (Err(JoinError::Panic(Id(20), ...)), 1, [JoinHandle { id: Id(19) }, JoinHandle { id: Id(23) }])
* Update cargolock file
* Fixing clippy removing grpc_inspect
* merging main with production (#290)
* remove block_debug_listen (#286)
* remove block_debug_listen
caused a panic - need more time to investigate
2024-01-17T20:31:42.913 app[683d392fd45368] ams [info] thread 'tokio-runtime-worker' panicked at cluster-endpoints/src/grpc_inspect.rs:59:21:
2024-01-17T20:31:42.913 app[683d392fd45368] ams [info] Error receiving block: Closed
2024-01-17T20:31:42.922 app[683d392fd45368] ams [info] 2024-01-17T20:31:42.912597Z ERROR lite_rpc: Services quit unexpectedly Err(cluster endpoint failure (Err(JoinError::Panic(Id(20), ...)), 1, [JoinHandle { id: Id(19) }, JoinHandle { id: Id(23) }])
* clippy
* Fixing message too long and overflow panics (#288)
* Update geyser grpc connector commit (#289)
---------
Co-authored-by: Groovie | Mango <95291500+grooviegermanikus@users.noreply.github.com>
* Merging MTU changes and setting up transportation config (#293)
* remove block_debug_listen (#286)
* remove block_debug_listen
caused a panic - need more time to investigate
2024-01-17T20:31:42.913 app[683d392fd45368] ams [info] thread 'tokio-runtime-worker' panicked at cluster-endpoints/src/grpc_inspect.rs:59:21:
2024-01-17T20:31:42.913 app[683d392fd45368] ams [info] Error receiving block: Closed
2024-01-17T20:31:42.922 app[683d392fd45368] ams [info] 2024-01-17T20:31:42.912597Z ERROR lite_rpc: Services quit unexpectedly Err(cluster endpoint failure (Err(JoinError::Panic(Id(20), ...)), 1, [JoinHandle { id: Id(19) }, JoinHandle { id: Id(23) }])
* clippy
* Fixing message too long and overflow panics (#288)
* Update geyser grpc connector commit (#289)
* Updating the transport config to match with solana endpoint (#292)
* Updating the transport config to match with solana endpoint
* Setting max MTU after groovies comments
---------
Co-authored-by: Groovie | Mango <95291500+grooviegermanikus@users.noreply.github.com>
* Making block subscription processed and moving confirmed block subscr… (#291)
* Making block subscription processed and moving confirmed block subscription to meta
* Sending both processed and confirmed blocks, if block has already been confirmed
* Minor bug, subscribing to processed blocks instead of confirmed (#295)
* reduce log
* Restart connection if any stream fails (#306)
* Main into production 01/02/2024 (#312)
* Production into main (#303)
* fix: panic on geyser close, multiplex bug
https://github.com/blockworks-foundation/geyser-grpc-connector/issues/3
* update Cargo.lock
* reverting cargo.lock
* Fix issues with grpc and postgres
* Solving merge issues
* Fixing cargo fmt
* Increase finish quic timeout (#280) (#281)
* integrate geyser slot subscription (#283)
* Increase finish quic timeout (#280)
* Moving geyser slot subscription from stream to channels (#282)
* Moving geyser slot subscription from stream to channels
* Closing all the slot subscription tasks incase of restart
* Making slot channel unbounded (bug)
* remove block_debug_listen
caused a panic - need more time to investigate
2024-01-17T20:31:42.913 app[683d392fd45368] ams [info] thread 'tokio-runtime-worker' panicked at cluster-endpoints/src/grpc_inspect.rs:59:21:
2024-01-17T20:31:42.913 app[683d392fd45368] ams [info] Error receiving block: Closed
2024-01-17T20:31:42.922 app[683d392fd45368] ams [info] 2024-01-17T20:31:42.912597Z ERROR lite_rpc: Services quit unexpectedly Err(cluster endpoint failure (Err(JoinError::Panic(Id(20), ...)), 1, [JoinHandle { id: Id(19) }, JoinHandle { id: Id(23) }])
* Update cargolock file
* Fixing clippy removing grpc_inspect
* merging main with production (#290)
* remove block_debug_listen (#286)
* remove block_debug_listen
caused a panic - need more time to investigate
2024-01-17T20:31:42.913 app[683d392fd45368] ams [info] thread 'tokio-runtime-worker' panicked at cluster-endpoints/src/grpc_inspect.rs:59:21:
2024-01-17T20:31:42.913 app[683d392fd45368] ams [info] Error receiving block: Closed
2024-01-17T20:31:42.922 app[683d392fd45368] ams [info] 2024-01-17T20:31:42.912597Z ERROR lite_rpc: Services quit unexpectedly Err(cluster endpoint failure (Err(JoinError::Panic(Id(20), ...)), 1, [JoinHandle { id: Id(19) }, JoinHandle { id: Id(23) }])
* clippy
* Fixing message too long and overflow panics (#288)
* Update geyser grpc connector commit (#289)
---------
Co-authored-by: Groovie | Mango <95291500+grooviegermanikus@users.noreply.github.com>
* Merging MTU changes and setting up transportation config (#293)
* remove block_debug_listen (#286)
* remove block_debug_listen
caused a panic - need more time to investigate
2024-01-17T20:31:42.913 app[683d392fd45368] ams [info] thread 'tokio-runtime-worker' panicked at cluster-endpoints/src/grpc_inspect.rs:59:21:
2024-01-17T20:31:42.913 app[683d392fd45368] ams [info] Error receiving block: Closed
2024-01-17T20:31:42.922 app[683d392fd45368] ams [info] 2024-01-17T20:31:42.912597Z ERROR lite_rpc: Services quit unexpectedly Err(cluster endpoint failure (Err(JoinError::Panic(Id(20), ...)), 1, [JoinHandle { id: Id(19) }, JoinHandle { id: Id(23) }])
* clippy
* Fixing message too long and overflow panics (#288)
* Update geyser grpc connector commit (#289)
* Updating the transport config to match with solana endpoint (#292)
* Updating the transport config to match with solana endpoint
* Setting max MTU after groovies comments
---------
Co-authored-by: Groovie | Mango <95291500+grooviegermanikus@users.noreply.github.com>
* Making block subscription processed and moving confirmed block subscr… (#291)
* Making block subscription processed and moving confirmed block subscription to meta
* Sending both processed and confirmed blocks, if block has already been confirmed
* Minor bug, subscribing to processed blocks instead of confirmed (#295)
---------
Co-authored-by: Groovie | Mango <95291500+grooviegermanikus@users.noreply.github.com>
Co-authored-by: GroovieGermanikus <groovie@mango.markets>
* Restart connection if any stream fails
* Updating to version 0.2.4
* Updating the change logs
* Fixing all the sub overflows by using saturating sub (#309)
* Fixing by cu computation adding more tests to check by CU (#311)
* Fixing by cu computation adding more tests to check by CU
* Adding more tests
---------
Co-authored-by: Groovie | Mango <95291500+grooviegermanikus@users.noreply.github.com>
Co-authored-by: GroovieGermanikus <groovie@mango.markets>
---------
Co-authored-by: Groovie | Mango <95291500+grooviegermanikus@users.noreply.github.com>
Co-authored-by: GroovieGermanikus <groovie@mango.markets>
2024-02-01 06:16:24 -08:00
|
|
|
log::warn!("finalized block meta received for blockhash {} which was never seen or already emitted", blockhash);
|
2024-01-12 05:21:54 -08:00
|
|
|
}
|
|
|
|
},
|
|
|
|
_ = cleanup_tick.tick() => {
|
2024-01-22 23:17:48 -08:00
|
|
|
// timebased restart
|
|
|
|
if
|
|
|
|
cleanup_without_finalized_recv_blocks_meta > MAX_ALLOWED_CLEANUP_WITHOUT_RECV
|
2024-01-26 13:23:59 -08:00
|
|
|
|| cleanup_without_recv_blocks > MAX_ALLOWED_CLEANUP_WITHOUT_RECV
|
|
|
|
|| cleanup_without_confirmed_recv_blocks_meta > MAX_ALLOWED_CLEANUP_WITHOUT_RECV {
|
2024-01-12 05:21:54 -08:00
|
|
|
log::error!("block or block meta stream stopped restaring blocks");
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
cleanup_without_recv_blocks += 1;
|
2024-01-21 04:56:40 -08:00
|
|
|
cleanup_without_finalized_recv_blocks_meta += 1;
|
|
|
|
cleanup_without_confirmed_recv_blocks_meta += 1;
|
|
|
|
let size_before = recent_processed_blocks.len();
|
|
|
|
recent_processed_blocks.retain(|_blockhash, block| {
|
2024-01-30 06:24:35 -08:00
|
|
|
last_finalized_slot == 0 || block.slot > last_finalized_slot.saturating_sub(CLEANUP_SLOTS_BEHIND_FINALIZED)
|
2024-01-12 05:21:54 -08:00
|
|
|
});
|
2024-01-30 06:24:35 -08:00
|
|
|
let cnt_cleaned = size_before.saturating_sub(recent_processed_blocks.len());
|
2024-01-12 05:21:54 -08:00
|
|
|
if cnt_cleaned > 0 {
|
2024-01-22 23:17:48 -08:00
|
|
|
debug!("cleaned {} processed blocks from cache", cnt_cleaned);
|
2024-01-08 03:11:48 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2024-01-17 06:17:49 -08:00
|
|
|
// abort all the tasks
|
2024-01-22 23:17:48 -08:00
|
|
|
processed_blocks_tasks.iter().for_each(|task| task.abort());
|
2024-01-08 03:11:48 -08:00
|
|
|
}
|
|
|
|
})
|
2023-12-22 05:42:20 -08:00
|
|
|
};
|
|
|
|
|
2024-01-08 05:14:59 -08:00
|
|
|
(blocks_output_stream, jh_block_emitter_task)
|
2023-12-22 05:42:20 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
struct SlotExtractor {}
|
|
|
|
|
|
|
|
impl FromYellowstoneExtractor for crate::grpc_multiplex::SlotExtractor {
|
|
|
|
type Target = SlotNotification;
|
|
|
|
fn map_yellowstone_update(&self, update: SubscribeUpdate) -> Option<(Slot, Self::Target)> {
|
|
|
|
match update.update_oneof {
|
|
|
|
Some(UpdateOneof::Slot(update_slot_message)) => {
|
|
|
|
let slot = SlotNotification {
|
|
|
|
estimated_processed_slot: update_slot_message.slot,
|
|
|
|
processed_slot: update_slot_message.slot,
|
|
|
|
};
|
|
|
|
Some((update_slot_message.slot, slot))
|
|
|
|
}
|
|
|
|
_ => None,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn create_grpc_multiplex_slots_subscription(
|
|
|
|
grpc_sources: Vec<GrpcSourceConfig>,
|
|
|
|
) -> (Receiver<SlotNotification>, AnyhowJoinHandle) {
|
|
|
|
info!("Setup grpc multiplexed slots connection...");
|
|
|
|
if grpc_sources.is_empty() {
|
|
|
|
info!("- no grpc connection configured");
|
|
|
|
}
|
|
|
|
for grpc_source in &grpc_sources {
|
|
|
|
info!("- connection to {}", grpc_source);
|
|
|
|
}
|
|
|
|
|
2024-01-17 06:17:49 -08:00
|
|
|
let (multiplexed_messages_sender, multiplexed_messages_rx) =
|
|
|
|
tokio::sync::broadcast::channel(1000);
|
2024-01-12 05:21:54 -08:00
|
|
|
|
|
|
|
let jh = tokio::spawn(async move {
|
|
|
|
loop {
|
2024-01-17 06:17:49 -08:00
|
|
|
let mut streams_tasks = Vec::new();
|
|
|
|
let mut recievers = Vec::new();
|
|
|
|
for grpc_source in &grpc_sources {
|
2024-01-17 10:45:10 -08:00
|
|
|
let (sx, rx) = async_channel::unbounded();
|
2024-01-17 06:17:49 -08:00
|
|
|
let task = create_slot_stream_task(
|
|
|
|
grpc_source.grpc_addr.clone(),
|
|
|
|
grpc_source.grpc_x_token.clone(),
|
|
|
|
sx,
|
|
|
|
yellowstone_grpc_proto::geyser::CommitmentLevel::Processed,
|
|
|
|
);
|
|
|
|
streams_tasks.push(task);
|
|
|
|
recievers.push(rx);
|
|
|
|
}
|
2023-12-22 05:42:20 -08:00
|
|
|
|
2024-01-17 06:17:49 -08:00
|
|
|
while let Ok(slot_update) = tokio::time::timeout(
|
|
|
|
Duration::from_secs(30),
|
|
|
|
futures::stream::select_all(recievers.clone()).next(),
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
{
|
|
|
|
if let Some(slot_update) = slot_update {
|
|
|
|
multiplexed_messages_sender.send(SlotNotification {
|
|
|
|
processed_slot: slot_update.slot,
|
|
|
|
estimated_processed_slot: slot_update.slot,
|
|
|
|
})?;
|
2024-01-12 05:21:54 -08:00
|
|
|
}
|
|
|
|
}
|
2024-01-17 06:17:49 -08:00
|
|
|
|
|
|
|
streams_tasks.iter().for_each(|task| task.abort());
|
2023-12-22 05:42:20 -08:00
|
|
|
}
|
2024-01-12 05:21:54 -08:00
|
|
|
});
|
2023-12-22 05:42:20 -08:00
|
|
|
|
2024-01-17 06:17:49 -08:00
|
|
|
(multiplexed_messages_rx, jh)
|
2023-12-22 05:42:20 -08:00
|
|
|
}
|