opt: reduce calls for get_payload_content

GroundWu · GroundWu · commit 9dc35b94b9f6 · 2025-03-05T19:36:38.000+08:00
diff --git a/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/AbstractPayload.java b/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/AbstractPayload.java
@@ -177,10 +177,10 @@ public void setSequence(long sequence) {
      * encode unis header
      */
     protected int encodeHeader(byte[] bytes, int idx) {
-        int headerLen = (int) getObUniVersionHeaderLength(getVersion(), getPayloadContentSize());
-        System.arraycopy(encodeObUniVersionHeader(getVersion(), getPayloadContentSize()), 0, bytes,
-            idx, headerLen);
-        idx += headerLen;
+        byte[] versionHeaderBytes = encodeObUniVersionHeader(getVersion(), getPayloadContentSize());
+        System.arraycopy(versionHeaderBytes, 0, bytes,
+            idx, versionHeaderBytes.length);
+        idx += versionHeaderBytes.length;
         return idx;
     }
 
diff --git a/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableLSOpRequest.java b/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableLSOpRequest.java
@@ -72,9 +72,8 @@ public byte[] encode() {
         idx++;
 
         // 4. encode lsOperation
-        int len = (int) lsOperation.getPayloadSize();
-        System.arraycopy(lsOperation.encode(), 0, bytes, idx, len);
-        idx += len;
+        byte[] lsOpBytes = lsOperation.encode();
+        System.arraycopy(lsOpBytes, 0, bytes, idx, lsOpBytes.length);
 
         return bytes;
     }
diff --git a/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableLSOperation.java b/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableLSOperation.java
@@ -70,48 +70,48 @@ public byte[] encode() {
         idx += 8;
 
         // 2. encode table name
-        int len =  Serialization.getNeedBytes(tableName);
-        System.arraycopy(Serialization.encodeVString(tableName), 0, bytes, idx, len);
-        idx += len;
+        byte[] tableNameBytes = Serialization.encodeVString(tableName);
+        System.arraycopy(tableNameBytes, 0, bytes, idx, tableNameBytes.length);
+        idx += tableNameBytes.length;
 
         // 3. encode table id
-        len = Serialization.getNeedBytes(tableId);
-        System.arraycopy(Serialization.encodeVi64(tableId), 0, bytes, idx, len);
-        idx += len;
+        byte[] tableIdBytes = Serialization.encodeVi64(tableId);
+        System.arraycopy(tableIdBytes, 0, bytes, idx, tableIdBytes.length);
+        idx += tableIdBytes.length;
 
         // 4. encode rowKey names
-        len = Serialization.getNeedBytes(rowKeyNames.size());
-        System.arraycopy(Serialization.encodeVi64(rowKeyNames.size()), 0, bytes, idx, len);
-        idx += len;
+        byte[] rowKeyNameLenBytes = Serialization.encodeVi64(rowKeyNames.size());
+        System.arraycopy(rowKeyNameLenBytes, 0, bytes, idx, rowKeyNameLenBytes.length);
+        idx += rowKeyNameLenBytes.length;
         for (String rowKeyName : rowKeyNames) {
-            len =  Serialization.getNeedBytes(rowKeyName);
-            System.arraycopy(Serialization.encodeVString(rowKeyName), 0, bytes, idx, len);
-            idx += len;
+            byte[] rowKeyNameBytes =  Serialization.encodeVString(rowKeyName);
+            System.arraycopy(rowKeyNameBytes, 0, bytes, idx, rowKeyNameBytes.length);
+            idx += rowKeyNameBytes.length;
         }
 
         // 5. encode properties names
-        len = Serialization.getNeedBytes(propertiesNames.size());
-        System.arraycopy(Serialization.encodeVi64(propertiesNames.size()), 0, bytes, idx, len);
-        idx += len;
+        byte[] propertiesNamesLenBytes = Serialization.encodeVi64(propertiesNames.size());
+        System.arraycopy(propertiesNamesLenBytes, 0, bytes, idx, propertiesNamesLenBytes.length);
+        idx += propertiesNamesLenBytes.length;
         for (String propertyName : propertiesNames) {
-            len =  Serialization.getNeedBytes(propertyName);
-            System.arraycopy(Serialization.encodeVString(propertyName), 0, bytes, idx, len);
-            idx += len;
+            byte[] propertyNameByte = Serialization.encodeVString(propertyName);
+            System.arraycopy(propertyNameByte, 0, bytes, idx, propertyNameByte.length);
+            idx += propertyNameByte.length;
         }
 
         // 6. encode option flag
-        len = Serialization.getNeedBytes(optionFlag.getValue());
-        System.arraycopy(Serialization.encodeVi64(optionFlag.getValue()), 0, bytes, idx, len);
-        idx += len;
+        byte[] optionFlagBytes = Serialization.encodeVi64(optionFlag.getValue());
+        System.arraycopy(optionFlagBytes, 0, bytes, idx, optionFlagBytes.length);
+        idx += optionFlagBytes.length;
 
         // 7. encode Operation
-        len = Serialization.getNeedBytes(tabletOperations.size());
-        System.arraycopy(Serialization.encodeVi64(tabletOperations.size()), 0, bytes, idx, len);
-        idx += len;
+        byte[] tabletOperationsLenBytes = Serialization.encodeVi64(tabletOperations.size());
+        System.arraycopy(tabletOperationsLenBytes, 0, bytes, idx, tabletOperationsLenBytes.length);
+        idx += tabletOperationsLenBytes.length;
         for (ObTableTabletOp tabletOperation : tabletOperations) {
-            len = (int) tabletOperation.getPayloadSize();
-            System.arraycopy(tabletOperation.encode(), 0, bytes, idx, len);
-            idx += len;
+            byte[] tabletOperationBytes = tabletOperation.encode();
+            System.arraycopy(tabletOperationBytes, 0, bytes, idx, tabletOperationBytes.length);
+            idx += tabletOperationBytes.length;
         }
 
         return bytes;
diff --git a/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableSingleOp.java b/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableSingleOp.java
@@ -52,25 +52,25 @@ public byte[] encode() {
 
         // 2. encode op flag
         long flag = singleOpFlag.getValue();
-        int len = Serialization.getNeedBytes(flag);
-        System.arraycopy(Serialization.encodeVi64(flag), 0, bytes, idx, len);
-        idx += len;
+        byte[] singleOpFlagLenBytes = Serialization.encodeVi64(flag);
+        System.arraycopy(singleOpFlagLenBytes, 0, bytes, idx, singleOpFlagLenBytes.length);
+        idx += singleOpFlagLenBytes.length;
 
         // 3. encode single op query
         if (ObTableOperationType.needEncodeQuery(singleOpType)) {
-            len = (int) query.getPayloadSize();
-            System.arraycopy(query.encode(), 0, bytes, idx, len);
-            idx += len;
+            byte[] queryBytes = query.encode();
+            System.arraycopy(queryBytes, 0, bytes, idx, queryBytes.length);
+            idx += queryBytes.length;
         }
 
         // 4. encode entities
-        len = Serialization.getNeedBytes(entities.size());
-        System.arraycopy(Serialization.encodeVi64(entities.size()), 0, bytes, idx, len);
-        idx += len;
+        byte[] entitiesSizeBytes = Serialization.encodeVi64(entities.size());
+        System.arraycopy(entitiesSizeBytes, 0, bytes, idx, entitiesSizeBytes.length);
+        idx += entitiesSizeBytes.length;
         for (ObTableSingleOpEntity entity : entities) {
-            len = (int) entity.getPayloadSize();
-            System.arraycopy(entity.encode(), 0, bytes, idx, len);
-            idx += len;
+            byte[] entityBytes = entity.encode();
+            System.arraycopy(entityBytes, 0, bytes, idx, entityBytes.length);
+            idx += entityBytes.length;
         }
 
         return bytes;
diff --git a/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableSingleOpEntity.java b/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableSingleOpEntity.java
@@ -56,47 +56,47 @@ public byte[] encode() {
         idx = encodeHeader(bytes, idx);
 
         // 1. encode rowKey bitmap
-        int len = Serialization.getNeedBytes(rowKeyBitLen);
-        System.arraycopy(Serialization.encodeVi64(rowKeyBitLen), 0, bytes, idx, len);
-        idx += len;
+        byte[] tmpBytes = Serialization.encodeVi64(rowKeyBitLen);
+        System.arraycopy(tmpBytes, 0, bytes, idx, tmpBytes.length);
+        idx += tmpBytes.length;
         for (byte b : rowKeyBitMap) {
             System.arraycopy(Serialization.encodeI8(b), 0, bytes, idx, 1);
             idx += 1;
         }
 
         // 2. encode rowkey
-        len = Serialization.getNeedBytes(rowkey.size());
-        System.arraycopy(Serialization.encodeVi64(rowkey.size()), 0, bytes, idx, len);
-        idx += len;
+        tmpBytes = Serialization.encodeVi64(rowkey.size());
+        System.arraycopy(tmpBytes, 0, bytes, idx, tmpBytes.length);
+        idx += tmpBytes.length;
         for (ObObj obj : rowkey) {
-            len =  ObTableSerialUtil.getEncodedSize(obj);
-            System.arraycopy(ObTableSerialUtil.encode(obj), 0, bytes, idx, len);
-            idx += len;
+            tmpBytes =  ObTableSerialUtil.encode(obj);
+            System.arraycopy(tmpBytes, 0, bytes, idx, tmpBytes.length);
+            idx += tmpBytes.length;
         }
 
         // 3. encode property bitmap
         if (ignoreEncodePropertiesColumnNames) {
-            len = Serialization.getNeedBytes(0L);
-            System.arraycopy(Serialization.encodeVi64(0L), 0, bytes, idx, len);
-            idx += len;
+            tmpBytes = Serialization.encodeVi64(0L);
+            System.arraycopy(tmpBytes, 0, bytes, idx, tmpBytes.length);
+            idx += tmpBytes.length;
         } else {
-            len = Serialization.getNeedBytes(propertiesBitLen);
-            System.arraycopy(Serialization.encodeVi64(propertiesBitLen), 0, bytes, idx, len);
-            idx += len;
+            tmpBytes = Serialization.encodeVi64(propertiesBitLen);
+            System.arraycopy(tmpBytes, 0, bytes, idx, tmpBytes.length);
+            idx += tmpBytes.length;
             for (byte b : propertiesBitMap) {
                 System.arraycopy(Serialization.encodeI8(b), 0, bytes, idx, 1);
                 idx += 1;
             }
         }
 
         // 4. encode properties values
-        len = Serialization.getNeedBytes(propertiesValues.size());
-        System.arraycopy(Serialization.encodeVi64(propertiesValues.size()), 0, bytes, idx, len);
-        idx += len;
+        tmpBytes = Serialization.encodeVi64(propertiesValues.size());
+        System.arraycopy(tmpBytes, 0, bytes, idx, tmpBytes.length);
+        idx += tmpBytes.length;
         for (ObObj obj : propertiesValues) {
-            len =  ObTableSerialUtil.getEncodedSize(obj);
-            System.arraycopy(ObTableSerialUtil.encode(obj), 0, bytes, idx, len);
-            idx += len;
+            tmpBytes = ObTableSerialUtil.encode(obj);
+            System.arraycopy(tmpBytes, 0, bytes, idx, tmpBytes.length);
+            idx += tmpBytes.length;
         }
 
         return bytes;
diff --git a/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableSingleOpQuery.java b/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableSingleOpQuery.java
@@ -48,60 +48,57 @@ public byte[] encode() {
         idx = encodeHeader(bytes, idx);
 
         // 1. encode index name
-        int len =  Serialization.getNeedBytes(indexName);
-        System.arraycopy(Serialization.encodeVString(indexName), 0, bytes, idx, len);
-        idx += len;
+        byte[] indexNameBytes =  Serialization.encodeVString(indexName);
+        System.arraycopy(indexNameBytes, 0, bytes, idx, indexNameBytes.length);
+        idx += indexNameBytes.length;
 
         // 2. encode scan ranges columns
-        len = Serialization.getNeedBytes(scanRangeBitLen);
-        System.arraycopy(Serialization.encodeVi64(scanRangeBitLen), 0, bytes, idx, len);
-        idx += len;
+        byte[] scanRangeBitLenBytes = Serialization.encodeVi64(scanRangeBitLen);
+        System.arraycopy(scanRangeBitLenBytes, 0, bytes, idx, scanRangeBitLenBytes.length);
+        idx += scanRangeBitLenBytes.length;
         for (byte b : scanRangeBitMap) {
             System.arraycopy(Serialization.encodeI8(b), 0, bytes, idx, 1);
             idx += 1;
         }
 
         // 3. encode scan ranges
-        len = Serialization.getNeedBytes(keyRanges.size());
-        System.arraycopy(Serialization.encodeVi64(keyRanges.size()), 0, bytes, idx, len);
-        idx += len;
+        byte[] keyRangesBytes = Serialization.encodeVi64(keyRanges.size());
+        System.arraycopy(keyRangesBytes, 0, bytes, idx, keyRangesBytes.length);
+        idx += keyRangesBytes.length;
         for (ObNewRange range : keyRanges) {
-            len =  ObTableSerialUtil.getEncodedSize(range);
-            System.arraycopy(ObTableSerialUtil.encode(range), 0, bytes, idx, len);
-            idx += len;
+            byte[] rangeBytes =  ObTableSerialUtil.encode(range);
+            System.arraycopy(rangeBytes, 0, bytes, idx, rangeBytes.length);
+            idx += rangeBytes.length;
         }
 
         // 4. encode filter string
-        len =  Serialization.getNeedBytes(filterString);
-        System.arraycopy(Serialization.encodeVString(filterString), 0, bytes, idx, len);
-        idx += len;
+        byte[] filterStringBytes =  Serialization.encodeVString(filterString);
+        System.arraycopy(filterStringBytes, 0, bytes, idx, filterStringBytes.length);
+        idx += filterStringBytes.length;
 
         // encode HBase Batch Get required
         if (isHbaseQuery && ObGlobal.isHBaseBatchGetSupport()) {
-            len = Serialization.getNeedBytes(selectColumns.size());
-            System.arraycopy(Serialization.encodeVi64(selectColumns.size()), 0, bytes, idx, len);
-            idx += len;
+            byte[] selectColumnsLenByets = Serialization.encodeVi64(selectColumns.size());
+            System.arraycopy(selectColumnsLenByets, 0, bytes, idx, selectColumnsLenByets.length);
+            idx += selectColumnsLenByets.length;
             for (String selectColumn : selectColumns) {
-                len = Serialization.getNeedBytes(selectColumn);
-                System.arraycopy(Serialization.encodeVString(selectColumn), 0, bytes, idx, len);
-                idx += len;
+                byte[] selectColumnLenBytes = Serialization.encodeVString(selectColumn);
+                System.arraycopy(selectColumnLenBytes, 0, bytes, idx, selectColumnLenBytes.length);
+                idx += selectColumnLenBytes.length;
             }
 
             System.arraycopy(Serialization.encodeI8(scanOrder.getByteValue()), 0, bytes, idx, 1);
             idx += 1;
 
-            len = (int) hTableFilter.getPayloadSize();
-            System.arraycopy(hTableFilter.encode(), 0, bytes, idx, len);
-            idx += len;
+            byte[] hTableFilterLenBytes = hTableFilter.encode();
+            System.arraycopy(hTableFilterLenBytes, 0, bytes, idx, hTableFilterLenBytes.length);
+            idx += hTableFilterLenBytes.length;
 
             if (obKVParams != null) {
-                len = (int) obKVParams.getPayloadSize();
-                System.arraycopy(obKVParams.encode(), 0, bytes, idx, len);
-                idx += len;
+                byte[] obKVParamsBytes = obKVParams.encode();
+                System.arraycopy(obKVParamsBytes, 0, bytes, idx, obKVParamsBytes.length);
             } else {
-                len = HTABLE_DUMMY_BYTES.length;
-                System.arraycopy(HTABLE_DUMMY_BYTES, 0, bytes, idx, len);
-                idx += len;
+                System.arraycopy(HTABLE_DUMMY_BYTES, 0, bytes, idx, HTABLE_DUMMY_BYTES.length);
             }
         }
         return bytes;
diff --git a/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableTabletOp.java b/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableTabletOp.java
@@ -57,18 +57,18 @@ public byte[] encode() {
         idx += 8;
 
         // 2. encode option flag
-        int len = Serialization.getNeedBytes(optionFlag.getValue());
-        System.arraycopy(Serialization.encodeVi64(optionFlag.getValue()), 0, bytes, idx, len);
-        idx += len;
+        byte[] optionFlagLenBytes = Serialization.encodeVi64(optionFlag.getValue());
+        System.arraycopy(optionFlagLenBytes, 0, bytes, idx, optionFlagLenBytes.length);
+        idx += optionFlagLenBytes.length;
 
         // 4. encode Operation
-        len = Serialization.getNeedBytes(singleOperations.size());
-        System.arraycopy(Serialization.encodeVi64(singleOperations.size()), 0, bytes, idx, len);
-        idx += len;
+        byte[] singleOperationsLenBytes = Serialization.encodeVi64(singleOperations.size());
+        System.arraycopy(singleOperationsLenBytes, 0, bytes, idx, singleOperationsLenBytes.length);
+        idx += singleOperationsLenBytes.length;
         for (ObTableSingleOp singleOperation : singleOperations) {
-            len = (int) singleOperation.getPayloadSize();
-            System.arraycopy(singleOperation.encode(), 0, bytes, idx, len);
-            idx += len;
+            byte[] singleOperationLenBytes = singleOperation.encode();
+            System.arraycopy(singleOperationLenBytes, 0, bytes, idx, singleOperationLenBytes.length);
+            idx += singleOperationLenBytes.length;
         }
 
         return bytes;
diff --git a/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/query/ObHTableFilter.java b/src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/query/ObHTableFilter.java
@@ -61,42 +61,42 @@ public byte[] encode() {
         int idx = 0;
 
         // 0. encode header
-        int headerLen = (int) getObUniVersionHeaderLength(getVersion(), getPayloadContentSize());
-        System.arraycopy(encodeObUniVersionHeader(getVersion(), getPayloadContentSize()), 0, bytes,
-            idx, headerLen);
-        idx += headerLen;
+        byte[] headerBytes = encodeObUniVersionHeader(getVersion(), getPayloadContentSize());
+        System.arraycopy(headerBytes, 0, bytes,
+            idx, headerBytes.length);
+        idx += headerBytes.length;
 
         // 1. encode
         System.arraycopy(Serialization.encodeI8(isValid ? (byte) 1 : (byte) 0), 0, bytes, idx, 1);
         idx++;
-        int len = Serialization.getNeedBytes(selectColumnQualifier.size());
+        byte[] selectColumnQualifierBytes = Serialization.encodeVi64(selectColumnQualifier.size());
         System
-            .arraycopy(Serialization.encodeVi64(selectColumnQualifier.size()), 0, bytes, idx, len);
-        idx += len;
+            .arraycopy(selectColumnQualifierBytes, 0, bytes, idx, selectColumnQualifierBytes.length);
+        idx += selectColumnQualifierBytes.length;
         for (ObBytesString q : selectColumnQualifier) {
-            len = Serialization.getNeedBytes(q);
-            System.arraycopy(Serialization.encodeBytesString(q), 0, bytes, idx, len);
-            idx += len;
+            byte[] QualifierBytes = Serialization.encodeBytesString(q);
+            System.arraycopy(QualifierBytes, 0, bytes, idx, QualifierBytes.length);
+            idx += QualifierBytes.length;
         }
 
-        len = Serialization.getNeedBytes(minStamp);
-        System.arraycopy(Serialization.encodeVi64(minStamp), 0, bytes, idx, len);
-        idx += len;
-        len = Serialization.getNeedBytes(maxStamp);
-        System.arraycopy(Serialization.encodeVi64(maxStamp), 0, bytes, idx, len);
-        idx += len;
-        len = Serialization.getNeedBytes(maxVersions);
-        System.arraycopy(Serialization.encodeVi32(maxVersions), 0, bytes, idx, len);
-        idx += len;
-        len = Serialization.getNeedBytes(limitPerRowPerCf);
-        System.arraycopy(Serialization.encodeVi32(limitPerRowPerCf), 0, bytes, idx, len);
-        idx += len;
-        len = Serialization.getNeedBytes(offsetPerRowPerCf);
-        System.arraycopy(Serialization.encodeVi32(offsetPerRowPerCf), 0, bytes, idx, len);
-        idx += len;
-        len = Serialization.getNeedBytes(filterString);
-        System.arraycopy(Serialization.encodeBytesString(filterString), 0, bytes, idx, len);
-        idx += len;
+        byte[] minStampBytes = Serialization.encodeVi64(minStamp);
+        System.arraycopy(minStampBytes, 0, bytes, idx, minStampBytes.length);
+        idx += minStampBytes.length;
+        byte[] maxStampBytes = Serialization.encodeVi64(maxStamp);
+        System.arraycopy(maxStampBytes, 0, bytes, idx, maxStampBytes.length);
+        idx += maxStampBytes.length;
+        byte[] maxVersionsBytes = Serialization.encodeVi32(maxVersions);
+        System.arraycopy(maxVersionsBytes, 0, bytes, idx, maxVersionsBytes.length);
+        idx += maxVersionsBytes.length;
+        byte[] limitPerRowPerCfBytes = Serialization.encodeVi32(limitPerRowPerCf);
+        System.arraycopy(Serialization.encodeVi32(limitPerRowPerCf), 0, bytes, idx, limitPerRowPerCfBytes.length);
+        idx += limitPerRowPerCfBytes.length;
+        byte[] offsetPerRowPerCfBytes = Serialization.encodeVi32(offsetPerRowPerCf);
+        System.arraycopy(offsetPerRowPerCfBytes, 0, bytes, idx, offsetPerRowPerCfBytes.length);
+        idx += offsetPerRowPerCfBytes.length;
+        byte[] filterStringBytes = Serialization.encodeBytesString(filterString);
+        System.arraycopy(filterStringBytes, 0, bytes, idx, filterStringBytes.length);
+        idx += filterStringBytes.length;
 
         return bytes;
     }
diff --git a/src/main/java/com/alipay/oceanbase/rpc/table/ObHBaseParams.java b/src/main/java/com/alipay/oceanbase/rpc/table/ObHBaseParams.java
diff --git a/src/main/java/com/alipay/oceanbase/rpc/table/ObKVParams.java b/src/main/java/com/alipay/oceanbase/rpc/table/ObKVParams.java
diff --git a/src/main/java/com/alipay/oceanbase/rpc/util/Serialization.java b/src/main/java/com/alipay/oceanbase/rpc/util/Serialization.java
