blockworks-foundation
/
geyser-grpc-connector
mirror of https://github.com/blockworks-foundation/geyser-grpc-connector.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

use channel_plugger in test

This commit is contained in:
GroovieGermanikus 2024-01-19 11:46:19 +01:00
parent 46f56872e5
commit 87a4bbc484
No known key found for this signature in database
GPG Key ID: 5B6EB831A5CD2015
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/stream_blocks_autoconnect.rs
View File

@ -18,6 +18,7 @@ use tracing::warn;
use yellowstone_grpc_proto::geyser::subscribe_update::UpdateOneof;
use yellowstone_grpc_proto::geyser::SubscribeUpdate;
use yellowstone_grpc_proto::prost::Message as _;
use geyser_grpc_connector::channel_plugger::spawn_plugger_mpcs_to_broadcast;
fn start_example_blockmini_consumer(
multiplex_stream: impl Stream<Item = BlockMini> + Send + 'static,
@ -106,10 +107,13 @@ pub async fn main() {
info!("Write Block stream..");
let (broadcast_tx, broadcast_rx) = tokio::sync::broadcast::channel(100);
let (jh_geyser_task, mut message_channel) = create_geyser_autoconnection_task(
green_config.clone(),
GeyserFilter(CommitmentConfig::confirmed()).blocks_and_txs(),
);
spawn_plugger_mpcs_to_broadcast(message_channel, broadcast_tx);
let mut message_channel = broadcast_rx;
tokio::spawn(async move {
if let TestCases::SlowReceiverStartup = TEST_CASE {
@ -117,7 +121,7 @@ pub async fn main() {
}
let mut message_count = 0;
while let Some(message) = message_channel.recv().await {
while let Ok(message) = message_channel.recv().await {
if let TestCases::AbortTaskFromOutside = TEST_CASE {
if message_count > 5 {
info!("(testcase) aborting task from outside");

2
src/lib.rs
View File

@ -8,11 +8,11 @@ use yellowstone_grpc_proto::geyser::{
};
use yellowstone_grpc_proto::tonic::transport::ClientTlsConfig;
mod channel_plugger;
pub mod grpc_subscription;
pub mod grpc_subscription_autoreconnect_streams;
pub mod grpc_subscription_autoreconnect_tasks;
pub mod grpcmultiplex_fastestwins;
pub mod channel_plugger;
mod obfuscate;
#[derive(Clone, Debug)]