Add bolt reconnection parameter: jraft.bolt.conn.reconnect (sofastack…

…#1136) * add param "jraft.bolt.conn.reconnect" * add param "jraft.bolt.conn.reconnect" * refactor --------- Co-authored-by: yuanyuan.liu <yuanyuan.liu@signalplus.com>
shihuili1218 · Aug 7, 2024 · 79df847 · 79df847
1 parent 4d53aa5
commit 79df847
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/jraft-core/src/main/java/com/alipay/sofa/jraft/option/RpcOptions.java b/jraft-core/src/main/java/com/alipay/sofa/jraft/option/RpcOptions.java
@@ -16,6 +16,7 @@
  */
 package com.alipay.sofa.jraft.option;
 
+import com.alipay.sofa.jraft.util.SystemPropertyUtil;
 import com.alipay.sofa.jraft.util.concurrent.FixedThreadsExecutorGroup;
 import com.codahale.metrics.MetricRegistry;
 
@@ -62,6 +63,12 @@ public class RpcOptions {
      */
     private FixedThreadsExecutorGroup appendEntriesExecutors;
 
+    /**
+     * Whether to enable bolt reconnection, default enabled(true).
+     */
+    private boolean                   enableBoltReconnect        = SystemPropertyUtil.getBoolean(
+                                                                     "jraft.bolt.conn.reconnect", true);
+
     public int getRpcConnectTimeoutMs() {
         return this.rpcConnectTimeoutMs;
     }
@@ -118,11 +125,20 @@ public void setAppendEntriesExecutors(FixedThreadsExecutorGroup appendEntriesExe
         this.appendEntriesExecutors = appendEntriesExecutors;
     }
 
+    public boolean isEnableBoltReconnect() {
+        return enableBoltReconnect;
+    }
+
+    public void setEnableBoltReconnect(boolean enableBoltReconnect) {
+        this.enableBoltReconnect = enableBoltReconnect;
+    }
+
     @Override
     public String toString() {
         return "RpcOptions{" + "rpcConnectTimeoutMs=" + rpcConnectTimeoutMs + ", rpcDefaultTimeout="
                + rpcDefaultTimeout + ", rpcInstallSnapshotTimeout=" + rpcInstallSnapshotTimeout
                + ", rpcProcessorThreadPoolSize=" + rpcProcessorThreadPoolSize + ", enableRpcChecksum="
-               + enableRpcChecksum + ", metricRegistry=" + metricRegistry + '}';
+               + enableRpcChecksum + ", metricRegistry=" + metricRegistry + ", enableBoltReconnect="
+               + enableBoltReconnect + '}';
     }
 }
diff --git a/jraft-core/src/main/java/com/alipay/sofa/jraft/rpc/impl/BoltRpcClient.java b/jraft-core/src/main/java/com/alipay/sofa/jraft/rpc/impl/BoltRpcClient.java
@@ -57,7 +57,7 @@ public boolean init(final RpcOptions opts) {
         this.rpcClient.option(BoltClientOption.NETTY_FLUSH_CONSOLIDATION, true);
         this.rpcClient.initWriteBufferWaterMark(BoltRaftRpcFactory.CHANNEL_WRITE_BUF_LOW_WATER_MARK,
             BoltRaftRpcFactory.CHANNEL_WRITE_BUF_HIGH_WATER_MARK);
-        this.rpcClient.enableReconnectSwitch();
+        this.rpcClient.option(BoltClientOption.CONN_RECONNECT_SWITCH, opts.isEnableBoltReconnect());
         this.rpcClient.startup();
         return true;
     }