io.muserver.MuStatsImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mu-server Show documentation

A simple but powerful web server framework

There is a newer version: 2.0.3

package io.muserver;

import io.netty.handler.traffic.TrafficCounter;

import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicLong;

class MuStatsImpl implements MuStats {
    private final TrafficCounter trafficCounter;
    private final AtomicLong completedRequests = new AtomicLong(0);
    private final AtomicLong invalidHttpRequests = new AtomicLong(0);
    private final Set activeRequests = ConcurrentHashMap.newKeySet();

    MuStatsImpl(TrafficCounter trafficCounter) {
        this.trafficCounter = trafficCounter;
    }

    @Override
    public long activeConnections() {
        return activeRequests.size();
    }

    @Override
    public long completedRequests() {
        return completedRequests.get();
    }

    @Override
    public long invalidHttpRequests() {
        return invalidHttpRequests.get();
    }

    @Override
    public long bytesSent() {
        return trafficCounter.cumulativeWrittenBytes();
    }

    @Override
    public long bytesRead() {
        return trafficCounter.cumulativeReadBytes();
    }

    @Override
    public Set activeRequests() {
        return activeRequests;
    }

    void onRequestStarted(MuRequest request) {
        activeRequests.add(request);
    }
    void onRequestEnded(MuRequest request) {
        activeRequests.remove(request);
        completedRequests.incrementAndGet();
    }
    void onInvalidRequest() {
        invalidHttpRequests.incrementAndGet();
    }

    @Override
    public String toString() {
        return "Completed requests: " + completedRequests() + "; active: " + activeConnections() + "; invalid requests: " + invalidHttpRequests() + "; bytes received: " + bytesRead() + "; bytes sent: " + bytesSent();
    }
}