Get CF level disk usage

anujphadke · Aug 5, 2022 · 8b95ee1 · 8b95ee1
1 parent 4d1d0ab
commit 8b95ee1
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/priam/src/main/java/com/netflix/priam/connection/JMXNodeTool.java b/priam/src/main/java/com/netflix/priam/connection/JMXNodeTool.java
@@ -271,6 +271,22 @@ public JSONObject estimateKeys() throws JSONException {
         return object;
     }
 
+    public JSONObject getCFDiskUsage() {
+        Iterator<Entry<String, ColumnFamilyStoreMBean>> it =
+                super.getColumnFamilyStoreMBeanProxies();
+        JSONObject object = new JSONObject();
+        JSONObject object = new JSONObject();
+
+        while (it.hasNext()) {
+            Entry<String, ColumnFamilyStoreMBean> entry = it.next();
+            object.put("keyspace", entry.getKey());
+            object.put("column_family", entry.getValue().getColumnFamilyName());
+            object.put("live_disk_size", (long) super.getColumnFamilyMetric(entry.getKey(),
+                    entry.getValue().getColumnFamilyName(), "liveDiskSpaceUsed"));
+        }
+        return object;
+    }
+
     public JSONObject info() throws JSONException {
         JSONObject object = new JSONObject();
         object.put("gossip_active", isInitialized());