2 * Copyright (c) 2014 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.controller.cluster.datastore.jmx.mbeans.shardmanager;
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;
23 public class ShardManagerInfo extends AbstractMXBean implements ShardManagerInfoMBean {
25 public static final String JMX_CATEGORY_SHARD_MANAGER = "ShardManager";
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());
31 private static final Logger LOG = LoggerFactory.getLogger(ShardManagerInfo.class);
33 private final String memberName;
34 private final List<String> localShards;
36 private boolean syncStatus = false;
38 private ShardManager shardManager;
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;
46 public static ShardManagerInfo createShardManagerMBean(String memberName, String name, String mxBeanType,
47 List<String> localShards){
48 ShardManagerInfo shardManagerInfo = new ShardManagerInfo(memberName, name, mxBeanType, localShards);
50 shardManagerInfo.registerMBean();
52 return shardManagerInfo;
55 public void addLocalShard(String shardName) {
56 localShards.add(shardName);
60 public List<String> getLocalShards() {
65 public boolean getSyncStatus() {
66 return this.syncStatus;
70 public String getMemberName() {
75 public void switchAllLocalShardsState(String newState, long term) {
76 LOG.info("switchAllLocalShardsState called newState = {}, term = {}", newState, term);
78 for(String shardName : localShards){
79 switchShardState(shardName, newState, term);
84 public void switchShardState(String shardName, String newState, long term) {
85 LOG.info("switchShardState called shardName = {}, newState = {}, term = {}", shardName, newState, term);
87 Preconditions.checkArgument(localShards.contains(shardName), shardName + " is not local");
88 Preconditions.checkArgument(ACCEPTABLE_STATES.contains(newState));
90 shardManager.getSelf().tell(new SwitchShardBehavior(shardName, RaftState.valueOf(newState), term),
94 public void setSyncStatus(boolean syncStatus){
95 this.syncStatus = syncStatus;
98 public void setShardManager(ShardManager shardManager){
99 this.shardManager = shardManager;