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

fix double-increment of attempt

This commit is contained in:
GroovieGermanikus 2024-03-11 15:36:56 +01:00
parent 48efa7bbe8
commit 4ffe39849f
No known key found for this signature in database
GPG Key ID: 5B6EB831A5CD2015
2 changed files with 34 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/grpc_subscription_autoreconnect_streams.rs
View File

@ -66,7 +66,7 @@ pub fn create_geyser_reconnecting_stream(
}
});
(ConnectionState::Connecting(attempt + 1, connection_task), Message::Connecting(attempt + 1))
(ConnectionState::Connecting(attempt, connection_task), Message::Connecting(attempt + 1))
}
ConnectionState::Connecting(attempt, connection_task) => {

46
src/grpc_subscription_autoreconnect_tasks.rs
View File

@ -38,7 +38,8 @@ pub fn create_geyser_autoconnection_task(
) -> (AbortHandle, Receiver<Message>) {
let (sender, receiver_channel) = tokio::sync::mpsc::channel::<Message>(1);
let abort_handle = create_geyser_autoconnection_task_with_mpsc(grpc_source, subscribe_filter, sender);
let abort_handle =
create_geyser_autoconnection_task_with_mpsc(grpc_source, subscribe_filter, sender);
(abort_handle, receiver_channel)
}
@ -61,7 +62,6 @@ pub fn create_geyser_autoconnection_task_with_mpsc(
loop {
state = match state {
ConnectionState::NotConnected(attempt) => {
let addr = grpc_source.grpc_addr.clone();
let token = grpc_source.grpc_x_token.clone();
let config = grpc_source.tls_config.clone();
@ -88,14 +88,22 @@ pub fn create_geyser_autoconnection_task_with_mpsc(
.await;
match connect_result {
Ok(client) => ConnectionState::Connecting(attempt + 1, client),
Ok(client) => ConnectionState::Connecting(attempt, client),
Err(GeyserGrpcClientError::InvalidUri(_)) => ConnectionState::FatalError(
attempt + 1, FatalErrorReason::ConfigurationError,
), Err(GeyserGrpcClientError::MetadataValueError(_)) => {
ConnectionState::FatalError(attempt + 1, FatalErrorReason::ConfigurationError)
attempt + 1,
FatalErrorReason::ConfigurationError,
),
Err(GeyserGrpcClientError::MetadataValueError(_)) => {
ConnectionState::FatalError(
attempt + 1,
FatalErrorReason::ConfigurationError,
)
}
Err(GeyserGrpcClientError::InvalidXTokenLength(_)) => {
ConnectionState::FatalError(attempt + 1, FatalErrorReason::ConfigurationError)
ConnectionState::FatalError(
attempt + 1,
FatalErrorReason::ConfigurationError,
)
}
Err(GeyserGrpcClientError::TonicError(tonic_error)) => {
warn!(
@ -135,13 +143,16 @@ pub fn create_geyser_autoconnection_task_with_mpsc(
match subscribe_result_timeout {
Ok(subscribe_result) => {
match subscribe_result {
Ok(geyser_stream) => ConnectionState::Ready(geyser_stream),
Ok(geyser_stream) => {
debug!("subscribed to {} after {} failed attempts", grpc_source, attempt);
ConnectionState::Ready(geyser_stream)
},
Err(GeyserGrpcClientError::TonicError(_)) => {
warn!("subscribe failed on {} - retrying", grpc_source);
warn!("subscribe failed on {} after {} attempts - retrying", grpc_source, attempt + 1);
ConnectionState::RecoverableConnectionError(attempt + 1)
}
Err(GeyserGrpcClientError::TonicStatus(_)) => {
warn!("subscribe failed on {} - retrying", grpc_source);
warn!("subscribe failed on {} {} attempts - retrying", grpc_source, attempt + 1);
ConnectionState::RecoverableConnectionError(attempt + 1)
}
// non-recoverable
@ -150,7 +161,10 @@ pub fn create_geyser_autoconnection_task_with_mpsc(
"subscribe to {} failed with unrecoverable error: {}",
grpc_source, unrecoverable_error
);
ConnectionState::FatalError(attempt + 1, FatalErrorReason::SubscribeError)
ConnectionState::FatalError(
attempt + 1,
FatalErrorReason::SubscribeError,
)
}
}
}
@ -252,13 +266,19 @@ pub fn create_geyser_autoconnection_task_with_mpsc(
}
Err(_send_error) => {
warn!("downstream receiver closed, message is lost - aborting");
break 'recv_loop ConnectionState::FatalError(0, FatalErrorReason::DownstreamChannelClosed);
break 'recv_loop ConnectionState::FatalError(
0,
FatalErrorReason::DownstreamChannelClosed,
);
}
}
}
Err(SendTimeoutError::Closed(_)) => {
warn!("downstream receiver closed - aborting");
break 'recv_loop ConnectionState::FatalError(0, FatalErrorReason::DownstreamChannelClosed);
break 'recv_loop ConnectionState::FatalError(
0,
FatalErrorReason::DownstreamChannelClosed,
);
}
}
}