package org.opendaylight.controller.cluster.datastore.jmx.mbeans.shard;
-import org.opendaylight.controller.cluster.datastore.jmx.mbeans.AbstractBaseMBean;
-
+import akka.actor.ActorRef;
+import akka.pattern.Patterns;
+import akka.util.Timeout;
+import com.google.common.base.Joiner;
+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 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 scala.concurrent.Await;
/**
+ * Maintains statistics for a shard.
+ *
* @author Basheeruddin syedbahm@cisco.com
*/
-public class ShardStats extends AbstractBaseMBean implements ShardStatsMBean {
+public class ShardStats extends AbstractMXBean implements ShardStatsMXBean {
+ public static String JMX_CATEGORY_SHARD = "Shards";
- private final String shardName;
+ private static final SimpleDateFormat DATE_FORMAT = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS");
- private long committedTransactionsCount = 0L;
+ private static final Cache<String, OnDemandRaftState> ONDEMAND_RAFT_STATE_CACHE =
+ CacheBuilder.newBuilder().expireAfterWrite(2, TimeUnit.SECONDS).build();
- private long readOnlyTransactionCount = 0L;
+ private long committedTransactionsCount;
- private long writeOnlyTransactionCount = 0L;
+ private long readOnlyTransactionCount;
- private long readWriteTransactionCount = 0L;
+ private long writeOnlyTransactionCount;
- private String leader;
+ private long readWriteTransactionCount;
- private String raftState;
+ private long lastCommittedTransactionTime;
- private long lastLogTerm = -1L;
+ private long failedTransactionsCount;
- private long lastLogIndex = -1L;
+ private final AtomicLong failedReadTransactionsCount = new AtomicLong();
- private long currentTerm = -1L;
+ private long abortTransactionsCount;
- private long commitIndex = -1L;
+ private boolean followerInitialSyncStatus = false;
- private long lastApplied = -1L;
+ private Shard shard;
- private Date lastCommittedTransactionTime = new Date(0L);
+ private String statRetrievalError;
- private long failedTransactionsCount = 0L;
+ private String statRetrievalTime;
- private long failedReadTransactionsCount = 0L;
+ private long leadershipChangeCount;
- private long abortTransactionsCount = 0L;
+ private long lastLeadershipChangeTime;
- private SimpleDateFormat sdf =
- new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS");
+ public ShardStats(final String shardName, final String mxBeanType) {
+ super(shardName, mxBeanType, JMX_CATEGORY_SHARD);
+ }
- ShardStats(String shardName) {
- this.shardName = shardName;
+ public void setShard(Shard shard) {
+ this.shard = shard;
}
+ @SuppressWarnings("checkstyle:IllegalCatch")
+ private OnDemandRaftState getOnDemandRaftState() {
+ String name = getShardName();
+ OnDemandRaftState state = ONDEMAND_RAFT_STATE_CACHE.getIfPresent(name);
+ if (state == null) {
+ statRetrievalError = null;
+ statRetrievalTime = null;
+
+ if (shard != null) {
+ Timeout timeout = new Timeout(10, TimeUnit.SECONDS);
+ try {
+ Stopwatch timer = Stopwatch.createStarted();
+
+ state = (OnDemandRaftState) Await.result(Patterns.ask(shard.getSelf(),
+ GetOnDemandRaftState.INSTANCE, timeout), timeout.duration());
+
+ statRetrievalTime = timer.stop().toString();
+ ONDEMAND_RAFT_STATE_CACHE.put(name, state);
+ } catch (Exception e) {
+ statRetrievalError = e.toString();
+ }
+ }
+
+ state = state != null ? state : OnDemandRaftState.builder().build();
+ }
+
+ return state;
+ }
@Override
public String getShardName() {
- return shardName;
+ return getMBeanName();
}
@Override
return committedTransactionsCount;
}
- @Override public String getLeader() {
- return leader;
+ @Override
+ public String getLeader() {
+ return getOnDemandRaftState().getLeader();
}
- @Override public String getRaftState() {
- return raftState;
+ @Override
+ public String getRaftState() {
+ return getOnDemandRaftState().getRaftState();
}
- @Override public long getReadOnlyTransactionCount() {
+ @Override
+ public long getReadOnlyTransactionCount() {
return readOnlyTransactionCount;
}
- @Override public long getWriteOnlyTransactionCount() {
+ @Override
+ public long getWriteOnlyTransactionCount() {
return writeOnlyTransactionCount;
}
- @Override public long getReadWriteTransactionCount() {
+ @Override
+ public long getReadWriteTransactionCount() {
return readWriteTransactionCount;
}
- @Override public long getLastLogIndex() {
- return lastLogIndex;
+ @Override
+ public long getLastLogIndex() {
+ return getOnDemandRaftState().getLastLogIndex();
}
- @Override public long getLastLogTerm() {
- return lastLogTerm;
+ @Override
+ public long getLastLogTerm() {
+ return getOnDemandRaftState().getLastLogTerm();
}
- @Override public long getCurrentTerm() {
- return currentTerm;
+ @Override
+ public long getCurrentTerm() {
+ return getOnDemandRaftState().getCurrentTerm();
}
- @Override public long getCommitIndex() {
- return commitIndex;
+ @Override
+ public long getCommitIndex() {
+ return getOnDemandRaftState().getCommitIndex();
}
- @Override public long getLastApplied() {
- return lastApplied;
+ @Override
+ public long getLastApplied() {
+ return getOnDemandRaftState().getLastApplied();
}
@Override
- public String getLastCommittedTransactionTime() {
+ public long getLastIndex() {
+ return getOnDemandRaftState().getLastIndex();
+ }
- return sdf.format(lastCommittedTransactionTime);
+ @Override
+ public long getLastTerm() {
+ return getOnDemandRaftState().getLastTerm();
}
- @Override public long getFailedTransactionsCount() {
- return failedTransactionsCount;
+ @Override
+ public long getSnapshotIndex() {
+ return getOnDemandRaftState().getSnapshotIndex();
}
- @Override public long getFailedReadTransactionsCount() {
- return failedReadTransactionsCount;
+ @Override
+ public long getSnapshotTerm() {
+ return getOnDemandRaftState().getSnapshotTerm();
}
- @Override public long getAbortTransactionsCount() {
- return abortTransactionsCount;
+ @Override
+ public long getReplicatedToAllIndex() {
+ return getOnDemandRaftState().getReplicatedToAllIndex();
}
- public long incrementCommittedTransactionCount() {
- return committedTransactionsCount++;
+ @Override
+ public String getVotedFor() {
+ return getOnDemandRaftState().getVotedFor();
}
- public long incrementReadOnlyTransactionCount() {
- return readOnlyTransactionCount++;
+ @Override
+ public boolean isVoting() {
+ return getOnDemandRaftState().isVoting();
}
- public long incrementWriteOnlyTransactionCount() {
- return writeOnlyTransactionCount++;
+ @Override
+ public String getPeerVotingStates() {
+ return toStringMap(getOnDemandRaftState().getPeerVotingStates());
}
- public long incrementReadWriteTransactionCount() {
- return readWriteTransactionCount++;
+ @Override
+ public boolean isSnapshotCaptureInitiated() {
+ return getOnDemandRaftState().isSnapshotCaptureInitiated();
}
- public long incrementFailedTransactionsCount() {
- return failedTransactionsCount++;
+ @Override
+ public String getLastCommittedTransactionTime() {
+ return DATE_FORMAT.format(new Date(lastCommittedTransactionTime));
}
- public long incrementFailedReadTransactionsCount() {
- return failedReadTransactionsCount++;
+ @Override
+ public long getFailedTransactionsCount() {
+ return failedTransactionsCount;
}
- public long incrementAbortTransactionsCount () { return abortTransactionsCount++;}
+ @Override
+ public long getFailedReadTransactionsCount() {
+ return failedReadTransactionsCount.get();
+ }
- public void setLeader(String leader) {
- this.leader = leader;
+ @Override
+ public long getAbortTransactionsCount() {
+ return abortTransactionsCount;
}
- public void setRaftState(String raftState) {
- this.raftState = raftState;
+ public long incrementCommittedTransactionCount() {
+ return ++committedTransactionsCount;
}
- public void setLastLogTerm(long lastLogTerm) {
- this.lastLogTerm = lastLogTerm;
+ public long incrementReadOnlyTransactionCount() {
+ return ++readOnlyTransactionCount;
}
- public void setLastLogIndex(long lastLogIndex) {
- this.lastLogIndex = lastLogIndex;
+ public long incrementWriteOnlyTransactionCount() {
+ return ++writeOnlyTransactionCount;
}
- public void setCurrentTerm(long currentTerm) {
- this.currentTerm = currentTerm;
+ public long incrementReadWriteTransactionCount() {
+ return ++readWriteTransactionCount;
}
- public void setCommitIndex(long commitIndex) {
- this.commitIndex = commitIndex;
+ public long incrementFailedTransactionsCount() {
+ return ++failedTransactionsCount;
}
- public void setLastApplied(long lastApplied) {
- this.lastApplied = lastApplied;
+ public long incrementFailedReadTransactionsCount() {
+ return failedReadTransactionsCount.incrementAndGet();
}
+ public long incrementAbortTransactionsCount() {
+ return ++abortTransactionsCount;
+ }
- public void setLastCommittedTransactionTime(
- Date lastCommittedTransactionTime) {
+ public void setLastCommittedTransactionTime(final long lastCommittedTransactionTime) {
this.lastCommittedTransactionTime = lastCommittedTransactionTime;
}
@Override
- protected String getMBeanName() {
- return shardName;
+ public long getInMemoryJournalDataSize() {
+ return getOnDemandRaftState().getInMemoryJournalDataSize();
}
@Override
- protected String getMBeanType() {
- return JMX_TYPE_DISTRIBUTED_DATASTORE;
+ public long getInMemoryJournalLogSize() {
+ return getOnDemandRaftState().getInMemoryJournalLogSize();
}
+ /**
+ * Resets the counters related to transactions.
+ */
@Override
- protected String getMBeanCategory() {
- return JMX_CATEGORY_SHARD;
+ public void resetTransactionCounters() {
+ committedTransactionsCount = 0;
+
+ readOnlyTransactionCount = 0;
+
+ writeOnlyTransactionCount = 0;
+
+ readWriteTransactionCount = 0;
+
+ lastCommittedTransactionTime = 0;
+
+ failedTransactionsCount = 0;
+
+ failedReadTransactionsCount.set(0);
+
+ abortTransactionsCount = 0;
+
}
- /**
- * resets the counters related to transactions
- */
+ public void setFollowerInitialSyncStatus(boolean followerInitialSyncStatus) {
+ this.followerInitialSyncStatus = followerInitialSyncStatus;
+ }
- public void resetTransactionCounters(){
- committedTransactionsCount = 0L;
+ @Override
+ public boolean getFollowerInitialSyncStatus() {
+ return followerInitialSyncStatus;
+ }
- readOnlyTransactionCount = 0L;
+ @Override
+ public List<FollowerInfo> getFollowerInfo() {
+ return getOnDemandRaftState().getFollowerInfoList();
+ }
- writeOnlyTransactionCount = 0L;
+ @Override
+ public String getPeerAddresses() {
+ return toStringMap(getOnDemandRaftState().getPeerAddresses());
+ }
- readWriteTransactionCount = 0L;
+ private static String toStringMap(Map<?, ?> map) {
+ return Joiner.on(", ").withKeyValueSeparator(": ").join(map);
+ }
- lastCommittedTransactionTime = new Date(0L);
+ @Override
+ public String getStatRetrievalTime() {
+ getOnDemandRaftState();
+ return statRetrievalTime;
+ }
- failedTransactionsCount = 0L;
+ @Override
+ public String getStatRetrievalError() {
+ getOnDemandRaftState();
+ return statRetrievalError;
+ }
- failedReadTransactionsCount = 0L;
+ @Override
+ public long getLeadershipChangeCount() {
+ return leadershipChangeCount;
+ }
+
+ public void incrementLeadershipChangeCount() {
+ leadershipChangeCount++;
+ lastLeadershipChangeTime = System.currentTimeMillis();
+ }
- abortTransactionsCount = 0L;
+ @Override
+ public String getLastLeadershipChangeTime() {
+ return DATE_FORMAT.format(new Date(lastLeadershipChangeTime));
+ }
+ @Override
+ public int getPendingTxCommitQueueSize() {
+ return shard.getPendingTxCommitQueueSize();
}
+ @Override
+ public int getTxCohortCacheSize() {
+ return shard.getCohortCacheSize();
+ }
+ @Override
+ public void captureSnapshot() {
+ if (shard != null) {
+ shard.getSelf().tell(new InitiateCaptureSnapshot(), ActorRef.noSender());
+ }
+ }
}