io.github.mike10004.vhs.BasicRequestTrackingState Maven / Gradle / Ivy

Go to download

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

There is a newer version: 0.32

package io.github.mike10004.vhs;

import com.google.common.primitives.Ints;
import com.google.common.util.concurrent.AtomicLongMap;
import io.github.mike10004.vhs.harbridge.HttpMethod;
import io.github.mike10004.vhs.harbridge.ParsedRequest;
import org.apache.commons.lang3.tuple.ImmutablePair;
import org.slf4j.LoggerFactory;

import java.net.URI;

class BasicRequestTrackingState implements ReplaySessionState {

    private final AtomicLongMap> requestCouples = AtomicLongMap.create();

    @Override
    public void register(ParsedRequest request) {
        ImmutablePair requestCouple = pairify(request);
        requestCouples.getAndIncrement(requestCouple);
    }

    private ImmutablePair pairify(ParsedRequest request) {
        return ImmutablePair.of(request.method, request.url);
    }

    @Override
    public int query(ParsedRequest request) {
        ImmutablePair requestCouple = pairify(request);
        long value = requestCouples.get(requestCouple);
        if (value == 0L) {
            LoggerFactory.getLogger(getClass()).info("queried request that was never registered: {}", request);
        }
        return Ints.saturatedCast(value - 1L);
    }

    @Override
    public String toString() {
        return String.format("BasicRequestTrackingState{registered=%d}", requestCouples.size());
    }
}