magmaus3/continuwuity
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

make service memory_usage()/clear_cache() async trait

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2025-03-15 04:07:53 +00:00
parent d8ea8b378c
commit 9ce95a7030
9 changed files with 61 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

57
src/service/services.rs
View file

@ -1,12 +1,12 @@
use std::{
any::Any,
collections::BTreeMap,
fmt::Write,
sync::{Arc, RwLock},
};
use conduwuit::{Result, Server, debug, debug_info, info, trace};
use conduwuit::{Result, Server, debug, debug_info, info, trace, utils::stream::IterStream};
use database::Database;
use futures::{Stream, StreamExt, TryStreamExt};
use tokio::sync::Mutex;
use crate::{
@ -171,40 +171,21 @@ impl Services {
}
pub async fn clear_cache(&self) {
for (service, ..) in self.service.read().expect("locked for reading").values() {
if let Some(service) = service.upgrade() {
service.clear_cache();
}
}
//TODO
self.rooms
.spaces
.roomid_spacehierarchy_cache
.lock()
.await
.clear();
self.services()
.for_each(|service| async move {
service.clear_cache().await;
})
.await;
}
pub async fn memory_usage(&self) -> Result<String> {
let mut out = String::new();
for (service, ..) in self.service.read().expect("locked for reading").values() {
if let Some(service) = service.upgrade() {
service.memory_usage(&mut out)?;
}
}
//TODO
let roomid_spacehierarchy_cache = self
.rooms
.spaces
.roomid_spacehierarchy_cache
.lock()
self.services()
.map(Ok)
.try_fold(String::new(), |mut out, service| async move {
service.memory_usage(&mut out).await?;
Ok(out)
})
.await
.len();
writeln!(out, "roomid_spacehierarchy_cache: {roomid_spacehierarchy_cache}")?;
Ok(out)
}
fn interrupt(&self) {
@ -217,6 +198,18 @@ impl Services {
}
}
/// Iterate from snapshot of the services map
fn services(&self) -> impl Stream<Item = Arc<dyn Service>> + Send {
self.service
.read()
.expect("locked for reading")
.values()
.filter_map(|val| val.0.upgrade())
.collect::<Vec<_>>()
.into_iter()
.stream()
}
#[inline]
pub fn try_get<T>(&self, name: &str) -> Result<Arc<T>>
where