Merge "BUG 2799: Migration of Message Bus from deprecated Helium MD-SAL APIs to Lithi...
[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / main / java / org / opendaylight / controller / cluster / datastore / jmx / mbeans / shard / ShardStats.java
index 4eb6a8cef96319d0ca885112c3a0812ee06e5431..fb59b7643f09f77ca19013a4cda3051bf4320bd8 100644 (file)
+/*
+ * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
 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.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.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.ThreadExecutorStats;
+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.ListenerNotificationQueueStats;
+import org.opendaylight.yangtools.util.concurrent.QueuedNotificationManager;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import scala.concurrent.Await;
 
 /**
- * @author: syedbahm
+ * Maintains statistics for a shard.
+ *
+ * @author  Basheeruddin syedbahm@cisco.com
  */
-public class ShardStats extends AbstractBaseMBean implements ShardStatsMBean {
-  private  Long committedTransactionsCount;
-  private Long journalMessagesCount;
-  final private String shardName;
-  private String leader;
-  private String raftState;
+public class ShardStats extends AbstractMXBean implements ShardStatsMXBean {
+    public static String JMX_CATEGORY_SHARD = "Shards";
+
+    private static final Logger LOG = LoggerFactory.getLogger(ShardStats.class);
+
+    private static final SimpleDateFormat DATE_FORMAT = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS");
+
+    private static final Cache<String, OnDemandRaftState> onDemandRaftStateCache =
+            CacheBuilder.newBuilder().expireAfterWrite(2, TimeUnit.SECONDS).build();
+
+    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 ThreadExecutorStatsMXBeanImpl notificationExecutorStatsBean;
+
+    private QueuedNotificationManagerMXBeanImpl notificationManagerStatsBean;
+
+    private boolean followerInitialSyncStatus = false;
+
+    private ActorRef shardActor;
+
+    private String statRetrievalError;
+
+    private String statRetrievalTime;
+
+    public ShardStats(final String shardName, final String mxBeanType) {
+        super(shardName, mxBeanType, JMX_CATEGORY_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;
+    }
+
+    private OnDemandRaftState getOnDemandRaftState() {
+        String name = getShardName();
+        OnDemandRaftState state = onDemandRaftStateCache.getIfPresent(name);
+        if(state == null) {
+            statRetrievalError = null;
+            statRetrievalTime = null;
+
+            if(shardActor != null) {
+                Timeout timeout = new Timeout(10, TimeUnit.SECONDS);
+                try {
+                    Stopwatch timer = Stopwatch.createStarted();
+
+                    state = (OnDemandRaftState) Await.result(Patterns.ask(shardActor,
+                            GetOnDemandRaftState.INSTANCE, timeout), timeout.duration());
+
+                    statRetrievalTime = timer.stop().toString();
+                    onDemandRaftStateCache.put(name, state);
+                } catch (Exception e) {
+                    statRetrievalError = e.toString();
+                }
+            }
+
+            state = state != null ? state : OnDemandRaftState.builder().build();
+        }
+
+        return state;
+    }
+
+    @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 isSnapshotCaptureInitiated() {
+        return getOnDemandRaftState().isSnapshotCaptureInitiated();
+    }
+
+    @Override
+    public String getLastCommittedTransactionTime() {
+        return DATE_FORMAT.format(new Date(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();
+    }
+
+    @Override
+    public ThreadExecutorStats getDataStoreExecutorStats() {
+        // FIXME: this particular thing does not work, as it really is DS-specific
+        return null;
+    }
 
-  ShardStats(String shardName){
-    this.shardName = shardName;
-    committedTransactionsCount =0L;
-    journalMessagesCount = 0L;
-  };
+    @Override
+    public ThreadExecutorStats getNotificationMgrExecutorStats() {
+        return notificationExecutorStatsBean.toThreadExecutorStats();
+    }
 
+    @Override
+    public List<ListenerNotificationQueueStats> getCurrentNotificationMgrListenerQueueStats() {
+        return notificationManagerStatsBean.getCurrentListenerQueueStats();
+    }
 
-  @Override
-  public String getShardName() {
-    return shardName;
-  }
+    @Override
+    public int getMaxNotificationMgrListenerQueueSize() {
+        return notificationManagerStatsBean.getMaxListenerQueueSize();
+    }
 
-  @Override
-  public Long getCommittedTransactionsCount() {
-    return committedTransactionsCount;
-  }
+    /**
+     * resets the counters related to transactions
+     */
+    @Override
+    public void resetTransactionCounters(){
+        committedTransactionsCount = 0;
 
-  @Override
-  public Long getJournalMessagesCount() {
-    //FIXME: this will be populated once after integration with Raft stuff
-    return journalMessagesCount;
-  }
+        readOnlyTransactionCount = 0;
 
-  @Override public String getLeader() {
-    return leader;
-  }
+        writeOnlyTransactionCount = 0;
 
-  @Override public String getRaftState() {
-    return raftState;
-  }
+        readWriteTransactionCount = 0;
 
-  public Long incrementCommittedTransactionCount() {
-    return committedTransactionsCount++;
-  }
+        lastCommittedTransactionTime = 0;
 
+        failedTransactionsCount = 0;
 
-  public void updateCommittedTransactionsCount(long currentCount){
-     committedTransactionsCount = currentCount;
+        failedReadTransactionsCount.set(0);
 
-  }
+        abortTransactionsCount = 0;
 
-  public void updateJournalMessagesCount(long currentCount){
-    journalMessagesCount  = currentCount;
+    }
 
-  }
+    public void setDataStore(final InMemoryDOMDataStore store) {
+        setNotificationManager(store.getDataChangeListenerNotificationManager());
+    }
 
-  public void setLeader(String leader){
-    this.leader = leader;
-  }
+    public void setFollowerInitialSyncStatus(boolean followerInitialSyncStatus) {
+        this.followerInitialSyncStatus = followerInitialSyncStatus;
+    }
 
-  public void setRaftState(String raftState){
-    this.raftState = raftState;
-  }
+    @Override
+    public boolean getFollowerInitialSyncStatus() {
+        return followerInitialSyncStatus;
+    }
 
+    @Override
+    public List<FollowerInfo> getFollowerInfo() {
+        return getOnDemandRaftState().getFollowerInfoList();
+    }
 
-  @Override
-  protected String getMBeanName() {
-    return  shardName;
-  }
+    @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(", ");
+            }
 
-  @Override
-  protected String getMBeanType() {
-    return JMX_TYPE_DISTRIBUTED_DATASTORE;
-  }
+            builder.append(e.getKey()).append(": ").append(e.getValue());
+        }
 
-  @Override
-  protected String getMBeanCategory() {
-    return JMX_CATEGORY_SHARD;
-  }
+        return builder.toString();
+    }
 
+    @Override
+    public String getStatRetrievalTime() {
+        getOnDemandRaftState();
+        return statRetrievalTime;
+    }
 
+    @Override
+    public String getStatRetrievalError() {
+        getOnDemandRaftState();
+        return statRetrievalError;
+    }
 }