X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?p=controller.git;a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-distributed-datastore%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fcluster%2Fdatastore%2Fjmx%2Fmbeans%2Fshardmanager%2FShardManagerInfo.java;h=aaac644b558b3e1913facbd01bc5a07536bf30ee;hp=8adc8b24b27c1ad4d3eff65730cf9d27646fd3d0;hb=a88392383dd0f836de1a8f70cfce85ef5d7fd91f;hpb=c68d251880d95d6d2f8df70c67d2cdd3a3a47685

diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/jmx/mbeans/shardmanager/ShardManagerInfo.java b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/jmx/mbeans/shardmanager/ShardManagerInfo.java
index 8adc8b24b2..aaac644b55 100644
--- a/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/jmx/mbeans/shardmanager/ShardManagerInfo.java
+++ b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/jmx/mbeans/shardmanager/ShardManagerInfo.java
@@ -8,18 +8,33 @@
 
 package org.opendaylight.controller.cluster.datastore.jmx.mbeans.shardmanager;
 
+import akka.actor.ActorRef;
+import com.google.common.base.Preconditions;
+import com.google.common.collect.Lists;
 import java.util.List;
-
+import org.opendaylight.controller.cluster.datastore.ShardManager;
+import org.opendaylight.controller.cluster.datastore.messages.SwitchShardBehavior;
+import org.opendaylight.controller.cluster.raft.RaftState;
 import org.opendaylight.controller.md.sal.common.util.jmx.AbstractMXBean;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 public class ShardManagerInfo extends AbstractMXBean implements ShardManagerInfoMBean {
 
     public static String JMX_CATEGORY_SHARD_MANAGER = "ShardManager";
 
+    // The only states that you can switch to from outside. You cannot switch to Candidate/IsolatedLeader for example
+    private static final List<String> ACCEPTABLE_STATES
+            = Lists.newArrayList(RaftState.Leader.name(), RaftState.Follower.name());
+
+    private static final Logger LOG = LoggerFactory.getLogger(ShardManagerInfo.class);
+
     private final List<String> localShards;
 
     private boolean syncStatus = false;
 
+    private ShardManager shardManager;
+
     public ShardManagerInfo(String name, String mxBeanType, List<String> localShards) {
         super(name, mxBeanType, JMX_CATEGORY_SHARD_MANAGER);
         this.localShards = localShards;
@@ -44,7 +59,30 @@ public class ShardManagerInfo extends AbstractMXBean implements ShardManagerInfo
         return this.syncStatus;
     }
 
+    @Override
+    public void switchAllLocalShardsState(String newState, long term) {
+        LOG.info("switchAllLocalShardsState called newState = {}, term = {}", newState, term);
+
+        for(String shardName : localShards){
+            switchShardState(shardName, newState, term);
+        }
+    }
+
+    @Override
+    public void switchShardState(String shardName, String newState, long term) {
+        LOG.info("switchShardState called shardName = {}, newState = {}, term = {}", shardName, newState, term);
+
+        Preconditions.checkArgument(localShards.contains(shardName), shardName + " is not local");
+        Preconditions.checkArgument(ACCEPTABLE_STATES.contains(newState));
+
+        shardManager.getSelf().tell(new SwitchShardBehavior(shardName, newState, term), ActorRef.noSender());
+    }
+
     public void setSyncStatus(boolean syncStatus){
         this.syncStatus = syncStatus;
     }
+
+    public void setShardManager(ShardManager shardManager){
+        this.shardManager = shardManager;
+    }
 }