Aleksandr BogomiakovandClaude
9966b50407
The command line package is now `tsunagi` and builds the `tsng` executable, which the packages, the unit, the installers and the workflows install and run. The core library package becomes `tsunagi-core`, keeping the library name `tsunagi`, so no code changes `use tsunagi::...`. Directories follow (crates/tsunagi-core, crates/tsunagi). Every command shown to a person is now `tsng ...`. Co-Authored-By: Claude <noreply@anthropic.com>
76 lines
2.0 KiB
Rust
76 lines
2.0 KiB
Rust
//! A minimal cancellation primitive.
|
|
//!
|
|
//! Kept local so the crate does not pull in a utility dependency for one type,
|
|
//! and so that no global state is involved: every agent and every network
|
|
//! runtime owns its own token.
|
|
|
|
use std::sync::Arc;
|
|
|
|
use tokio::sync::watch;
|
|
|
|
/// A clonable cancellation token.
|
|
#[derive(Debug, Clone)]
|
|
pub(crate) struct Shutdown {
|
|
tx: Arc<watch::Sender<bool>>,
|
|
rx: watch::Receiver<bool>,
|
|
}
|
|
|
|
impl Shutdown {
|
|
/// Creates an untriggered token.
|
|
pub(crate) fn new() -> Self {
|
|
let (tx, rx) = watch::channel(false);
|
|
Self {
|
|
tx: Arc::new(tx),
|
|
rx,
|
|
}
|
|
}
|
|
|
|
/// Triggers cancellation. Idempotent.
|
|
pub(crate) fn trigger(&self) {
|
|
let _ = self.tx.send(true);
|
|
}
|
|
|
|
/// Whether cancellation has been triggered.
|
|
pub(crate) fn is_triggered(&self) -> bool {
|
|
*self.rx.borrow()
|
|
}
|
|
|
|
/// Resolves once cancellation has been triggered.
|
|
pub(crate) async fn wait(&self) {
|
|
let mut rx = self.rx.clone();
|
|
{
|
|
if *rx.borrow() {
|
|
return;
|
|
}
|
|
}
|
|
while rx.changed().await.is_ok() {
|
|
if *rx.borrow() {
|
|
return;
|
|
}
|
|
}
|
|
// The sender is gone, which for our purposes means "stop".
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
#![allow(clippy::unwrap_used, clippy::expect_used, clippy::panic)]
|
|
|
|
use super::*;
|
|
|
|
#[tokio::test]
|
|
async fn waiting_resolves_once_triggered_and_stays_resolved() {
|
|
let shutdown = Shutdown::new();
|
|
assert!(!shutdown.is_triggered());
|
|
let waiter = shutdown.clone();
|
|
let waiting = tokio::spawn(async move { waiter.wait().await });
|
|
shutdown.trigger();
|
|
waiting.await.unwrap();
|
|
assert!(shutdown.is_triggered());
|
|
// Triggering twice is not an error, and a token cloned afterwards
|
|
// sees the trigger that already happened.
|
|
shutdown.trigger();
|
|
shutdown.clone().wait().await;
|
|
}
|
|
}
|