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.Lists;
  14 import java.util.List;
  15 import org.opendaylight.controller.cluster.datastore.ShardManager;
  16 import org.opendaylight.controller.cluster.datastore.messages.SwitchShardBehavior;
  17 import org.opendaylight.controller.cluster.raft.RaftState;
  18 import org.opendaylight.controller.md.sal.common.util.jmx.AbstractMXBean;
  19 import org.slf4j.Logger;
  20 import org.slf4j.LoggerFactory;
  21 import org.opendaylight.controller.cluster.datastore.messages.AddShardReplica;
  22 import org.opendaylight.controller.cluster.datastore.messages.RemoveShardReplica;
  23 import akka.pattern.Patterns;
  24 import akka.util.Timeout;
  25 import java.util.concurrent.TimeUnit;
  26 import scala.concurrent.Await;
  27
  28 public class ShardManagerInfo extends AbstractMXBean implements ShardManagerInfoMBean {
  29
  30     public static String JMX_CATEGORY_SHARD_MANAGER = "ShardManager";
  31
  32     // The only states that you can switch to from outside. You cannot switch to Candidate/IsolatedLeader for example
  33     private static final List<String> ACCEPTABLE_STATES
  34             = Lists.newArrayList(RaftState.Leader.name(), RaftState.Follower.name());
  35
  36     private static final Logger LOG = LoggerFactory.getLogger(ShardManagerInfo.class);
  37
  38     private final String memberName;
  39     private final List<String> localShards;
  40
  41     private boolean syncStatus = false;
  42
  43     private ShardManager shardManager;
  44
  45     public ShardManagerInfo(String memberName, String name, String mxBeanType, List<String> localShards) {
  46         super(name, mxBeanType, JMX_CATEGORY_SHARD_MANAGER);
  47         this.memberName = memberName;
  48         this.localShards = localShards;
  49     }
  50
  51     public static ShardManagerInfo createShardManagerMBean(String memberName, String name, String mxBeanType,
  52             List<String> localShards){
  53         ShardManagerInfo shardManagerInfo = new ShardManagerInfo(memberName, name, mxBeanType, localShards);
  54
  55         shardManagerInfo.registerMBean();
  56
  57         return shardManagerInfo;
  58     }
  59
  60     public void addLocalShard(String shardName) {
  61         localShards.add(shardName);
  62     }
  63
  64     @Override
  65     public List<String> getLocalShards() {
  66         return localShards;
  67     }
  68
  69     @Override
  70     public boolean getSyncStatus() {
  71         return this.syncStatus;
  72     }
  73
  74     @Override
  75     public String getMemberName() {
  76         return memberName;
  77     }
  78
  79     @Override
  80     public void switchAllLocalShardsState(String newState, long term) {
  81         LOG.info("switchAllLocalShardsState called newState = {}, term = {}", newState, term);
  82
  83         for(String shardName : localShards){
  84             switchShardState(shardName, newState, term);
  85         }
  86     }
  87
  88     @Override
  89     public void switchShardState(String shardName, String newState, long term) {
  90         LOG.info("switchShardState called shardName = {}, newState = {}, term = {}", shardName, newState, term);
  91
  92         Preconditions.checkArgument(localShards.contains(shardName), shardName + " is not local");
  93         Preconditions.checkArgument(ACCEPTABLE_STATES.contains(newState));
  94
  95         shardManager.getSelf().tell(new SwitchShardBehavior(shardName, newState, term), ActorRef.noSender());
  96     }
  97
  98     public void setSyncStatus(boolean syncStatus){
  99         this.syncStatus = syncStatus;
 100     }
 101
 102     public void setShardManager(ShardManager shardManager){
 103         this.shardManager = shardManager;
 104     }
 105
 106     @Override
 107     public void setAddShardReplica (String shardName) {
 108         LOG.info ("addShardReplica initiated for shard {}", shardName);
 109
 110         // TODO addTimeout to be made configurable
 111         Timeout addTimeOut = new Timeout(1, TimeUnit.MINUTES);
 112         try {
 113             Await.result(Patterns.ask(shardManager.getSelf(),
 114                            new AddShardReplica(shardName), addTimeOut),
 115                            addTimeOut.duration());
 116         } catch (Exception ex) {
 117             LOG.debug ("Obtained an exception during addShardReplica", ex);
 118             throw (new RuntimeException(ex.getMessage()));
 119         }
 120         return;
 121     }
 122
 123     @Override
 124     public void setRemoveShardReplica (String shardName) {
 125         LOG.info ("removeShardReplica initiated for shard {}", shardName);
 126
 127         // TODO remTimeOut to be made configurable
 128         Timeout remTimeOut = new Timeout(30, TimeUnit.SECONDS);
 129         try {
 130             Await.result(Patterns.ask(shardManager.getSelf(),
 131                            new RemoveShardReplica(shardName), remTimeOut),
 132                            remTimeOut.duration());
 133         } catch (Exception ex) {
 134             LOG.debug ("Obtained an exception during removeShardReplica", ex);
 135             throw (new RuntimeException(ex.getMessage()));
 136         }
 137         return;
 138     }
 139 }