mirror of
https://github.com/EasyTier/EasyTier.git
synced 2026-09-02 09:09:17 +00:00
server: add register method
node: format node & server: fix rpc ownership
This commit is contained in:
@@ -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()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user