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

cleanup

This commit is contained in:
GroovieGermanikus 2023-12-20 08:54:20 +01:00
parent 93d192b7c5
commit ed944a64d5
No known key found for this signature in database
GPG Key ID: 5B6EB831A5CD2015
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/grpc_subscription_autoreconnect.rs
View File

@ -75,7 +75,7 @@ type Attempt = u32;
// wraps payload and status messages
pub enum Message {
GeyserSubscribeUpdate(SubscribeUpdate),
GeyserSubscribeUpdate(Box<SubscribeUpdate>),
// connect (attempt=1) or reconnect(attempt=2..)
Connecting(Attempt),
}
@ -148,7 +148,7 @@ pub fn create_geyser_reconnecting_stream(
(state, yield_value) = match state {
ConnectionState::NotConnected(mut attempt) => {
attempt = attempt + 1;
attempt += 1;
let connection_task = tokio::spawn({
let addr = grpc_source.grpc_addr.clone();
@ -233,7 +233,7 @@ pub fn create_geyser_reconnecting_stream(
match geyser_stream.next().await {
Some(Ok(update_message)) => {
trace!("> recv update message from {}", grpc_source);
(ConnectionState::Ready(attempt, geyser_stream), Message::GeyserSubscribeUpdate(update_message))
(ConnectionState::Ready(attempt, geyser_stream), Message::GeyserSubscribeUpdate(Box::new(update_message)))
}
Some(Err(tonic_status)) => {
// ATM we consider all errors recoverable

1
src/lib.rs
View File

@ -1,5 +1,4 @@
pub mod experimental;
pub mod grpc_stream_utils;
pub mod grpc_subscription;
pub mod grpc_subscription_autoreconnect;
pub mod grpcmultiplex_fastestwins;