Bug 2187: EOS shard recovery after AddShardReplica

[controller.git] / opendaylight / md-sal / sal-akka-raft / src / main / java / org / opendaylight / controller / cluster / raft / client / messages / OnDemandRaftState.java

diff --git a/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/client/messages/OnDemandRaftState.java b/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/client/messages/OnDemandRaftState.java
index 8c2986f6d19f402137d40efd551667b22a28c151..57f8beb0054118e8a26549c55dac770e8ee10593 100644 (file)
--- a/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/client/messages/OnDemandRaftState.java
+++ b/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/client/messages/OnDemandRaftState.java
@@ -33,6 +33,7 @@ public class OnDemandRaftState {
     private String raftState;
     private String votedFor;
     private boolean isSnapshotCaptureInitiated;
+    private String customRaftPolicyClassName;
 
     private List<FollowerInfo> followerInfoList = Collections.emptyList();
     private Map<String, String> peerAddresses = Collections.emptyMap();
@@ -116,6 +117,10 @@ public class OnDemandRaftState {
         return peerAddresses;
     }
 
+    public String getCustomRaftPolicyClassName() {
+        return customRaftPolicyClassName;
+    }
+
     public static class Builder {
         private final OnDemandRaftState stats = new OnDemandRaftState();
 
@@ -209,6 +214,11 @@ public class OnDemandRaftState {
             return this;
         }
 
+        public Builder customRaftPolicyClassName(String className) {
+            stats.customRaftPolicyClassName = className;
+            return this;
+        }
+
         public OnDemandRaftState build() {
             return stats;
         }