PersistentTopicStats.java

/**
 * Copyright 2016 Yahoo Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.yahoo.pulsar.common.policies.data;

import static com.google.common.base.Preconditions.checkNotNull;

import java.util.List;
import java.util.Map;

import com.google.common.collect.Lists;
import com.google.common.collect.Maps;

/**
 */
public class PersistentTopicStats {
    /** Total rate of messages published on the topic. msg/s */
    public double msgRateIn;

    /** Total throughput of messages published on the topic. byte/s */
    public double msgThroughputIn;

    /** Total rate of messages dispatched for the topic. msg/s */
    public double msgRateOut;

    /** Total throughput of messages dispatched for the topic. byte/s */
    public double msgThroughputOut;

    /** Average size of published messages. bytes */
    public double averageMsgSize;

    /** Space used to store the messages for the topic. bytes */
    public long storageSize;

    /** List of connected publishers on this topic w/ their stats */
    public List<PublisherStats> publishers;

    /** Map of subscriptions with their individual statistics */
    public Map<String, PersistentSubscriptionStats> subscriptions;

    /** Map of replication statistics by remote cluster context */
    public Map<String, ReplicatorStats> replication;

    public PersistentTopicStats() {
        this.publishers = Lists.newArrayList();
        this.subscriptions = Maps.newHashMap();
        this.replication = Maps.newTreeMap();
    }

    public void reset() {
        this.msgRateIn = 0;
        this.msgThroughputIn = 0;
        this.msgRateOut = 0;
        this.msgThroughputOut = 0;
        this.averageMsgSize = 0;
        this.storageSize = 0;
        this.publishers.clear();
        this.subscriptions.clear();
        this.replication.clear();
    }

    // if the stats are added for the 1st time, we will need to make a copy of these stats and add it to the current
    // stats
    public PersistentTopicStats add(PersistentTopicStats stats) {
        checkNotNull(stats);
        this.msgRateIn += stats.msgRateIn;
        this.msgThroughputIn += stats.msgThroughputIn;
        this.msgRateOut += stats.msgRateOut;
        this.msgThroughputOut += stats.msgThroughputOut;
        this.averageMsgSize += stats.averageMsgSize;
        this.storageSize += stats.storageSize;
        if (this.publishers.size() != stats.publishers.size()) {
            for (int i = 0; i < stats.publishers.size(); i++) {
                PublisherStats publisherStats = new PublisherStats();
                this.publishers.add(publisherStats.add(stats.publishers.get(i)));
            }
        } else {
            for (int i = 0; i < stats.publishers.size(); i++) {
                this.publishers.get(i).add(stats.publishers.get(i));
            }
        }
        if (this.subscriptions.size() != stats.subscriptions.size()) {
            for (String subscription : stats.subscriptions.keySet()) {
                PersistentSubscriptionStats subscriptionStats = new PersistentSubscriptionStats();
                this.subscriptions.put(subscription, subscriptionStats.add(stats.subscriptions.get(subscription)));
            }
        } else {
            for (String subscription : stats.subscriptions.keySet()) {
                this.subscriptions.get(subscription).add(stats.subscriptions.get(subscription));
            }
        }
        if (this.replication.size() != stats.replication.size()) {
            for (String repl : stats.replication.keySet()) {
                ReplicatorStats replStats = new ReplicatorStats();
                this.replication.put(repl, replStats.add(stats.replication.get(repl)));
            }
        } else {
            for (String repl : stats.replication.keySet()) {
                this.replication.get(repl).add(stats.replication.get(repl));
            }
        }
        return this;
    }
}