use super::*;
#[derive(Clone)]
pub struct Thread {
voter: Voter,
}
impl Thread {
async fn run_once(&self) -> Result<()> {
let election_state = self.voter.read_election_state();
ensure!(std::matches!(
election_state,
voter::ElectionState::Follower
));
if let Some(timeout) = self.voter.get_election_timeout() {
tokio::time::sleep(timeout).await;
}
if self.voter.get_election_timeout().is_some() {
info!("election timeout. try to become a leader");
self.voter.try_promote(false).await?;
}
Ok(())
}
fn do_loop(self) -> ThreadHandle {
let hdl = tokio::spawn(async move {
let mut interval = tokio::time::interval(Duration::from_millis(100));
loop {
interval.tick().await;
self.run_once().await.ok();
}
})
.abort_handle();
ThreadHandle(hdl)
}
}
pub fn new(voter: Voter) -> ThreadHandle {
Thread { voter }.do_loop()
}