import akka.actor.ActorRef;
import akka.pattern.Patterns;
import akka.util.Timeout;
+import com.google.common.base.Joiner;
+import com.google.common.base.Joiner.MapJoiner;
import com.google.common.base.Stopwatch;
import com.google.common.cache.Cache;
import com.google.common.cache.CacheBuilder;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
-import java.util.Map;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicLong;
+import javax.annotation.Nullable;
+import org.opendaylight.controller.cluster.datastore.Shard;
+import org.opendaylight.controller.cluster.raft.base.messages.InitiateCaptureSnapshot;
import org.opendaylight.controller.cluster.raft.client.messages.FollowerInfo;
import org.opendaylight.controller.cluster.raft.client.messages.GetOnDemandRaftState;
import org.opendaylight.controller.cluster.raft.client.messages.OnDemandRaftState;
import org.opendaylight.controller.md.sal.common.util.jmx.AbstractMXBean;
-import org.opendaylight.controller.md.sal.common.util.jmx.QueuedNotificationManagerMXBeanImpl;
-import org.opendaylight.controller.md.sal.common.util.jmx.ThreadExecutorStatsMXBeanImpl;
-import org.opendaylight.controller.md.sal.dom.store.impl.InMemoryDOMDataStore;
-import org.opendaylight.yangtools.util.concurrent.QueuedNotificationManager;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
import scala.concurrent.Await;
/**
* @author Basheeruddin syedbahm@cisco.com
*/
public class ShardStats extends AbstractMXBean implements ShardStatsMXBean {
- public static String JMX_CATEGORY_SHARD = "Shards";
-
- private static final Logger LOG = LoggerFactory.getLogger(ShardStats.class);
+ public static final String JMX_CATEGORY_SHARD = "Shards";
private static final SimpleDateFormat DATE_FORMAT = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS");
- private static final Cache<String, OnDemandRaftState> onDemandRaftStateCache =
+ private static final Cache<String, OnDemandRaftState> ONDEMAND_RAFT_STATE_CACHE =
CacheBuilder.newBuilder().expireAfterWrite(2, TimeUnit.SECONDS).build();
+ private static final MapJoiner MAP_JOINER = Joiner.on(", ").withKeyValueSeparator(": ");
+
private long committedTransactionsCount;
private long readOnlyTransactionCount;
private long abortTransactionsCount;
- private ThreadExecutorStatsMXBeanImpl notificationExecutorStatsBean;
-
- private QueuedNotificationManagerMXBeanImpl notificationManagerStatsBean;
-
private boolean followerInitialSyncStatus = false;
- private ActorRef shardActor;
+ private final Shard shard;
private String statRetrievalError;
private long lastLeadershipChangeTime;
- public ShardStats(final String shardName, final String mxBeanType) {
+ public ShardStats(final String shardName, final String mxBeanType, @Nullable final Shard shard) {
super(shardName, mxBeanType, JMX_CATEGORY_SHARD);
+ this.shard = shard;
}
- public void setNotificationManager(final QueuedNotificationManager<?, ?> manager) {
- this.notificationManagerStatsBean = new QueuedNotificationManagerMXBeanImpl(manager,
- "notification-manager", getMBeanType(), getMBeanCategory());
-
- this.notificationExecutorStatsBean = ThreadExecutorStatsMXBeanImpl.create(manager.getExecutor());
- }
-
- public void setShardActor(ActorRef shardActor) {
- this.shardActor = shardActor;
- }
-
+ @SuppressWarnings("checkstyle:IllegalCatch")
private OnDemandRaftState getOnDemandRaftState() {
String name = getShardName();
- OnDemandRaftState state = onDemandRaftStateCache.getIfPresent(name);
- if(state == null) {
+ OnDemandRaftState state = ONDEMAND_RAFT_STATE_CACHE.getIfPresent(name);
+ if (state == null) {
statRetrievalError = null;
statRetrievalTime = null;
- if(shardActor != null) {
+ if (shard != null) {
Timeout timeout = new Timeout(10, TimeUnit.SECONDS);
try {
Stopwatch timer = Stopwatch.createStarted();
- state = (OnDemandRaftState) Await.result(Patterns.ask(shardActor,
+ state = (OnDemandRaftState) Await.result(Patterns.ask(shard.getSelf(),
GetOnDemandRaftState.INSTANCE, timeout), timeout.duration());
statRetrievalTime = timer.stop().toString();
- onDemandRaftStateCache.put(name, state);
+ ONDEMAND_RAFT_STATE_CACHE.put(name, state);
} catch (Exception e) {
statRetrievalError = e.toString();
}
return getOnDemandRaftState().getVotedFor();
}
+ @Override
+ public boolean isVoting() {
+ return getOnDemandRaftState().isVoting();
+ }
+
+ @Override
+ public String getPeerVotingStates() {
+ return MAP_JOINER.join(getOnDemandRaftState().getPeerVotingStates());
+ }
+
@Override
public boolean isSnapshotCaptureInitiated() {
return getOnDemandRaftState().isSnapshotCaptureInitiated();
@Override
public String getLastCommittedTransactionTime() {
- return DATE_FORMAT.format(new Date(lastCommittedTransactionTime));
+ synchronized (DATE_FORMAT) {
+ return DATE_FORMAT.format(new Date(lastCommittedTransactionTime));
+ }
}
@Override
return failedReadTransactionsCount.incrementAndGet();
}
- public long incrementAbortTransactionsCount ()
- {
+ public long incrementAbortTransactionsCount() {
return ++abortTransactionsCount;
}
}
@Override
- public long getInMemoryJournalDataSize(){
+ public long getInMemoryJournalDataSize() {
return getOnDemandRaftState().getInMemoryJournalDataSize();
}
}
/**
- * resets the counters related to transactions
+ * Resets the counters related to transactions.
*/
@Override
- public void resetTransactionCounters(){
+ public void resetTransactionCounters() {
committedTransactionsCount = 0;
readOnlyTransactionCount = 0;
}
- public void setDataStore(final InMemoryDOMDataStore store) {
- setNotificationManager(store.getDataChangeListenerNotificationManager());
- }
-
public void setFollowerInitialSyncStatus(boolean followerInitialSyncStatus) {
this.followerInitialSyncStatus = followerInitialSyncStatus;
}
@Override
public String getPeerAddresses() {
- StringBuilder builder = new StringBuilder();
- int i = 0;
- for(Map.Entry<String, String> e: getOnDemandRaftState().getPeerAddresses().entrySet()) {
- if(i++ > 0) {
- builder.append(", ");
- }
-
- builder.append(e.getKey()).append(": ").append(e.getValue());
- }
-
- return builder.toString();
+ return MAP_JOINER.join(getOnDemandRaftState().getPeerAddresses());
}
@Override
@Override
public String getLastLeadershipChangeTime() {
- return DATE_FORMAT.format(new Date(lastLeadershipChangeTime));
+ synchronized (DATE_FORMAT) {
+ return DATE_FORMAT.format(new Date(lastLeadershipChangeTime));
+ }
+ }
+
+ @Override
+ public int getPendingTxCommitQueueSize() {
+ return shard != null ? shard.getPendingTxCommitQueueSize() : -1;
+ }
+
+ @Override
+ public int getTxCohortCacheSize() {
+ return shard != null ? shard.getCohortCacheSize() : -1;
+ }
+
+ @Override
+ public void captureSnapshot() {
+ if (shard != null) {
+ shard.getSelf().tell(new InitiateCaptureSnapshot(), ActorRef.noSender());
+ }
}
}