Fix panic in Uniform

mikedilger · Jun 29, 2024 · a1fa061 · a1fa061
1 parent ca98995
commit a1fa061
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/gossip-lib/src/overlord/mod.rs b/gossip-lib/src/overlord/mod.rs
@@ -526,13 +526,15 @@ impl Overlord {
         &mut self,
         url: RelayUrl,
         jobs: Vec<RelayJob>,
-        exclusion: u64,
+        mut exclusion: u64,
     ) {
         // Randomize the exclusion to between half and full
         use rand::Rng;
-        let exclusion = rand::thread_rng().sample(
-            rand::distributions::Uniform::new(exclusion/2, exclusion)
-        );
+        if exclusion > 1 {
+            exclusion = rand::thread_rng().sample(
+                rand::distributions::Uniform::new(exclusion/2, exclusion)
+            );
+        }
 
         // Let the relay picker know it disconnected
         GLOBALS