fix bug: index rebuild exceeds 80w limit and 65536 cassandra limit

fixed: #291 Change-Id: Iadf1b18c7a1be7a6343ae1fdeffbdb268ee1be90
apache · Dec 25, 2018 · cdf1dfe · cdf1dfe
1 parent 85383c9
commit cdf1dfe
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/tx/GraphTransaction.java b/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/tx/GraphTransaction.java
@@ -1411,7 +1411,7 @@ private <T> void traverseByLabel(SchemaLabel label,
          * Query.DEFAULT_CAPACITY to limit elements number per pass
          */
         query.limit(Query.DEFAULT_CAPACITY);
-        query.capacity(Query.DEFAULT_CAPACITY);
+        query.capacity(Query.NO_CAPACITY);
         query.eq(HugeKeys.LABEL, label.id());
         int pass = 0;
         int counter = 0;

diff --git a/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/tx/IndexableTransaction.java b/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/tx/IndexableTransaction.java
@@ -57,6 +57,15 @@ protected void commit2Backend() {
         this.commitMutation2Backend(mutation, txMutation);
     }
 
+    @Override
+    public void commitIfGtSize(int size) throws BackendException {
+        int totalSize = this.mutationSize() +
+                        this.indexTransaction().mutationSize();
+        if (totalSize >= size) {
+            this.commit();
+        }
+    }
+
     @Override
     public void rollback() throws BackendException {
         try {