chore: add missing accesors

moderately-ai · Jun 18, 2024 · e39f43a · e39f43a
1 parent 3a0db4c
commit e39f43a
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/mai-sdk-core/src/network.rs b/mai-sdk-core/src/network.rs
@@ -23,9 +23,9 @@ use std::{
 use tokio::{select, sync::RwLock};
 
 use crate::{
+    distributed_kv_store::{GetEvent, SetEvent},
     event_bridge::{EventBridge, PublishEvents},
     handler::Startable,
-    distributed_kv_store::{GetEvent, SetEvent},
 };
 
 use serde::{Deserialize, Serialize};
@@ -191,6 +191,14 @@ impl P2PNetwork {
             psk,
         }
     }
+
+    pub fn listen_addrs(&self) -> Vec<String> {
+        self.listen_addrs.iter().map(|a| a.to_string()).collect()
+    }
+
+    pub fn bootstrap_addrs(&self) -> Vec<String> {
+        self.bootstrap_addrs.iter().map(|a| a.to_string()).collect()
+    }
 }
 
 async fn handle_set_event(kad: &mut kad::Behaviour<MemoryStore>, event: SetEvent) -> Result<()> {