From be1229ae542486aca338452c9094b4c3682d728d Mon Sep 17 00:00:00 2001
From: Theodore Schnepper <tschnepper@espressosys.com>
Date: Thu, 3 Oct 2024 07:33:24 -0600
Subject: [PATCH] Replace custom retry attempts with `BackoffParams`

---
 node-metrics/src/api/node_validator/v0/mod.rs | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/node-metrics/src/api/node_validator/v0/mod.rs b/node-metrics/src/api/node_validator/v0/mod.rs
index 81805d2e92..8331ddcd08 100644
--- a/node-metrics/src/api/node_validator/v0/mod.rs
+++ b/node-metrics/src/api/node_validator/v0/mod.rs
@@ -5,7 +5,7 @@ use crate::service::client_message::{ClientMessage, InternalClientMessage};
 use crate::service::data_state::{LocationDetails, NodeIdentity};
 use crate::service::server_message::ServerMessage;
 use async_std::task::JoinHandle;
-use espresso_types::SeqTypes;
+use espresso_types::{BackoffParams, SeqTypes};
 use futures::channel::mpsc::SendError;
 use futures::future::Either;
 use futures::{
@@ -20,7 +20,6 @@ use hotshot_types::traits::{signature_key::StakeTableEntryType, stake_table::Sta
 use hotshot_types::PeerConfig;
 use prometheus_parse::{Sample, Scrape};
 use serde::{Deserialize, Serialize};
-use std::cmp::max;
 use std::fmt;
 use std::future::Future;
 use std::io::BufRead;
@@ -597,6 +596,9 @@ impl ProcessProduceLeafStreamTask {
     where
         R: LeafStreamRetriever<Item = Leaf<SeqTypes>>,
     {
+        let backoff_params = BackoffParams::default();
+        let mut delay = Duration::ZERO;
+
         for attempt in 1..=100 {
             let leaves_stream_result = leaf_stream_receiver.retrieve_stream(None).await;
 
@@ -614,13 +616,11 @@ impl ProcessProduceLeafStreamTask {
                     // For every failed iteration, we will double our delay, up
                     // to the maximum of 5 seconds.
 
-                    let retry_penalty = max(
-                        Duration::from_millis(100) * 2u32.pow(attempt),
-                        Duration::from_secs(5),
-                    );
-                    async_std::task::sleep(retry_penalty).await;
+                    delay = backoff_params.backoff(delay);
+                    async_std::task::sleep(delay).await;
                     continue;
                 }
+
                 Ok(leaves_stream) => leaves_stream,
             };