Get CF level disk usage via JMXNodetool

Netflix · Aug 5, 2022 · a1a7678 · a1a7678
1 parent 4d1d0ab
commit a1a7678
Showing 1 changed file with 24 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,30 @@ public JSONObject estimateKeys() throws JSONException {
         return object;
     }
 
+    /**
+     * Get CF level disk usage
+     *
+     * @return
+     */
+    public JSONObject getCFDiskUsage() throws JSONException {
+        Iterator<Entry<String, ColumnFamilyStoreMBean>> it =
+                super.getColumnFamilyStoreMBeanProxies();
+        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",
+                    super.getColumnFamilyMetric(
+                            entry.getKey(),
+                            entry.getValue().getColumnFamilyName(),
+                            "liveDiskSpaceUsed"));
+        }
+        return object;
+    }
+
     public JSONObject info() throws JSONException {
         JSONObject object = new JSONObject();
         object.put("gossip_active", isInitialized());