From ec3852953c6eb5d36d00375026e99cfc75cea7a5 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 1e5c47c9..0051d5e1 100644
--- a/src/database/admin.rs
+++ b/src/database/admin.rs
@@ -13,6 +13,7 @@ use tracing::warn;
 pub enum AdminCommand {
     RegisterAppservice(serde_yaml::Value),
     ListAppservices,
+    ShowMemoryUsage,
     SendMessage(RoomMessageEventContent),
 }
 
@@ -109,6 +110,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 ff18cd47..807a37b0 100644
--- a/src/database/rooms.rs
+++ b/src/database/rooms.rs
@@ -1680,6 +1680,9 @@ impl Rooms {
                                         ));
                                     }
                                 }
+                                "database_memory_usage" => {
+                                    db.admin.send(AdminCommand::ShowMemoryUsage);
+                                }
                                 _ => {
                                     db.admin.send(AdminCommand::SendMessage(
                                         RoomMessageEventContent::text_plain(format!(