add ObFetchPartitionMetaType in request, pass re-fetch meta

Author: JackShi148

src/main/java/com/alipay/oceanbase/rpc/ObTableClient.java

@@ -2955,7 +2955,8 @@ private TableEntry getOrFetchODPPartitionMeta(String tableName, boolean needRene
             do {
                 try {
                     ObFetchPartitionMetaRequest request = ObFetchPartitionMetaRequest.getInstance(
-                        tableName, clusterName, tenantName, database, forceRenew,
+                            ObFetchPartitionMetaType.GET_PARTITION_META.getIndex(),
+                            tableName, clusterName, tenantName, database, forceRenew,
                         odpTable.getObTableOperationTimeout()); // TODO: timeout setting need to be verified
                     ObPayload result = odpTable.execute(request);
                     checkObFetchPartitionMetaResult(lastOdpRefreshTimeMills, request, result);

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

@@ -30,7 +30,6 @@
  *   header_,
  *   private String credential_,
  *   private String tableName_,
- *   private long createTime_,
  *   private String clusterName_,
  *   private String tenantName_,
  *   private String databaseName_,
@@ -39,7 +38,7 @@
  *
  */
 public class ObFetchPartitionMetaRequest extends AbstractPayload implements Credentialable {
-    private long          createTime = System.currentTimeMillis();
+    private ObFetchPartitionMetaType obFetchPartitionMetaType;
     private ObBytesString credential;
     private String        tableName;
     private String        clusterName;
@@ -52,8 +51,12 @@ public int getPcode() {
         return Pcodes.OB_TABLE_API_PART_META_QUERY;
     }
 
-    public long getCreateTime() {
-        return this.createTime;
+    public void setObFetchPartitionMetaType(ObFetchPartitionMetaType type) {
+        this.obFetchPartitionMetaType = type;
+    }
+
+    public void setObFetchPartitionMetaType(int index) {
+        this.obFetchPartitionMetaType = ObFetchPartitionMetaType.valueOf(index);
     }
 
     public void setClusterName(String clusterName) {
@@ -95,6 +98,11 @@ public byte[] encode() {
         // ver + plen + payload
         idx = encodeHeader(bytes, idx);
 
+        // encode type
+        int len = Serialization.getNeedBytes(obFetchPartitionMetaType.getIndex());
+        System.arraycopy(Serialization.encodeVi32(obFetchPartitionMetaType.getIndex()), 0, bytes, idx, len);
+        idx += len;
+
         // encode credential
         idx = encodeCredential(bytes, idx);
 
@@ -119,7 +127,7 @@ public byte[] encode() {
         idx += strbytes.length;
 
         // encode force_renew for ODP route
-        System.arraycopy(Serialization.encodeI8(forceRenew ? (byte) 1 : (byte) 0), 0, bytes, idx, 1);
+        System.arraycopy(Serialization.encodeI8(forceRenew ? ((byte) 1) : ((byte) 0)), 0, bytes, idx, 1);
 
         return bytes;
     }
@@ -140,7 +148,8 @@ public Object decode(ByteBuf buf) {
 
     @Override
     public long getPayloadContentSize() {
-        return Serialization.getNeedBytes(credential)
+        return Serialization.getNeedBytes(obFetchPartitionMetaType.getIndex())
+                    + Serialization.getNeedBytes(credential)
                     + Serialization.getNeedBytes(tableName)
                     + Serialization.getNeedBytes(clusterName)
                     + Serialization.getNeedBytes(tenantName)
@@ -154,10 +163,11 @@ private int encodeCredential(byte[] bytes, int idx) {
         return idx;
     }
 
-    public static ObFetchPartitionMetaRequest getInstance(String tableName, String clusterName,
+    public static ObFetchPartitionMetaRequest getInstance(int typeIdx, String tableName, String clusterName,
                                                           String tenantName, String databaseName,
                                                           boolean forceRenew, long timeout) {
         ObFetchPartitionMetaRequest request = new ObFetchPartitionMetaRequest();
+        request.setObFetchPartitionMetaType(typeIdx);
         request.setTableName(tableName);
         if (clusterName == null) {
             clusterName = "";

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

@@ -0,0 +1,39 @@
+package com.alipay.oceanbase.rpc.protocol.payload.impl.execute;
+
+public enum ObFetchPartitionMetaType {
+    GET_PARTITION_META("GET_PARTITION_META", 0), INVALID("INVALID", 1);
+
+    private final String name;
+    private final int   index;
+
+    ObFetchPartitionMetaType(String name, int index) {
+        this.name = name;
+        this.index = index;
+    }
+
+    /*
+     * Value of.
+     */
+    public static ObFetchPartitionMetaType valueOf(int index) {
+        if (GET_PARTITION_META.index == index) {
+            return GET_PARTITION_META;
+        } else {
+            return INVALID;
+        }
+    }
+
+    /*
+     * Get index.
+     */
+    public int getIndex() {
+        return index;
+    }
+
+    /*
+     * Get name.
+     */
+    public String getName() {
+        return name;
+    }
+
+}