[Fix] fix kv_params is null

oceanbase · Aug 31, 2024 · cb92eca · cb92eca
1 parent 214ff68
commit cb92eca
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/query/ObTableQuery.java b/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/query/ObTableQuery.java
@@ -178,7 +178,7 @@ public byte[] encode() {
             idx += len;
         }
 
-        if (isHbaseQuery) {
+        if (isHbaseQuery && obKVParams != null) {
             len = (int) obKVParams.getPayloadSize();
             System.arraycopy(obKVParams.encode(), 0, bytes, idx, len);
             idx += len;
@@ -277,9 +277,13 @@ public long getPayloadContentSize() {
 
         if (isHbaseQuery) {
             contentSize += hTableFilter.getPayloadSize();
+        } else {
+            contentSize += HTABLE_DUMMY_BYTES.length;
+        }
+        if (isHbaseQuery && obKVParams != null) {
             contentSize += obKVParams.getPayloadSize();
         } else {
-            contentSize += 2 * HTABLE_DUMMY_BYTES.length;
+            contentSize += HTABLE_DUMMY_BYTES.length;
         }
         contentSize += Serialization.getNeedBytes(scanRangeColumns.size());
         for (String scanRangeColumn : scanRangeColumns) {