Fix inappropriate shard will cause base64 decode error (#1383)

* Fix shard start is 0 will cause base64 decode error Change-Id: Ibc90df4e792b00222d41806fee769082545dea40
apache · Mar 29, 2021 · 61466d6 · 61466d6
1 parent a495eea
commit 61466d6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
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
@@ -168,9 +168,9 @@ public List<Shard> getSplits(Session session, long splitSize) {
             long maxKey = this.maxKey();
             double each = maxKey / count;
 
-            long offset = 0L;
-            String last = this.position(offset);
             List<Shard> splits = new ArrayList<>((int) count);
+            String last = START;
+            long offset = 0L;
             while (offset < maxKey) {
                 offset += each;
                 if (offset > maxKey) {

diff --git a/hugegraph-rocksdb/src/main/java/com/baidu/hugegraph/backend/store/rocksdb/RocksDBTable.java b/hugegraph-rocksdb/src/main/java/com/baidu/hugegraph/backend/store/rocksdb/RocksDBTable.java
@@ -327,7 +327,7 @@ public long estimateNumKeys(Session session) {
 
         @Override
         public byte[] position(String position) {
-            if (END.equals(position)) {
+            if (START.equals(position) || END.equals(position)) {
                 return null;
             }
             return StringEncoding.decodeBase64(position);