server: add register method

node: format

node & server: fix rpc ownership
This commit is contained in:
Luna Yao
2026-04-06 11:54:50 +02:00
parent 6c47b0d86c
commit 093cd7c197
2 changed files with 10 additions and 12 deletions
+6 -7
View File
@@ -140,16 +140,10 @@ impl DnsServer {
pub fn new(
peer_mgr: Arc<PeerManager>,
global_ctx: ArcGlobalCtx,
rpc: StandAloneServer<TcpTunnelListener>,
#[cfg(feature = "tun")] nic_ctx: ArcNicCtx, // TODO: REMOVE THIS
) -> Self {
let mgr = Arc::new(DnsNodeMgr::new());
rpc.registry()
.register(DnsNodeMgrRpcServer::new_arc(mgr.clone()), "");
Self {
mgr,
mgr: Arc::new(DnsNodeMgr::new()),
nic_ctx,
peer_mgr,
global_ctx,
@@ -158,6 +152,11 @@ impl DnsServer {
}
}
pub fn register(&self, rpc: &StandAloneServer<TcpTunnelListener>) {
rpc.registry()
.register(DnsNodeMgrRpcServer::new_arc(self.mgr.clone()), "");
}
pub fn addresses(&self) -> HashSet<SocketAddr> {
self.addresses.read().iter().map(|a| a.addr).collect()
}