X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-akka-raft%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fcluster%2Fraft%2FDefaultConfigParamsImpl.java;h=c8cc88c3edd45256011f050a5a4c83b749b37730;hb=f3c38988cc31e07bab473cace946aebf5152c61f;hp=d4d13899eb770e4ee8bb88a634dd85ec2cfb7bbc;hpb=f78b7f15e24efdb5dd9f91b487bc63dad7517b1c;p=controller.git

diff --git a/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/DefaultConfigParamsImpl.java b/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/DefaultConfigParamsImpl.java
index d4d13899eb..c8cc88c3ed 100644
--- a/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/DefaultConfigParamsImpl.java
+++ b/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/DefaultConfigParamsImpl.java
@@ -7,7 +7,14 @@
  */
 package org.opendaylight.controller.cluster.raft;
 
+import com.google.common.base.Strings;
+import com.google.common.base.Supplier;
+import com.google.common.base.Suppliers;
 import java.util.concurrent.TimeUnit;
+import org.opendaylight.controller.cluster.raft.policy.DefaultRaftPolicy;
+import org.opendaylight.controller.cluster.raft.policy.RaftPolicy;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import scala.concurrent.duration.FiniteDuration;
 
 /**
@@ -17,6 +24,8 @@ import scala.concurrent.duration.FiniteDuration;
  */
 public class DefaultConfigParamsImpl implements ConfigParams {
 
+    private static final Logger LOG = LoggerFactory.getLogger(DefaultConfigParamsImpl.class);
+
     private static final int SNAPSHOT_BATCH_COUNT = 20000;
 
     private static final int JOURNAL_RECOVERY_LOG_BATCH_SIZE = 1000;
@@ -49,7 +58,12 @@ public class DefaultConfigParamsImpl implements ConfigParams {
     // in-memory journal can use before it needs to snapshot
     private int snapshotDataThresholdPercentage = 12;
 
+    private int snapshotChunkSize = SNAPSHOT_CHUNK_SIZE;
+
     private long electionTimeoutFactor = 2;
+    private String customRaftPolicyImplementationClass;
+
+    Supplier<RaftPolicy> policySupplier = Suppliers.memoize(new PolicySupplier());
 
     public void setHeartBeatInterval(FiniteDuration heartBeatInterval) {
         this.heartBeatInterval = heartBeatInterval;
@@ -64,6 +78,10 @@ public class DefaultConfigParamsImpl implements ConfigParams {
         this.snapshotDataThresholdPercentage = snapshotDataThresholdPercentage;
     }
 
+    public void setSnapshotChunkSize(int snapshotChunkSize) {
+        this.snapshotChunkSize = snapshotChunkSize;
+    }
+
     public void setJournalRecoveryLogBatchSize(int journalRecoveryLogBatchSize) {
         this.journalRecoveryLogBatchSize = journalRecoveryLogBatchSize;
     }
@@ -77,6 +95,10 @@ public class DefaultConfigParamsImpl implements ConfigParams {
         electionTimeOutInterval = null;
     }
 
+    public void setCustomRaftPolicyImplementationClass(String customRaftPolicyImplementationClass){
+        this.customRaftPolicyImplementationClass = customRaftPolicyImplementationClass;
+    }
+
     @Override
     public long getSnapshotBatchCount() {
         return snapshotBatchCount;
@@ -109,7 +131,7 @@ public class DefaultConfigParamsImpl implements ConfigParams {
 
     @Override
     public int getSnapshotChunkSize() {
-        return SNAPSHOT_CHUNK_SIZE;
+        return snapshotChunkSize;
     }
 
     @Override
@@ -126,4 +148,33 @@ public class DefaultConfigParamsImpl implements ConfigParams {
     public long getElectionTimeoutFactor() {
         return electionTimeoutFactor;
     }
+
+    @Override
+    public RaftPolicy getRaftPolicy() {
+        return policySupplier.get();
+    }
+
+    private class PolicySupplier implements Supplier<RaftPolicy>{
+        @Override
+        public RaftPolicy get() {
+            if(Strings.isNullOrEmpty(DefaultConfigParamsImpl.this.customRaftPolicyImplementationClass)){
+                LOG.debug("No custom RaftPolicy specified. Using DefaultRaftPolicy");
+                return DefaultRaftPolicy.INSTANCE;
+            }
+            try {
+                String className = DefaultConfigParamsImpl.this.customRaftPolicyImplementationClass;
+                LOG.info("Trying to use custom RaftPolicy {}", className);
+                Class c = Class.forName(className);
+                RaftPolicy obj = (RaftPolicy)c.newInstance();
+                return obj;
+            } catch (Exception e) {
+                if(LOG.isDebugEnabled()) {
+                    LOG.error("Could not create custom raft policy, will stick with default", e);
+                } else {
+                    LOG.error("Could not create custom raft policy, will stick with default : cause = {}", e.getMessage());
+                }
+            }
+            return DefaultRaftPolicy.INSTANCE;
+        }
+    }
 }