Feature hashrate distribution

.gitignore

@@ -1 +1,2 @@
 /target
+/cpuminer-opt

src/config.rs

@@ -8,9 +8,11 @@ use std::{
 use tracing::{error, info, warn};
 
 use crate::{HashUnit, DEFAULT_SV1_HASHPOWER};
+
 lazy_static! {
     pub static ref CONFIG: Configuration = Configuration::load_config();
 }
+
 #[derive(Parser)]
 struct Args {
     #[clap(long)]
@@ -39,12 +41,14 @@ struct Args {
     config_file: Option<PathBuf>,
     #[clap(long = "api-server-port", short = 's')]
     api_server_port: Option<String>,
+    #[clap(long = "hashrate-distribution", value_delimiter = ',')]
+    hashrate_distribution: Option<Vec<f32>>,
 }
 
 #[derive(Serialize, Deserialize)]
 struct ConfigFile {
-    token: Option<String>,
     tp_address: Option<String>,
+    token: Option<String>,
     pool_addresses: Option<Vec<String>>,
     test_pool_addresses: Option<Vec<String>>,
     interval: Option<u64>,
@@ -55,6 +59,7 @@ struct ConfigFile {
     test: Option<bool>,
     listening_addr: Option<String>,
     api_server_port: Option<String>,
+    hashrate_distribution: Option<Vec<f32>>,
 }
 
 pub struct Configuration {
@@ -70,7 +75,9 @@ pub struct Configuration {
     test: bool,
     listening_addr: Option<String>,
     api_server_port: String,
+    hashrate_distribution: Option<Vec<f32>>,
 }
+
 impl Configuration {
     pub fn token() -> Option<String> {
         CONFIG.token.clone()
@@ -137,6 +144,14 @@ impl Configuration {
         CONFIG.test
     }
 
+    pub fn hashrate_distribution() -> Option<Vec<f32>> {
+        CONFIG.hashrate_distribution.clone()
+    }
+
+    pub fn wants_hashrate_distribution() -> bool {
+        CONFIG.hashrate_distribution.is_some()
+    }
+
     // Loads config from CLI, file, or env vars with precedence: CLI > file > env.
     fn load_config() -> Self {
         let args = Args::parse();
@@ -157,6 +172,7 @@ impl Configuration {
                 test: None,
                 listening_addr: None,
                 api_server_port: None,
+                hashrate_distribution: None,
             });
 
         let token = args
@@ -285,6 +301,16 @@ impl Configuration {
             .unwrap_or("off".to_string());
 
         let test = args.test || config.test.unwrap_or(false) || std::env::var("TEST").is_ok();
+        let hashrate_distribution = args
+            .hashrate_distribution
+            .or(config.hashrate_distribution)
+            .or_else(|| {
+                std::env::var("HASHRATE_DISTRIBUTION").ok().and_then(|s| {
+                    s.split(',')
+                        .map(|s| s.trim().parse::<f32>().ok())
+                        .collect::<Option<Vec<f32>>>()
+                })
+            });
 
         Configuration {
             token,
@@ -299,6 +325,7 @@ impl Configuration {
             test,
             listening_addr,
             api_server_port,
+            hashrate_distribution,
         }
     }
 }