mirror/Easytier
1
0
Fork 0
mirror of https://github.com/EasyTier/EasyTier.git synced 2026-08-31 08:19:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

client_mgr: add

Browse Source 
format

client_mgr
This commit is contained in:
Luna Yao
2026-02-21 22:38:36 +01:00
parent b7073504d8
commit dc1e1929c8
5 changed files with 270 additions and 153 deletions
Download Patch File Download Diff File Expand all files Collapse all files
easytier/src/dns/client.rs
+5 -5
View File
@@ -1,7 +1,7 @@
use crate::dns::config::DNS_SERVER_RPC_ADDR;
use crate::dns::peer_mgr::DnsPeerMgr;
use crate::peers::peer_manager::PeerManager;
use crate::proto::dns::{DnsPeerMgrRpcServer, DnsServerRpcClientFactory, HeartbeatRequest};
use crate::proto::dns::{DnsClientMgrRpcClientFactory, DnsPeerMgrRpcServer, HeartbeatRequest};
use crate::proto::peer_rpc::RoutePeerInfo;
use crate::proto::rpc_impl::standalone::StandAloneClient;
use crate::proto::rpc_types::controller::BaseController;
@@ -63,22 +63,22 @@ impl DnsClient {
) -> anyhow::Result<()> {
let request = if heartbeat.snapshot.is_none() || self.mgr.dirty.peers.reset() {
heartbeat.update(self.mgr.snapshot());
heartbeat.clone().into()
heartbeat.clone()
} else {
let snapshot = heartbeat.snapshot.take();
let request = heartbeat.clone().into();
let request = heartbeat.clone();
heartbeat.snapshot = snapshot;
request
};
let client = rpc
.scoped_client::<DnsServerRpcClientFactory<BaseController>>("".to_string())
.scoped_client::<DnsClientMgrRpcClientFactory<BaseController>>("".to_string())
.await?;
let response = client.heartbeat(BaseController::default(), request).await?;
if response.resync {
client
.heartbeat(BaseController::default(), heartbeat.clone().into())
.heartbeat(BaseController::default(), heartbeat.clone())
.await?;
}