From ff243870f850c07907a6944151fd909c234da662 Mon Sep 17 00:00:00 2001
From: Andrej Kacian <andrej@kacian.sk>
Date: Sun, 9 Jan 2022 20:07:50 +0100
Subject: [PATCH] Add "database_memory_usage" AdminCommand

---
 src/database/admin.rs | 8 ++++++++
 src/database/rooms.rs | 3 +++
 2 files changed, 11 insertions(+)

diff --git a/src/database/admin.rs b/src/database/admin.rs
index 0702bcdd..c3083309 100644
--- a/src/database/admin.rs
+++ b/src/database/admin.rs
@@ -14,6 +14,7 @@ pub enum AdminCommand {
     RegisterAppservice(serde_yaml::Value),
     UnregisterAppservice(String),
     ListAppservices,
+    ShowMemoryUsage,
     SendMessage(RoomMessageEventContent),
 }
 
@@ -113,6 +114,13 @@ impl Admin {
                                     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) => {
                                 send_message(message, guard, &state_lock);
                             }
diff --git a/src/database/rooms.rs b/src/database/rooms.rs
index 600f46df..0ba6c9ba 100644
--- a/src/database/rooms.rs
+++ b/src/database/rooms.rs
@@ -1693,6 +1693,9 @@ impl Rooms {
                                         ));
                                     }
                                 }
+                                "database_memory_usage" => {
+                                    db.admin.send(AdminCommand::ShowMemoryUsage);
+                                }
                                 _ => {
                                     db.admin.send(AdminCommand::SendMessage(
                                         RoomMessageEventContent::text_plain(format!(