jitter using u64 only

permesi · Feb 2, 2024 · 02bd82d · 02bd82d
1 parent b5e155b
commit 02bd82d
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/vault/renew.rs b/src/vault/renew.rs
@@ -119,9 +119,10 @@ pub async fn try_renew(globals: &GlobalArgs, tx: mpsc::UnboundedSender<()>) -> R
 
                     match renew_token(&url, &token, None).await {
                         Ok(lease_duration) => {
-                            jittered_lease_duration = Duration::from_secs(
-                                (lease_duration as f64 * rng.gen_range(0.7..0.9)) as u64,
-                            );
+                            let factor = rng.gen_range(70..90);
+
+                            jittered_lease_duration =
+                                Duration::from_secs(lease_duration * factor / 100);
 
                             break;
                         }
@@ -173,9 +174,10 @@ pub async fn try_renew(globals: &GlobalArgs, tx: mpsc::UnboundedSender<()>) -> R
 
                     match renew_db_token(&url, &token, &db_lease_id, db_lease_duration).await {
                         Ok(lease_duration) => {
-                            jittered_lease_duration = Duration::from_secs(
-                                (lease_duration as f64 * rng.gen_range(0.7..0.9)) as u64,
-                            );
+                            let factor = rng.gen_range(70..90);
+
+                            jittered_lease_duration =
+                                Duration::from_secs(lease_duration * factor / 100);
 
                             break;
                         }