refresh tablet location to refresh leader observer when transport timeout

Author: JackShi148

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

@@ -572,6 +572,10 @@ private <T> T execute(String tableName, TableExecuteCallback<T> callback, ObServ
                         if (ex instanceof ObTableTransportException &&
                                 ((ObTableTransportException) ex).getErrorCode() == TransportCodes.BOLT_TIMEOUT) {
                             syncRefreshMetadata(true);
+                            TableEntry entry = tableRoute.getTableEntry(tableName);
+                            long partId = tableRoute.getPartId(entry, rowKey);
+                            long tabletId = tableRoute.getTabletIdByPartId(entry, partId);
+                            tableRoute.refreshPartitionLocation(tableName, tabletId, entry);
                         }
                         String logMessage;
                         if (ex instanceof ObTableException) {
@@ -786,6 +790,10 @@ private <T> T execute(String tableName, OperationExecuteCallback<T> callback,
                         if (ex instanceof ObTableTransportException &&
                                 ((ObTableTransportException) ex).getErrorCode() == TransportCodes.BOLT_TIMEOUT) {
                             syncRefreshMetadata(true);
+                            TableEntry entry = tableRoute.getTableEntry(tableName);
+                            long partId = tableRoute.getPartId(entry, callback.getRowKey());
+                            long tabletId = tableRoute.getTabletIdByPartId(entry, partId);
+                            tableRoute.refreshPartitionLocation(tableName, tabletId, entry);
                         }
                         String logMessage;
                         if (ex instanceof ObTableException) {

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

@@ -291,6 +291,8 @@ protected ObPayload commonExecute(ObTableClient client, Logger logger,
                             if (e instanceof ObTableTransportException
                                 && ((ObTableTransportException) e).getErrorCode() == TransportCodes.BOLT_TIMEOUT) {
                                 client.syncRefreshMetadata(true);
+                                long tabletId = partIdWithIndex.getRight().getTabletId();
+                                client.refreshTableLocationByTabletId(indexTableName, tabletId);
                             }
                             client.calculateContinuousFailure(indexTableName, e.getMessage());
                             throw e;

src/main/java/com/alipay/oceanbase/rpc/table/ObTableClientBatchOpsImpl.java

@@ -455,6 +455,7 @@ public void partitionExecute(ObTableOperationResult[] results,
                         if (ex instanceof ObTableTransportException
                             && ((ObTableTransportException) ex).getErrorCode() == TransportCodes.BOLT_TIMEOUT) {
                             obTableClient.syncRefreshMetadata(true);
+                            obTableClient.refreshTableLocationByTabletId(tableName, partId);
                         }
                         obTableClient.calculateContinuousFailure(tableName, ex.getMessage());
                         throw ex;

src/main/java/com/alipay/oceanbase/rpc/table/ObTableClientLSBatchOpsImpl.java

@@ -676,7 +676,7 @@ public void partitionExecute(ObTableSingleOpResult[] results,
                     }
                 } else if (ex instanceof ObTableException) {
                     if (((ObTableException) ex).isNeedRefreshTableEntry()) {
-                        logger.warn("meet need refresh exception, errCode: {}, ls id: {}", ((ObTableException) ex).getErrorCode(), lsId);
+                        logger.warn("meet need refresh exception, errCode: {}, ls id: {}, errMsg: {}", ((ObTableException) ex).getErrorCode(), lsId, ex.getMessage());
                         if ((((ObTableException) ex).getErrorCode() == ResultCodes.OB_TABLE_NOT_EXIST.errorCode ||
                                 ((ObTableException) ex).getErrorCode() == ResultCodes.OB_SCHEMA_ERROR.errorCode)
                                 && obTableClient.isTableGroupName(tableName)
@@ -731,6 +731,7 @@ public void partitionExecute(ObTableSingleOpResult[] results,
                         if (ex instanceof ObTableTransportException &&
                                 ((ObTableTransportException) ex).getErrorCode() == TransportCodes.BOLT_TIMEOUT) {
                             obTableClient.syncRefreshMetadata(true);
+                            obTableClient.refreshTabletLocationBatch(realTableName);
                         }
                         obTableClient.calculateContinuousFailure(realTableName, ex.getMessage());
                         throw ex;