org.browsermob.proxy.ProxyServerLog Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of browsermob-proxy Show documentation

A programmatic HTTP/S designed for performance and functional testing

There is a newer version: 2.0-beta-7

package org.browsermob.proxy;

import java.util.ArrayList;
import java.util.List;
import java.util.Set;
import java.util.Map;
import java.util.HashMap;
import java.util.HashSet;
import java.lang.IllegalArgumentException;

import net.jcip.annotations.GuardedBy;
import net.jcip.annotations.ThreadSafe;

/** Keep track of what the proxy server is doing.*/
@ThreadSafe
public class ProxyServerLog {
    @GuardedBy("this")
    private List recentBlocks = new ArrayList();
    private int maxRecentBlocks = 100;
    @GuardedBy("this")
    private Map> namedBlocks;
    @GuardedBy("this")
    private Set activeNamesBlocks;

    public ProxyServerLog(int maxRecentBlocks) {
        this.maxRecentBlocks = maxRecentBlocks;
        namedBlocks = new HashMap>();
        activeNamesBlocks = new HashSet();
    }


    public synchronized void clearRecentBlocks() {
        recentBlocks.clear();
    }


    public synchronized List getRecentBlocks() {
        return new ArrayList(recentBlocks);
    }


    public synchronized List getLastNRecentBlocks(int n) {
        return new ArrayList(recentBlocks.subList(0, n));
    }


    public synchronized void startNewNamedSession(String key) {
        if (activeNamesBlocks.contains(key)) {
            throw new IllegalArgumentException("session with key: " + key + "already exists");
        }
        activeNamesBlocks.add(key);
        namedBlocks.put(key, new ArrayList());
    }


    public synchronized void clearNamedSession(String key) {
        if (!namedBlocks.containsKey(key)) {
            throw new IllegalArgumentException("session with key: " + key + "does not exist");
        }
        activeNamesBlocks.remove(key);
        namedBlocks.remove(key);
    }


    public synchronized void stopSession(String key) {
        activeNamesBlocks.remove(key);
    }


    public synchronized List getNamedSession(String key) {
        if (!namedBlocks.containsKey(key)) {
            throw new IllegalArgumentException("session with key: " + key + "does not exist");
        }
        return namedBlocks.get(key);
    }



    private void recordBlocks(List blocks, HttpObject httpObject, int maxSize) {
        if (blocks.isEmpty()) {
            // put in the first block
            blocks.add(new Block());
        }
        
        Block block = blocks.get(0);
        if (!block.add(httpObject)) {
            Block newBlock = new Block();
            newBlock.add(httpObject);
            blocks.add(0, newBlock);

            if (maxSize > 0 &&  blocks.size() > maxSize) {
                blocks.remove(maxSize);
            }
        }
    }


    public synchronized void record(HttpObject httpObject) {
        recordBlocks(recentBlocks, httpObject, maxRecentBlocks);
        for (String key : activeNamesBlocks) {
            recordBlocks(namedBlocks.get(key), httpObject, -1);
        }
    }

}