mirrors/activitypub-federation-rust
1
0
Fork 0
mirror of https://github.com/LemmyNet/activitypub-federation-rust.git synced 2024-06-02 13:29:36 +00:00
Code Issues Projects Releases Wiki Activity

simplify shutdown interface

Browse source
This commit is contained in:
phiresky 2023-07-04 12:20:28 +00:00
parent 8065813b8f
commit e9f01583a4
1 changed files with 14 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
src/config.rs
View file

@ -184,22 +184,22 @@ impl<T: Clone> FederationConfig<T> {
pub fn domain(&self) -> &str {
&self.domain
}
/// Shut down this federation, waiting for the outgoing queue to be sent
/// If the activityqueue is still in use in other requests, returns an error
/// If wait_retries is true, also wait for requests that have initially failed and are being retried
/// Shut down this federation, waiting for the outgoing queue to be sent.
/// If the activityqueue is still in use in other requests or was never constructed, returns an error.
/// If wait_retries is true, also wait for requests that have initially failed and are being retried.
/// Returns a stats object that can be printed for debugging (structure currently not part of the public interface).
///
/// Currently, this method does not work correctly if worker_count = 0 (unlimited)
pub async fn shutdown(mut self, wait_retries: bool) -> anyhow::Result<()> {
if let Some(q) = self.activity_queue.take() {
let stats = Arc::<ActivityQueue>::into_inner(q)
.context(
"Could not cleanly shut down: activityqueue arc was still in use elsewhere ",
)?
.shutdown(wait_retries)
.await?;
tracing::info!("{:?}", stats);
}
Ok(())
pub async fn shutdown(mut self, wait_retries: bool) -> anyhow::Result<impl std::fmt::Debug> {
let q = self
.activity_queue
.take()
.context("ActivityQueue never constructed, build() not called?")?;
let stats = Arc::<ActivityQueue>::into_inner(q)
.context("Could not cleanly shut down: activityqueue arc was still in use elsewhere ")?
.shutdown(wait_retries)
.await?;
return Ok(stats);
}
}