mirror/Easytier
1
0
Fork 0
mirror of https://github.com/EasyTier/EasyTier.git synced 2026-08-30 15:59:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

node: add Notify for server election

Browse Source
This commit is contained in:
Luna Yao
2026-02-22 23:58:40 +01:00
parent 14155d706c
commit 0c8da66a74
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
easytier/src/dns/node.rs
+9 -4
View File
@@ -9,7 +9,7 @@ use crate::proto::rpc_types::controller::BaseController;
use crate::tunnel::tcp::TcpTunnelConnector; use crate::tunnel::tcp::TcpTunnelConnector;
use std::sync::Arc; use std::sync::Arc;
use std::time::Duration; use std::time::Duration;
use tokio::sync::broadcast; use tokio::sync::{broadcast, Notify};
use tokio::task::JoinSet; use tokio::task::JoinSet;
use tokio::time::{sleep_until, Instant}; use tokio::time::{sleep_until, Instant};
use uuid::Uuid; use uuid::Uuid;
@@ -17,10 +17,12 @@ use uuid::Uuid;
#[derive(Debug)] #[derive(Debug)]
pub struct DnsNode { pub struct DnsNode {
mgr: Arc<DnsPeerMgr>, mgr: Arc<DnsPeerMgr>,
election: Arc<Notify>,
} }
impl DnsNode { impl DnsNode {
pub fn new(peer_mgr: Arc<PeerManager>) -> Self { pub fn new(peer_mgr: Arc<PeerManager>, election: Arc<Notify>) -> Self {
let mgr = Arc::new(DnsPeerMgr::new(peer_mgr.clone())); let mgr = Arc::new(DnsPeerMgr::new(peer_mgr.clone()));
peer_mgr peer_mgr
.get_peer_rpc_mgr() .get_peer_rpc_mgr()
@@ -31,7 +33,10 @@ impl DnsNode {
&peer_mgr.get_global_ctx_ref().get_network_name(), &peer_mgr.get_global_ctx_ref().get_network_name(),
); );
Self { mgr } Self {
mgr,
election,
}
} }
pub fn id(&self) -> Uuid { pub fn id(&self) -> Uuid {
@@ -67,8 +72,8 @@ impl DnsNode {
_ = &mut sleep => { _ = &mut sleep => {
if let Err(e) = self.heartbeat(&mut rpc, &mut heartbeat).await { if let Err(e) = self.heartbeat(&mut rpc, &mut heartbeat).await {
// TODO: try to start server
tracing::error!("heartbeat failed: {:?}", e); tracing::error!("heartbeat failed: {:?}", e);
self.election.notify_one();
} }
last_heartbeat = Instant::now(); last_heartbeat = Instant::now();