add file listing to database abstraction.

Signed-off-by: Jason Volk <[email protected]>

src/database/abstraction.rs

@@ -34,6 +34,8 @@ pub(crate) trait KeyValueDatabaseEngine: Send + Sync {
 	fn backup(&self) -> Result<(), Box<dyn Error>> { unimplemented!() }
 
 	fn backup_list(&self) -> Result<String> { Ok(String::new()) }
+
+	fn file_list(&self) -> Result<String> { Ok(String::new()) }
 }
 
 pub(crate) trait KvTree: Send + Sync {

src/database/abstraction/rocksdb.rs

@@ -304,6 +304,30 @@ impl KeyValueDatabaseEngine for Arc<Engine> {
 		Ok(res)
 	}
 
+	fn file_list(&self) -> Result<String> {
+		match self.rocks.live_files() {
+			Err(e) => Ok(String::from(e)),
+			Ok(files) => {
+				let mut res = String::new();
+				for file in files {
+					let _ = std::fmt::write(
+						&mut res,
+						format_args!(
+							"<code>L{} {:<13} {:7}+ {:4}- {:9}</code> {}<br>",
+							file.level,
+							file.name,
+							file.num_entries,
+							file.num_deletions,
+							file.size,
+							file.column_family_name,
+						),
+					);
+				}
+				Ok(res)
+			},
+		}
+	}
+
 	// TODO: figure out if this is needed for rocksdb
 	#[allow(dead_code)]
 	fn clear_caches(&self) {}

src/database/key_value/globals.rs

@@ -286,4 +286,6 @@ lasttimelinecount_cache: {lasttimelinecount_cache}\n"
 	fn backup(&self) -> Result<(), Box<dyn std::error::Error>> { self.db.backup() }
 
 	fn backup_list(&self) -> Result<String> { self.db.backup_list() }
+
+	fn file_list(&self) -> Result<String> { self.db.file_list() }
 }

src/service/globals/data.rs

@@ -37,4 +37,5 @@ pub trait Data: Send + Sync {
 	fn bump_database_version(&self, new_version: u64) -> Result<()>;
 	fn backup(&self) -> Result<(), Box<dyn Error>> { unimplemented!() }
 	fn backup_list(&self) -> Result<String> { Ok(String::new()) }
+	fn file_list(&self) -> Result<String> { Ok(String::new()) }
 }