print memory usage for queries

1 files changed, 18 insertions, 9 deletions

diff --git a/crates/ra_prof/src/memory_usage.rs b/crates/ra_prof/src/memory_usage.rs
index 2bde8fb5f..7f857942d 100644
--- a/crates/ra_prof/src/memory_usage.rs
+++ b/crates/ra_prof/src/memory_usage.rs
@@ -27,21 +27,23 @@ impl fmt::Display for MemoryUsage {
     }
 }
 
-#[derive(Default)]
+#[derive(Default, PartialEq, Eq, PartialOrd, Ord, Hash, Clone, Copy)]
 pub struct Bytes(usize);
 
 impl fmt::Display for Bytes {
     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
         let bytes = self.0;
-        if bytes < 4096 {
-            return write!(f, "{} bytes", bytes);
+        let mut value = bytes;
+        let mut suffix = "b";
+        if value > 4096 {
+            value /= 1024;
+            suffix = "kb";
+            if value > 4096 {
+                value /= 1024;
+                suffix = "mb";
+            }
         }
-        let kb = bytes / 1024;
-        if kb < 4096 {
-            return write!(f, "{}kb", kb);
-        }
-        let mb = kb / 1024;
-        write!(f, "{}mb", mb)
+        f.pad(&format!("{}{}", value, suffix))
     }
 }
 
@@ -50,3 +52,10 @@ impl std::ops::AddAssign<usize> for Bytes {
         self.0 += x;
     }
 }
+
+impl std::ops::Sub for Bytes {
+    type Output = Bytes;
+    fn sub(self, rhs: Bytes) -> Bytes {
+        Bytes(self.0 - rhs.0)
+    }
+}