diff --git a/Cargo.lock b/Cargo.lock index 0a07a71..96c9115 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -2325,6 +2325,7 @@ dependencies = [ "prost 0.11.9", "prost-build", "qp2p", + "rand 0.8.5", "regex", "serde", "serde_json", diff --git a/Cargo.toml b/Cargo.toml index 53317aa..ef8bded 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -6,7 +6,7 @@ edition = "2021" # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html [dependencies] -qp2p = "0.36.4" #{ path = "qp2p" } +qp2p = "0.36.4" #{ path = "qp2p" } tokio = { version = "1.32.0", features = ["full"] } thiserror = "1.0.50" async-trait = "0.1.74" @@ -39,6 +39,7 @@ wasmedge-sdk = { version = "0.10.1" } async-channel = "2.1.0" sysinfo = "0.29.10" ssh2 = "0.9.4" +rand = "0.8.5" # slog-envlogger = { version = "2.1.0", optional = true } diff --git a/src/schedule/master.rs b/src/schedule/master.rs index 94677cd..688dc09 100644 --- a/src/schedule/master.rs +++ b/src/schedule/master.rs @@ -5,7 +5,9 @@ use axum::{ http::StatusCode, response::{IntoResponse, Redirect, Response}, }; +use rand::{distributions::Uniform, prelude::Distribution, thread_rng}; use ws_derive::LogicalModule; +// use use super::{executor::Executor, http_handler::RequestHandler}; use crate::{ @@ -24,6 +26,43 @@ pub struct ScheMaster { // view: ScheMasterView, } +trait NodeWeighteFetcher: Send + Sync + 'static { + // NOTE: get weight return node weight + // larger is better + fn get_node_weight(&self, id: NodeID) -> f64; +} + +struct StrawNodeSelector { + weight_fetcher: Box, +} + +impl StrawNodeSelector { + fn new(weight_fetcher: Box) -> Self { + Self { weight_fetcher } + } +} + +// NOTE: Straw2 algorithm +impl NodeSelector for StrawNodeSelector { + fn select_node(&self, all_node_cnt: usize, _fn_name: &str) -> NodeID { + // NOTE: 1 is an impossible value for straw + let mut max_straw: f64 = 1.0; + let mut node_id: NodeID = 1; + let range = Uniform::new(0.0, 65536.0); + // NOTE: node id is [1,all_node_cnt] + for i in 1..all_node_cnt + 1 { + let weight = self.weight_fetcher.get_node_weight(i as NodeID); + let mut straw: f64 = range.sample(&mut thread_rng()); + straw = (straw / 65536.0).ln() / weight; + if (max_straw - 1.0).abs() < 0.000001 || max_straw < straw { + max_straw = straw; + node_id = i as NodeID; + } + } + return node_id; + } +} + trait NodeSelector: Send + Sync + 'static { fn select_node(&self, all_node_cnt: usize, fn_name: &str) -> NodeID; }