p2pool: use indexing for nodes instead of for loop comparisons

This commit is contained in:
hinto-janaiyo 2022-11-19 13:03:33 -05:00
parent ff98b2a303
commit 2f6bb53027
No known key found for this signature in database
GPG key ID: B1C5A64B80691E45
3 changed files with 72 additions and 32 deletions

View file

@ -90,6 +90,7 @@ pub const P2POOL_NODE_IP: &'static str = "Specify the Monero Node IP to connect
pub const P2POOL_RPC_PORT: &'static str = "Specify the RPC port of the Monero node; [1-65535]"; pub const P2POOL_RPC_PORT: &'static str = "Specify the RPC port of the Monero node; [1-65535]";
pub const P2POOL_ZMQ_PORT: &'static str = "Specify the ZMQ port of the Monero node; [1-65535]"; pub const P2POOL_ZMQ_PORT: &'static str = "Specify the ZMQ port of the Monero node; [1-65535]";
pub const P2POOL_ADD: &'static str = "Add the current values to the list"; pub const P2POOL_ADD: &'static str = "Add the current values to the list";
pub const P2POOL_SAVE: &'static str = "Save the current values to the already existing entry";
pub const P2POOL_DELETE: &'static str = "Delete the currently selected node"; pub const P2POOL_DELETE: &'static str = "Delete the currently selected node";
pub const P2POOL_CLEAR: &'static str = "Clear all current values"; pub const P2POOL_CLEAR: &'static str = "Clear all current values";

View file

@ -248,7 +248,7 @@ pub async fn ping(ping: Arc<Mutex<Ping>>, og: Arc<Mutex<State>>) -> Result<(), a
} }
for handle in handles { for handle in handles {
handle.await; handle.await?;
} }
let node_vec = node_vec.lock().unwrap().clone(); let node_vec = node_vec.lock().unwrap().clone();

View file

