+public class ShardStats extends AbstractMXBean implements ShardStatsMXBean {
+ public static final String JMX_CATEGORY_SHARD = "Shards";
+
+ @GuardedBy("DATE_FORMAT")
+ private static final SimpleDateFormat DATE_FORMAT = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS");
+
+ private static final MapJoiner MAP_JOINER = Joiner.on(", ").withKeyValueSeparator(": ");
+
+ private final Shard shard;
+
+ private final OnDemandShardStateCache stateCache;
+
+ private long committedTransactionsCount;
+
+ private long readOnlyTransactionCount;
+
+ private long writeOnlyTransactionCount;
+
+ private long readWriteTransactionCount;
+
+ private long lastCommittedTransactionTime;
+
+ private long failedTransactionsCount;
+
+ private final AtomicLong failedReadTransactionsCount = new AtomicLong();
+
+ private long abortTransactionsCount;
+
+ private boolean followerInitialSyncStatus = false;
+
+ private String statRetrievalError;
+
+ private long leadershipChangeCount;
+
+ private long lastLeadershipChangeTime;
+
+ public ShardStats(final String shardName, final String mxBeanType, @Nullable final Shard shard) {
+ super(shardName, mxBeanType, JMX_CATEGORY_SHARD);
+ this.shard = shard;
+ stateCache = new OnDemandShardStateCache(shardName, shard != null ? shard.self() : null);
+ }
+
+ @SuppressWarnings("checkstyle:IllegalCatch")
+ private OnDemandRaftState getOnDemandRaftState() {
+ try {
+ final OnDemandRaftState state = stateCache.get();
+ statRetrievalError = null;
+ return state;
+ } catch (Exception e) {
+ statRetrievalError = e.getCause().toString();
+ return OnDemandRaftState.builder().build();
+ }
+ }
+
+ private static String formatMillis(final long timeMillis) {
+ synchronized (DATE_FORMAT) {
+ return DATE_FORMAT.format(new Date(timeMillis));
+ }
+ }
+
+ @Override
+ public String getShardName() {
+ return getMBeanName();
+ }
+
+ @Override
+ public long getCommittedTransactionsCount() {
+ return committedTransactionsCount;
+ }
+
+ @Override
+ public String getLeader() {
+ return getOnDemandRaftState().getLeader();
+ }
+
+ @Override
+ public String getRaftState() {
+ return getOnDemandRaftState().getRaftState();
+ }
+
+ @Override
+ public long getReadOnlyTransactionCount() {
+ return readOnlyTransactionCount;
+ }
+
+ @Override
+ public long getWriteOnlyTransactionCount() {
+ return writeOnlyTransactionCount;
+ }
+
+ @Override
+ public long getReadWriteTransactionCount() {
+ return readWriteTransactionCount;
+ }
+
+ @Override
+ public long getLastLogIndex() {
+ return getOnDemandRaftState().getLastLogIndex();
+ }
+
+ @Override
+ public long getLastLogTerm() {
+ return getOnDemandRaftState().getLastLogTerm();
+ }
+
+ @Override
+ public long getCurrentTerm() {
+ return getOnDemandRaftState().getCurrentTerm();
+ }
+
+ @Override
+ public long getCommitIndex() {
+ return getOnDemandRaftState().getCommitIndex();
+ }
+
+ @Override
+ public long getLastApplied() {
+ return getOnDemandRaftState().getLastApplied();
+ }
+
+ @Override
+ public long getLastIndex() {
+ return getOnDemandRaftState().getLastIndex();
+ }
+
+ @Override
+ public long getLastTerm() {
+ return getOnDemandRaftState().getLastTerm();
+ }
+
+ @Override
+ public long getSnapshotIndex() {
+ return getOnDemandRaftState().getSnapshotIndex();
+ }
+
+ @Override
+ public long getSnapshotTerm() {
+ return getOnDemandRaftState().getSnapshotTerm();
+ }
+
+ @Override
+ public long getReplicatedToAllIndex() {
+ return getOnDemandRaftState().getReplicatedToAllIndex();
+ }
+
+ @Override
+ public String getVotedFor() {
+ 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 formatMillis(lastCommittedTransactionTime);
+ }
+
+ @Override
+ public long getFailedTransactionsCount() {
+ return failedTransactionsCount;
+ }
+
+ @Override
+ public long getFailedReadTransactionsCount() {
+ return failedReadTransactionsCount.get();
+ }
+
+ @Override
+ public long getAbortTransactionsCount() {
+ return abortTransactionsCount;
+ }
+
+ public long incrementCommittedTransactionCount() {
+ return ++committedTransactionsCount;
+ }
+
+ public long incrementReadOnlyTransactionCount() {
+ return ++readOnlyTransactionCount;
+ }
+
+ public long incrementWriteOnlyTransactionCount() {
+ return ++writeOnlyTransactionCount;
+ }
+
+ public long incrementReadWriteTransactionCount() {
+ return ++readWriteTransactionCount;
+ }
+
+ public long incrementFailedTransactionsCount() {
+ return ++failedTransactionsCount;
+ }
+
+ public long incrementFailedReadTransactionsCount() {
+ return failedReadTransactionsCount.incrementAndGet();
+ }
+
+ public long incrementAbortTransactionsCount() {
+ return ++abortTransactionsCount;
+ }
+
+ public void setLastCommittedTransactionTime(final long lastCommittedTransactionTime) {
+ this.lastCommittedTransactionTime = lastCommittedTransactionTime;
+ }
+
+ @Override
+ public long getInMemoryJournalDataSize() {
+ return getOnDemandRaftState().getInMemoryJournalDataSize();
+ }
+
+ @Override
+ public long getInMemoryJournalLogSize() {
+ return getOnDemandRaftState().getInMemoryJournalLogSize();
+ }