Avoid unnecessary panic

This commit is contained in:
Marcin Kulik
2025-06-06 22:01:35 +02:00
parent 2e932e5f0e
commit 32626d0ee8

View File

@@ -33,17 +33,13 @@ pub async fn forward<N: Notifier>(
subscriber: Subscriber, subscriber: Subscriber,
mut notifier: N, mut notifier: N,
shutdown_token: tokio_util::sync::CancellationToken, shutdown_token: tokio_util::sync::CancellationToken,
) { ) -> anyhow::Result<()> {
info!("forwarding to {url}"); info!("forwarding to {url}");
let mut reconnect_attempt = 0; let mut reconnect_attempt = 0;
let mut connection_count: u64 = 0; let mut connection_count: u64 = 0;
loop { loop {
let session_stream = subscriber let session_stream = subscriber.subscribe().await?;
.subscribe()
.await
.expect("stream should be alive");
let conn = connect_and_forward(&url, session_stream); let conn = connect_and_forward(&url, session_stream);
tokio::pin!(conn); tokio::pin!(conn);
@@ -129,6 +125,8 @@ pub async fn forward<N: Notifier>(
_ = shutdown_token.cancelled() => break _ = shutdown_token.cancelled() => break
} }
} }
Ok(())
} }
async fn connect_and_forward( async fn connect_and_forward(