Merge pull request #113 from carlaKC/111-randomactivity

Random Activity Generator

sim-lib/Cargo.toml

@@ -25,3 +25,4 @@ rand = "0.8.5"
 hex = "0.4.3"
 csv = "1.2.2"
 serde_millis = "0.1.1"
+rand_distr = "0.4.3"

sim-lib/src/cln.rs

@@ -2,8 +2,8 @@ use async_trait::async_trait;
 use bitcoin::secp256k1::PublicKey;
 use cln_grpc::pb::{
     listpays_pays::ListpaysPaysStatus, node_client::NodeClient, Amount, GetinfoRequest,
-    GetinfoResponse, KeysendRequest, KeysendResponse, ListnodesRequest, ListpaysRequest,
-    ListpaysResponse,
+    GetinfoResponse, KeysendRequest, KeysendResponse, ListchannelsRequest, ListnodesRequest,
+    ListpaysRequest, ListpaysResponse,
 };
 use lightning::ln::features::NodeFeatures;
 use lightning::ln::PaymentHash;
@@ -85,6 +85,36 @@ impl ClnNode {
             },
         })
     }
+
+    /// Fetch channels belonging to the local node, initiated locally if is_source is true, and initiated remotely if
+    /// is_source is false. Introduced as a helper function because CLN doesn't have a single API to list all of our
+    /// node's channels.
+    async fn node_channels(&mut self, is_source: bool) -> Result<Vec<u64>, LightningError> {
+        let req = if is_source {
+            ListchannelsRequest {
+                source: Some(self.info.pubkey.serialize().to_vec()),
+                ..Default::default()
+            }
+        } else {
+            ListchannelsRequest {
+                destination: Some(self.info.pubkey.serialize().to_vec()),
+                ..Default::default()
+            }
+        };
+
+        let resp = self
+            .client
+            .list_channels(req)
+            .await
+            .map_err(|err| LightningError::ListChannelsError(err.to_string()))?
+            .into_inner();
+
+        Ok(resp
+            .channels
+            .into_iter()
+            .map(|channel| channel.amount_msat.unwrap_or_default().msat)
+            .collect())
+    }
 }
 
 #[async_trait]
@@ -198,6 +228,12 @@ impl LightningNode for ClnNode {
             ))
         }
     }
+
+    async fn list_channels(&mut self) -> Result<Vec<u64>, LightningError> {
+        let mut node_channels = self.node_channels(true).await?;
+        node_channels.extend(self.node_channels(false).await?);
+        Ok(node_channels)
+    }
 }
 
 async fn reader(filename: &str) -> Result<Vec<u8>, Error> {