make cassandra compact in the scope of keyspace

Change-Id: I321121bd43ef4895a4a9fe815d818d62eb74c925
apache · Jun 2, 2021 · 5f347a9 · 5f347a9
1 parent 8549b68
commit 5f347a9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/...cassandra/src/main/java/com/baidu/hugegraph/backend/store/cassandra/CassandraMetrics.java b/...cassandra/src/main/java/com/baidu/hugegraph/backend/store/cassandra/CassandraMetrics.java
@@ -39,6 +39,7 @@
 import com.baidu.hugegraph.backend.store.cassandra.CassandraTables.Vertex;
 import com.baidu.hugegraph.config.CoreOptions;
 import com.baidu.hugegraph.config.HugeConfig;
+import com.baidu.hugegraph.testutil.Whitebox;
 import com.baidu.hugegraph.util.E;
 import com.baidu.hugegraph.util.InsertionOrderUtil;
 import com.baidu.hugegraph.util.UnitUtil;
@@ -157,7 +158,7 @@ protected void appendExtraMetrics(Map<String, Object> metrics,
         appendCounterMetrics(metrics, probe, this.keyspace, this.tables,
                              "BloomFilterFalseRatio");
 
-        //Table timer Metrics
+        // Table timer Metrics
         appendTimerMetrics(metrics, probe, this.keyspace, "WriteLatency");
         appendTimerMetrics(metrics, probe, this.keyspace, "ReadLatency");
         appendTimerMetrics(metrics, probe, null, "WriteLatency");
@@ -266,6 +267,13 @@ public Map<String, Object> compact() {
     private Object compactHost(String host) {
         try (NodeProbe probe = this.newNodeProbe(host)) {
             Compact compact = new Compact();
+            /*
+             * Set the keyspace to be compacted
+             * NOTE: use Whitebox due to no public api is provided, args format
+             * is [<keyspace> <tables>...], the first arg means keyspace.
+             */
+            Whitebox.invoke(compact, "args", new Class<?>[]{Object.class},
+                            "add", this.keyspace);
             compact.execute(probe);
             return "OK";
         } catch (Throwable e) {

diff --git a/...ph-scylladb/src/main/java/com/baidu/hugegraph/backend/store/scylladb/ScyllaDBMetrics.java b/...ph-scylladb/src/main/java/com/baidu/hugegraph/backend/store/scylladb/ScyllaDBMetrics.java
@@ -50,7 +50,7 @@ protected void appendExtraMetrics(Map<String, Object> metrics,
         appendCounterMetrics(metrics, probe, this.keyspace(), this.tables(),
                              "BloomFilterFalseRatio");
 
-        //Table timer Metrics
+        // Table timer Metrics
         //appendTimerMetrics(metrics, probe, this.keyspace(), "WriteLatency");
         //appendTimerMetrics(metrics, probe, this.keyspace(), "ReadLatency");
         //appendTimerMetrics(metrics, probe, null, "WriteLatency");