refresh configuration for membership change (#652)

* refresh configuration for membership change * by CR
sofastack · Aug 10, 2021 · 17cbbd0 · 17cbbd0
1 parent 1ef3067
commit 17cbbd0
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/...ore/src/main/java/com/alipay/sofa/jraft/rhea/client/pd/AbstractPlacementDriverClient.java b/...ore/src/main/java/com/alipay/sofa/jraft/rhea/client/pd/AbstractPlacementDriverClient.java
@@ -18,6 +18,7 @@
 
 import java.util.List;
 import java.util.Map;
+import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
 
 import org.slf4j.Logger;
@@ -60,9 +61,12 @@
  */
 public abstract class AbstractPlacementDriverClient implements PlacementDriverClient {
 
-    private static final Logger         LOG              = LoggerFactory.getLogger(AbstractPlacementDriverClient.class);
+    private static final Logger         LOG                      = LoggerFactory
+                                                                     .getLogger(AbstractPlacementDriverClient.class);
 
-    protected final RegionRouteTable    regionRouteTable = new RegionRouteTable();
+    private static final long           AT_LEAST_REQUIRED_MILLIS = TimeUnit.SECONDS.toMillis(1);
+
+    protected final RegionRouteTable    regionRouteTable         = new RegionRouteTable();
     protected final long                clusterId;
     protected final String              clusterName;
 
@@ -277,6 +281,17 @@ protected PeerId getLeader(final String raftGroupId, final boolean forceRefresh,
                         : new RouteTableException(error.toString());
                 }
             }
+
+            // we need refresh configuration for membership change
+            final long leftTime = deadline - System.currentTimeMillis();
+            if (leftTime > AT_LEAST_REQUIRED_MILLIS) {
+                try {
+                    RouteTable.getInstance().refreshConfiguration(this.cliClientService, raftGroupId, (int) leftTime);
+                } catch (final InterruptedException e) {
+                    ThrowUtil.throwException(e);
+                } catch (final TimeoutException ignored) {
+                }
+            }
         }
         return routeTable.selectLeader(raftGroupId);
     }