fix hbase/rocksdb shard bug

Change-Id: Ifce3d450712104c067a57c361351c9e047bc0cfa
apache · Dec 21, 2020 · 0fae667 · 0fae667
1 parent f0d1146
commit 0fae667
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/store/BackendTable.java b/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/store/BackendTable.java
@@ -254,8 +254,9 @@ public List<Shard> splitEven(int count) {
                                     length);
                 byte[] offset = start;
                 byte[] last = offset;
+                boolean finished = false;
                 List<Shard> shards = new ArrayList<>(count);
-                while (Bytes.compare(offset, end) < 0) {
+                while (Bytes.compare(offset, end) < 0 && !finished) {
                     offset = add(offset, each);
                     if (offset.length > end.length ||
                         Bytes.compare(offset, end) > 0) {
@@ -267,6 +268,7 @@ public List<Shard> splitEven(int count) {
                     }
                     if (endChanged && Arrays.equals(offset, end)) {
                         offset = this.endKey;
+                        finished = true;
                     }
                     shards.add(new Shard(startKey(last), endKey(offset), 0));
                     last = offset;