opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/jmx/mbeans/shardmanager/ShardManagerInfo.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.controller.cluster.datastore.jmx.mbeans.shardmanager;
  10
  11 import akka.actor.ActorRef;
  12 import com.google.common.base.Preconditions;
  13 import com.google.common.collect.ImmutableList;
  14 import java.util.Collection;
  15 import java.util.List;
  16 import org.opendaylight.controller.cluster.datastore.ShardManager;
  17 import org.opendaylight.controller.cluster.datastore.messages.SwitchShardBehavior;
  18 import org.opendaylight.controller.cluster.raft.RaftState;
  19 import org.opendaylight.controller.md.sal.common.util.jmx.AbstractMXBean;
  20 import org.slf4j.Logger;
  21 import org.slf4j.LoggerFactory;
  22
  23 public class ShardManagerInfo extends AbstractMXBean implements ShardManagerInfoMBean {
  24
  25     public static final String JMX_CATEGORY_SHARD_MANAGER = "ShardManager";
  26
  27     // The only states that you can switch to from outside. You cannot switch to Candidate/IsolatedLeader for example
  28     private static final Collection<String> ACCEPTABLE_STATES
  29             = ImmutableList.of(RaftState.Leader.name(), RaftState.Follower.name());
  30
  31     private static final Logger LOG = LoggerFactory.getLogger(ShardManagerInfo.class);
  32
  33     private final String memberName;
  34     private final List<String> localShards;
  35
  36     private boolean syncStatus = false;
  37
  38     private ShardManager shardManager;
  39
  40     public ShardManagerInfo(String memberName, String name, String mxBeanType, List<String> localShards) {
  41         super(name, mxBeanType, JMX_CATEGORY_SHARD_MANAGER);
  42         this.memberName = memberName;
  43         this.localShards = localShards;
  44     }
  45
  46     public static ShardManagerInfo createShardManagerMBean(String memberName, String name, String mxBeanType,
  47             List<String> localShards){
  48         ShardManagerInfo shardManagerInfo = new ShardManagerInfo(memberName, name, mxBeanType, localShards);
  49
  50         shardManagerInfo.registerMBean();
  51
  52         return shardManagerInfo;
  53     }
  54
  55     public void addLocalShard(String shardName) {
  56         localShards.add(shardName);
  57     }
  58
  59     @Override
  60     public List<String> getLocalShards() {
  61         return localShards;
  62     }
  63
  64     @Override
  65     public boolean getSyncStatus() {
  66         return this.syncStatus;
  67     }
  68
  69     @Override
  70     public String getMemberName() {
  71         return memberName;
  72     }
  73
  74     @Override
  75     public void switchAllLocalShardsState(String newState, long term) {
  76         LOG.info("switchAllLocalShardsState called newState = {}, term = {}", newState, term);
  77
  78         for(String shardName : localShards){
  79             switchShardState(shardName, newState, term);
  80         }
  81     }
  82
  83     @Override
  84     public void switchShardState(String shardName, String newState, long term) {
  85         LOG.info("switchShardState called shardName = {}, newState = {}, term = {}", shardName, newState, term);
  86
  87         Preconditions.checkArgument(localShards.contains(shardName), shardName + " is not local");
  88         Preconditions.checkArgument(ACCEPTABLE_STATES.contains(newState));
  89
  90         shardManager.getSelf().tell(new SwitchShardBehavior(shardName, newState, term), ActorRef.noSender());
  91     }
  92
  93     public void setSyncStatus(boolean syncStatus){
  94         this.syncStatus = syncStatus;
  95     }
  96
  97     public void setShardManager(ShardManager shardManager){
  98         this.shardManager = shardManager;
  99     }
 100 }