fix lsop encode error

Author: GroundWu

src/main/java/com/alipay/oceanbase/rpc/protocol/payload/AbstractPayload.java

@@ -240,4 +240,13 @@ protected byte[] encodeDefaultBytes() {
 
     protected boolean isUseDefaultEncode() { return false; }
 
+    /**
+     * Reset the cached payload content size to force recalculation
+     * Subclasses should override this method to reset their own cached sizes
+     * and call this method on their child objects
+     */
+    public void resetPayloadContentSize() {
+        this.payLoadContentSize = INVALID_PAYLOAD_CONTENT_SIZE;
+    }
+
 }

src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableLSOpRequest.java

@@ -160,4 +160,15 @@ public void setCredential(ObBytesString credential) {
     public void setTableId(long tableId) {
         this.lsOperation.setTableId(tableId);
     }
+
+    /**
+     * Reset the cached payload content size and propagate to child objects
+     */
+    @Override
+    public void resetPayloadContentSize() {
+        super.resetPayloadContentSize();
+        if (lsOperation != null) {
+            lsOperation.resetPayloadContentSize();
+        }
+    }
 }

src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableLSOperation.java

@@ -230,6 +230,19 @@ public long getPayloadContentSize() {
 
     }
 
+    /**
+     * Reset the cached payload content size and propagate to child objects
+     */
+    @Override
+    public void resetPayloadContentSize() {
+        super.resetPayloadContentSize();
+        for (ObTableTabletOp operation : tabletOperations) {
+            if (operation != null) {
+                operation.resetPayloadContentSize();
+            }
+        }
+    }
+
     /*
      * Get table operations.
      */
@@ -372,6 +385,7 @@ public void prepareColumnNamesBitMap() {
     }
 
     public void prepare() {
+        this.resetPayloadContentSize(); // to avoid use wrong cached when do retry
         this.collectColumnNamesIdxMap();
         this.beforeOption();
         this.prepareOption();

src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableSingleOp.java

@@ -143,6 +143,22 @@ public long getPayloadContentSize() {
         return this.payLoadContentSize;
     }
 
+    /**
+     * Reset the cached payload content size and propagate to child objects
+     */
+    @Override
+    public void resetPayloadContentSize() {
+        super.resetPayloadContentSize();
+        if (query != null) {
+            query.resetPayloadContentSize();
+        }
+        for (ObTableSingleOpEntity entity : entities) {
+            if (entity != null) {
+                entity.resetPayloadContentSize();
+            }
+        }
+    }
+
     public List<ObNewRange> getScanRange() {
         return query.getScanRanges();
     }

src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableSingleOpEntity.java

@@ -224,6 +224,15 @@ public long getPayloadContentSize() {
         return this.payLoadContentSize;
     }
 
+    /**
+     * Reset the cached payload content size
+     * This class has no child objects, so just call super
+     */
+    @Override
+    public void resetPayloadContentSize() {
+        super.resetPayloadContentSize();
+    }
+
     public static boolean areArraysSameLengthOrBothNull(Object[] a, Object[] b) {
         if (a == null && b == null) {
             return true;

src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableSingleOpQuery.java

@@ -239,6 +239,20 @@ public long getPayloadContentSize() {
         return this.payLoadContentSize;
     }
 
+    /**
+     * Reset the cached payload content size and propagate to child objects
+     */
+    @Override
+    public void resetPayloadContentSize() {
+        super.resetPayloadContentSize();
+        if (hTableFilter != null) {
+            hTableFilter.resetPayloadContentSize();
+        }
+        if (obKVParams != null) {
+            obKVParams.resetPayloadContentSize();
+        }
+    }
+
     // Support class, which is used for column name sorted
     private static class ColumnNamePair implements Comparable<ColumnNamePair> {
         long number;

src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/execute/ObTableTabletOp.java

@@ -135,6 +135,19 @@ public long getPayloadContentSize() {
         return this.payLoadContentSize;
     }
 
+    /**
+     * Reset the cached payload content size and propagate to child objects
+     */
+    @Override
+    public void resetPayloadContentSize() {
+        super.resetPayloadContentSize();
+        for (ObTableSingleOp operation : singleOperations) {
+            if (operation != null) {
+                operation.resetPayloadContentSize();
+            }
+        }
+    }
+
     /*
      * Get table operations.
      */