org.openfeed.client.api.impl.ConnectionStats Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of openfeed-client Show documentation

Openfeed Java Client

The newest version!

package org.openfeed.client.api.impl;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

public class ConnectionStats {
    private final MessageStats overallStats = new MessageStats();

    // channel => { exchangeCode => Stats }
    private final Map> channelToExchangeToMessageStats = new HashMap<>();

    public MessageStats getMessageStats() {
        return this.overallStats;
    }

    public MessageStats getExchangeMessageStats(int channel, String exchangeCode) {
        if (!channelToExchangeToMessageStats.containsKey(channel)) {
            channelToExchangeToMessageStats.put(channel,new HashMap<>());
        }
        return this.channelToExchangeToMessageStats.get(channel)
                .computeIfAbsent(exchangeCode, key -> new MessageStats(channel,exchangeCode));
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder("\nOverall: " + overallStats + "\n");

        for(Map channelStats : channelToExchangeToMessageStats.values()) {
            channelStats.values().forEach(stats ->  sb.append("\t " + stats + "\n"));
        }
        return sb.toString();
    }

    public void clear() {
        overallStats.clear();
        channelToExchangeToMessageStats.clear();
    }
}