diff --git a/easytier/src/dns/client.rs b/easytier/src/dns/client.rs index ece4bed3..c996da87 100644 --- a/easytier/src/dns/client.rs +++ b/easytier/src/dns/client.rs @@ -61,7 +61,7 @@ impl DnsClient { rpc: &mut StandAloneClient, heartbeat: &mut HeartbeatRequest, ) -> anyhow::Result<()> { - let request = if heartbeat.snapshot.is_none() || self.mgr.dirty.peers.reset() { + let request = if heartbeat.snapshot.is_none() || self.mgr.dirty.reset() { heartbeat.update(self.mgr.snapshot()); heartbeat.clone() } else { diff --git a/easytier/src/dns/peer_mgr.rs b/easytier/src/dns/peer_mgr.rs index 3af82c52..2bb3b6eb 100644 --- a/easytier/src/dns/peer_mgr.rs +++ b/easytier/src/dns/peer_mgr.rs @@ -39,15 +39,10 @@ impl TryFrom for DnsPeerInfo { const DNS_PEER_TTL: Duration = Duration::from_secs(3); -#[derive(Debug, Default)] -pub struct DnsPeerMgrDirtyFlags { - pub(crate) peers: DirtyFlag, -} - #[derive(Debug, Deref)] pub struct DnsPeerMgr { peers: Cache, - pub(super) dirty: DirtyState, + pub(super) dirty: DirtyState, #[deref] mgr: Arc, @@ -92,7 +87,7 @@ impl DnsPeerMgr { } }; - self.dirty.peers.mark(); + self.dirty.mark(); match self.fetch(peer_id).await { Ok(info) => {