oss/conduit
oss/conduit
1
0
Fork 0
mirror of https://gitlab.com/famedly/conduit.git synced 2025-06-27 16:35:59 +00:00
Code Wiki Activity

Add "database_memory_usage" AdminCommand

Browse source
This commit is contained in:
Andrej Kacian 2022-01-09 20:07:50 +01:00
parent dd8429bc7f
commit ec3852953c
2 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/database/admin.rs
View file

@ -13,6 +13,7 @@ use tracing::warn;
pub enum AdminCommand { pub enum AdminCommand {
RegisterAppservice(serde_yaml::Value), RegisterAppservice(serde_yaml::Value),
ListAppservices, ListAppservices,
ShowMemoryUsage,
SendMessage(RoomMessageEventContent), SendMessage(RoomMessageEventContent),
} }
@ -109,6 +110,13 @@ impl Admin {
send_message(RoomMessageEventContent::text_plain("Failed to get appservices."), guard, &state_lock); send_message(RoomMessageEventContent::text_plain("Failed to get appservices."), guard, &state_lock);
} }
} }
AdminCommand::ShowMemoryUsage => {
if let Ok(response) = guard._db.memory_usage() {
send_message(RoomMessageEventContent::text_plain(response), guard, &state_lock);
} else {
send_message(RoomMessageEventContent::text_plain("Failed to get database memory usage".to_string()), guard, &state_lock);
}
}
AdminCommand::SendMessage(message) => { AdminCommand::SendMessage(message) => {
send_message(message, guard, &state_lock); send_message(message, guard, &state_lock);
} }

3
src/database/rooms.rs
View file

@ -1680,6 +1680,9 @@ impl Rooms {
)); ));
} }
} }
"database_memory_usage" => {
db.admin.send(AdminCommand::ShowMemoryUsage);
}
_ => { _ => {
db.admin.send(AdminCommand::SendMessage( db.admin.send(AdminCommand::SendMessage(
RoomMessageEventContent::text_plain(format!( RoomMessageEventContent::text_plain(format!(