feat: workers provide a price at connection

Lagrange-Labs · Nov 27, 2024 · e64476f · e64476f
1 parent 281f545
commit e64476f
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/lgn-worker/src/main.rs b/lgn-worker/src/main.rs
@@ -28,6 +28,9 @@ use lgn_messages::types::UpstreamPayload;
 use lgn_worker::avs::utils::read_keystore;
 use metrics::counter;
 use mimalloc::MiMalloc;
+use rand::distributions::Uniform;
+use rand::prelude::Distribution;
+use serde_json::Number;
 use tokio::io::AsyncWriteExt;
 use tokio_stream::StreamExt;
 use tonic::metadata::MetadataValue;
@@ -622,6 +625,10 @@ fn get_claims(config: &Config) -> Result<Claims>
     };
 
     let version = env!("CARGO_PKG_VERSION");
+    let price_range = Uniform::new(
+        3000,
+        5000,
+    );
     let private = [
         (
             "version".to_string(),
@@ -636,6 +643,10 @@ fn get_claims(config: &Config) -> Result<Claims>
                     .to_string(),
             ),
         ),
+        (
+            "price".to_string(),
+            serde_json::Value::Number(Number::from(price_range.sample(&mut rand::thread_rng()))),
+        ),
     ]
     .into_iter()
     .collect::<BTreeMap<String, serde_json::Value>>();