@ -291,56 +291,95 @@ impl P2pool {
ui.spacing_mut().slider_width = width - 8.0; ui.spacing_mut().slider_width = width - 8.0;
ui.spacing_mut().icon_width = width / 25.0; ui.spacing_mut().icon_width = width / 25.0;
// [Ping List] // [Ping List]
let text = RichText::new(format!("{}. {} | {}", self.selected_index, self.selected_name, self.selected_ip)); let text = RichText::new(format!("{}. {}", self.selected_index, self.selected_name));
ComboBox::from_id_source("manual_nodes").selected_text(RichText::text_style(text, Monospace)).show_ui(ui, |ui| { ComboBox::from_id_source("manual_nodes").selected_text(RichText::text_style(text, Monospace)).show_ui(ui, |ui| {
let mut n = 1; let mut n = 1;
for (name, node) in node_vec.iter() { for (name, node) in node_vec.iter() {
let text = RichText::text_style(RichText::new(format!("{}. {}\n IP: {}\n RPC: {}\n ZMQ: {}", n, name, node.ip, node.rpc, node.zmq)), Monospace); let text = RichText::text_style(RichText::new(format!("{}. {}\n IP: {}\n RPC: {}\n ZMQ: {}", n, name, node.ip, node.rpc, node.zmq)), Monospace);
if ui.add(SelectableLabel::new(self.selected_name == *name, text)).clicked() { if ui.add(SelectableLabel::new(self.selected_name == *name, text)).clicked() {
self.selected_index = n; self.selected_index = n;
let node = node.clone();
self.selected_name = name.clone(); self.selected_name = name.clone();
self.selected_ip = node.ip.clone();
self.selected_rpc = node.rpc.clone();
self.selected_zmq = node.zmq.clone();
self.name = name.clone();
self.ip = node.ip;
self.rpc = node.rpc;
self.zmq = node.zmq;
} }
n += 1; n += 1;
} }
}); });
// [Add] + [Delete] // [Add/Save]
let node_vec_len = node_vec.len(); let node_vec_len = node_vec.len();
ui.horizontal(|ui| { let mut exists = false;
let mut exists = false; let mut save_diff = true;
for (name, _) in node_vec.iter() { let mut existing_index = 0;
if *name == self.name { exists = true; } for (name, node) in node_vec.iter() {
if *name == self.name {
exists = true;
if self.ip == node.ip && self.rpc == node.rpc && self.zmq == node.zmq {
save_diff = false;
}
break
} }
ui.set_enabled(!incorrect_input && !exists && node_vec_len < 1000); existing_index += 1;
let text = format!("{}\n Currently selected node: {}. {}\n Current amount of nodes: {}/1000", P2POOL_ADD, self.selected_index, self.selected_name, node_vec_len); }
if ui.add_sized([width, text_edit], Button::new("Add")).on_hover_text(text).clicked() { ui.horizontal(|ui| {
let node = Node { let text;
ip: self.ip.clone(), if exists { text = P2POOL_SAVE } else { text = P2POOL_ADD }
rpc: self.rpc.clone(), let text = format!("{}\n Currently selected node: {}. {}\n Current amount of nodes: {}/1000", text, self.selected_index, self.selected_name, node_vec_len);
zmq: self.zmq.clone(), // If the node already exists, show [Save] and mutate the already existing node
}; if exists {
node_vec.push((self.name.clone(), node)); ui.set_enabled(save_diff);
info!("Node | Added [index: {}, name: \"{}\", ip: \"{}\", rpc: {}, zmq: {}]", node_vec_len+1, self.name, self.ip, self.rpc, self.zmq); if ui.add_sized([width, text_edit], Button::new("Save")).on_hover_text(text).clicked() {
let node = Node {
ip: self.ip.clone(),
rpc: self.rpc.clone(),
zmq: self.zmq.clone(),
};
node_vec[existing_index as usize].1 = node;
info!("Node | S | [index: {}, name: \"{}\", ip: \"{}\", rpc: {}, zmq: {}]", existing_index+1, self.name, self.ip, self.rpc, self.zmq);
}
// Else, add to the list
} else {
ui.set_enabled(!incorrect_input && node_vec_len < 1000);
if ui.add_sized([width, text_edit], Button::new("Add")).on_hover_text(text).clicked() {
let node = Node {
ip: self.ip.clone(),
rpc: self.rpc.clone(),
zmq: self.zmq.clone(),
};
node_vec.push((self.name.clone(), node));
self.selected_index = (node_vec_len+1) as u16;
self.selected_name = self.name.clone();
self.selected_ip = self.ip.clone();
self.selected_rpc = self.rpc.clone();
self.selected_zmq = self.zmq.clone();
info!("Node | A | [index: {}, name: \"{}\", ip: \"{}\", rpc: {}, zmq: {}]", node_vec_len+1, self.name, self.ip, self.rpc, self.zmq);
}
} }
}); });
// [Delete]
ui.horizontal(|ui| { ui.horizontal(|ui| {
ui.set_enabled(node_vec_len > 1); ui.set_enabled(node_vec_len > 1);
let text = format!("{}\n Currently selected node: {}. {}\n Current amount of nodes: {}/1000", P2POOL_DELETE, self.selected_index, self.selected_name, node_vec_len); let text = format!("{}\n Currently selected node: {}. {}\n Current amount of nodes: {}/1000", P2POOL_DELETE, self.selected_index, self.selected_name, node_vec_len);
if ui.add_sized([width, text_edit], Button::new("Delete")).on_hover_text(text).clicked() { if ui.add_sized([width, text_edit], Button::new("Delete")).on_hover_text(text).clicked() {
let mut n = 0; let new_index = self.selected_index-1;
for (name, _) in node_vec.iter() { let new_name = node_vec[(new_index-1) as usize].0.clone();
if *name == self.selected_name { let new_node = node_vec[(new_index-1) as usize].1.clone();
// If deleting [0], make selected = [1] self.selected_index = new_index;
// instead of attempting to [0-1] (panic!) self.selected_name = new_name.clone();
match n { self.selected_ip = new_node.ip.clone();
0 => { self.selected_name = node_vec[1].0.clone(); self.selected_index = 1; }, self.selected_rpc = new_node.rpc.clone();
_ => { self.selected_name = node_vec[n-1].0.clone(); self.selected_index = n as u16; }, self.selected_zmq = new_node.zmq.clone();
}; self.name = new_name;
node_vec.remove(n); self.ip = new_node.ip;
info!("Node | Deleted [index: {}, name: \"{}\", ip: \"{}\", rpc: {}, zmq: {}]", n+1, self.selected_name, self.selected_ip, self.selected_rpc, self.selected_zmq); self.rpc = new_node.rpc;
break self.zmq = new_node.zmq;
} node_vec.remove(self.selected_index as usize);
n += 1; info!("Node | D | [index: {}, name: \"{}\", ip: \"{}\", rpc: {}, zmq: {}]", self.selected_index, self.selected_name, self.selected_ip, self.selected_rpc, self.selected_zmq);
}
} }
}); });
ui.horizontal(|ui| { ui.horizontal(|ui| {