Merge pull request stratum-mining#1378 from Shourya742/2025-01-22-let…

…-pool-wait-for-template-provider

Add retry logic to ensure the pool waits and connect to TP

roles/pool/src/lib/template_receiver/mod.rs

@@ -18,7 +18,7 @@ use roles_logic_sv2::{
 };
 use std::{convert::TryInto, net::SocketAddr, sync::Arc};
 use tokio::{net::TcpStream, task};
-use tracing::info;
+use tracing::{info, warn};
 
 mod message_handler;
 mod setup_connection;
@@ -45,7 +45,15 @@ impl TemplateRx {
         coinbase_out_len: u32,
         expected_tp_authority_public_key: Option<Secp256k1PublicKey>,
     ) -> PoolResult<()> {
-        let stream = TcpStream::connect(address).await?;
+        let stream = loop {
+            match TcpStream::connect(address).await {
+                Ok(stream) => break stream,
+                Err(err) => {
+                    warn!("Failed to connect to {}: {}. Retrying...", address, err);
+                    tokio::time::sleep(std::time::Duration::from_secs(1)).await;
+                }
+            }
+        };
         info!("Connected to template distribution server at {}", address);
 
         let initiator = match expected_tp_authority_public_key {