use in grpc_subscription_autoreconnect_streams

This commit is contained in:
GroovieGermanikus 2024-03-22 18:05:49 +01:00
parent 066f9a8800
commit 130845f30a
No known key found for this signature in database
GPG Key ID: 5B6EB831A5CD2015
1 changed files with 17 additions and 7 deletions

View File

@ -1,4 +1,4 @@
use crate::{Attempt, GrpcSourceConfig, Message};
use crate::{Attempt, GrpcSourceConfig, Message, yellowstone_grpc_util};
use async_stream::stream;
use futures::{Stream, StreamExt};
use log::{debug, info, log, trace, warn, Level};
@ -45,12 +45,22 @@ pub fn create_geyser_reconnecting_stream(
log!(if attempt > 1 { Level::Warn } else { Level::Debug }, "Connecting attempt #{} to {}", attempt, addr);
async move {
let connect_result = GeyserGrpcClient::connect_with_timeout(
addr, token, config,
connect_timeout,
request_timeout,
false)
.await;
// let connect_result = GeyserGrpcClient::connect_with_timeout(
// addr, token, config,
// connect_timeout,
// request_timeout,
// false)
// .await;
warn!("Use HACKED version of connect_with_timeout_hacked");
let connect_result = yellowstone_grpc_util::connect_with_timeout_hacked(
addr,
token,
// config,
// connect_timeout,
// request_timeout,
// false,
)
.await;
let mut client = connect_result?;