Compare commits

..

1 Commits

Author SHA1 Message Date
cuddle-please
0eea32b57b chore(release): 0.1.1
All checks were successful
continuous-integration/drone/push Build is passing
continuous-integration/drone/pr Build is passing
2025-07-04 10:54:11 +00:00
4 changed files with 22 additions and 73 deletions

View File

@ -6,11 +6,6 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
## [Unreleased] ## [Unreleased]
## [0.1.2] - 2025-07-04
### Added
- update basic example with a more simple acquire and run function
## [0.1.1] - 2025-07-04 ## [0.1.1] - 2025-07-04
### Added ### Added

View File

@ -3,7 +3,7 @@ members = ["crates/*"]
resolver = "2" resolver = "2"
[workspace.package] [workspace.package]
version = "0.1.2" version = "0.1.1"
license = "MIT" license = "MIT"
[workspace.dependencies] [workspace.dependencies]

View File

@ -23,21 +23,28 @@ async fn main() -> anyhow::Result<()> {
tracing::info!("creating bucket"); tracing::info!("creating bucket");
leader.create_bucket().await?; leader.create_bucket().await?;
tokio::spawn({
let leader = leader.clone();
let leader_id = leader_id.clone();
async move {
tracing::debug!(leader_id, "starting leader");
leader
.start(CancellationToken::default())
.await
.expect("to succeed");
}
});
leader leader
.acquire_and_run({ .do_while_leader(move |token| async move {
move |token| { loop {
let leader_id = leader_id.clone(); if token.is_cancelled() {
return Ok(());
async move {
loop {
if token.is_cancelled() {
return Ok(());
}
tracing::info!(leader_id, "do work as leader");
tokio::time::sleep(std::time::Duration::from_secs(1)).await;
}
} }
tracing::info!("do work as leader");
tokio::time::sleep(std::time::Duration::from_secs(1)).await;
} }
}) })
.await?; .await?;

View File

@ -31,72 +31,19 @@ impl Leader {
} }
} }
pub async fn acquire_and_run<F, Fut>(&self, f: F) -> anyhow::Result<()>
where
F: Fn(CancellationToken) -> Fut,
Fut: Future<Output = anyhow::Result<()>> + Send + 'static,
{
let parent_token = CancellationToken::default();
let s = self.clone();
let server_token = parent_token.child_token();
// Start the server election process in another task, this is because start is blocking
let handle = tokio::spawn({
let server_token = server_token.child_token();
async move {
match s.start(server_token).await {
Ok(_) => {}
Err(e) => tracing::error!("leader election process failed: {}", e),
}
tracing::info!("shutting down noleader");
parent_token.cancel();
}
});
// Do the work if we're leader
let res = self
.do_while_leader_inner(server_token.child_token(), f)
.await;
// Stop the server election process if our provided functions returns an error;
server_token.cancel();
// Close down the task as well, it should already be stopped, but this forces the task to close
handle.abort();
res?;
Ok(())
}
pub async fn do_while_leader<F, Fut>(&self, f: F) -> anyhow::Result<()> pub async fn do_while_leader<F, Fut>(&self, f: F) -> anyhow::Result<()>
where
F: Fn(CancellationToken) -> Fut,
Fut: Future<Output = anyhow::Result<()>> + Send + 'static,
{
self.do_while_leader_inner(CancellationToken::new(), f)
.await
}
async fn do_while_leader_inner<F, Fut>(
&self,
cancellation_token: CancellationToken,
f: F,
) -> anyhow::Result<()>
where where
F: Fn(CancellationToken) -> Fut, F: Fn(CancellationToken) -> Fut,
Fut: Future<Output = anyhow::Result<()>> + Send + 'static, Fut: Future<Output = anyhow::Result<()>> + Send + 'static,
{ {
loop { loop {
let cancellation_token = cancellation_token.child_token();
let is_leader = self.is_leader.clone(); let is_leader = self.is_leader.clone();
if !is_leader.load(Ordering::Relaxed) { if !is_leader.load(Ordering::Relaxed) {
tokio::time::sleep(Duration::from_secs(1)).await; tokio::time::sleep(Duration::from_secs(1)).await;
continue; continue;
} }
let cancellation_token = CancellationToken::new();
let child_token = cancellation_token.child_token(); let child_token = cancellation_token.child_token();
let guard = tokio::spawn(async move { let guard = tokio::spawn(async move {