mirror/Easytier
1
0
Fork 0
mirror of https://github.com/EasyTier/EasyTier.git synced 2026-09-02 17:15:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

utils: add DirtyFlag

Browse Source
This commit is contained in:
Luna Yao
2026-04-06 11:54:01 +02:00
parent e3ed91aced
commit 3d3dba28e8
4 changed files with 59 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files
easytier/src/dns/client.rs
+9 -11
View File
@@ -6,7 +6,6 @@ use crate::proto::peer_rpc::RoutePeerInfo;
use crate::proto::rpc_impl::standalone::StandAloneClient;
use crate::proto::rpc_types::controller::BaseController;
use crate::tunnel::tcp::TcpTunnelConnector;
use std::sync::atomic::Ordering;
use std::sync::Arc;
use std::time::Duration;
use tokio::task::JoinSet;
@@ -61,16 +60,15 @@ impl DnsClient {
rpc: &mut StandAloneClient<TcpTunnelConnector>,
heartbeat: &mut HeartbeatRequest,
) -> anyhow::Result<()> {
let request =
if heartbeat.snapshot.is_none() || self.mgr.dirty.swap(false, Ordering::Acquire) {
heartbeat.update(self.mgr.snapshot());
heartbeat.clone().into()
} else {
let snapshot = heartbeat.snapshot.take();
let request = heartbeat.clone().into();
heartbeat.snapshot = snapshot;
request
};
let request = if heartbeat.snapshot.is_none() || self.mgr.dirty.reset() {
heartbeat.update(self.mgr.snapshot());
heartbeat.clone().into()
} else {
let snapshot = heartbeat.snapshot.take();
let request = heartbeat.clone().into();
heartbeat.snapshot = snapshot;
request
};
let client = rpc
.scoped_client::<DnsServerRpcClientFactory<BaseController>>("".to_string())