fix online-node selector

PolyhedraZK · Oct 9, 2024 · 2e4bd0b · 2e4bd0b
1 parent 496ce21
commit 2e4bd0b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/src/ddev.rs b/src/ddev.rs
@@ -22,7 +22,7 @@ use chaindev::{
 };
 use ruc::*;
 use serde::{Deserialize, Serialize};
-use std::{env, fs, str::FromStr};
+use std::{collections::HashSet, env, fs, str::FromStr};
 
 #[derive(Clone, Debug, Serialize, Deserialize)]
 pub struct EnvCfg {
@@ -333,7 +333,10 @@ fi "#
             let online_nodes = e
                 .nodes_should_be_online
                 .iter()
-                .copied()
+                .map(|(k, _)| k)
+                .filter(|k| *k != n.id)
+                .collect::<HashSet<_>>() // for random purpose
+                .into_iter()
                 .take(5)
                 .collect::<Vec<_>>();
 

diff --git a/src/dev.rs b/src/dev.rs
@@ -19,7 +19,7 @@ use chaindev::{
 };
 use ruc::*;
 use serde::{Deserialize, Serialize};
-use std::fs;
+use std::{collections::HashSet, fs};
 
 #[derive(Clone, Debug, Serialize, Deserialize)]
 pub struct EnvCfg {
@@ -221,7 +221,10 @@ fi "#
             let online_nodes = e
                 .nodes_should_be_online
                 .iter()
-                .copied()
+                .map(|(k, _)| k)
+                .filter(|k| *k != n.id)
+                .collect::<HashSet<_>>() // for random purpose
+                .into_iter()
                 .take(5)
                 .collect::<Vec<_>>();