com.github.tomakehurst.wiremock.verification.journal.BoundedInMemoryRequestJournal.java Source code

Introduction

Here is the source code for com.github.tomakehurst.wiremock.verification.journal.BoundedInMemoryRequestJournal.java

Source

/*
 * Copyright (C) 2013 Roger Abelenda
 *
 * 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.github.tomakehurst.wiremock.verification.journal;

import com.github.tomakehurst.wiremock.http.Request;
import com.github.tomakehurst.wiremock.matching.RequestPattern;
import com.github.tomakehurst.wiremock.verification.LoggedRequest;
import com.google.common.base.Predicate;
import com.google.common.collect.ImmutableList;
import org.apache.commons.collections.buffer.CircularFifoBuffer;

import java.util.Collection;
import java.util.List;

import static com.google.common.collect.Iterables.filter;
import static com.google.common.collect.Iterables.size;

public class BoundedInMemoryRequestJournal implements ImmutableCapacityJournal {

    private final CircularFifoBuffer requests;

    public BoundedInMemoryRequestJournal(int capacity) {
        requests = new CircularFifoBuffer(capacity);
    }

    @SuppressWarnings("unchecked")
    @Override
    public int countRequestsMatching(RequestPattern requestPattern) {
        return size(filter((Collection<LoggedRequest>) requests, matchedBy(requestPattern)));
    }

    @SuppressWarnings("unchecked")
    @Override
    public List<LoggedRequest> getRequestsMatching(RequestPattern requestPattern) {
        return ImmutableList.copyOf(filter((Collection<LoggedRequest>) requests, matchedBy(requestPattern)));
    }

    private Predicate<Request> matchedBy(final RequestPattern requestPattern) {
        return new Predicate<Request>() {
            public boolean apply(Request input) {
                return requestPattern.isMatchedBy(input).isMatched();
            }
        };
    }

    @Override
    public void requestReceived(Request request) {
        requests.add(LoggedRequest.createFrom(request));
    }

    @Override
    public void reset() {
        requests.clear();
    }

    @Override
    public void load(List<LoggedRequest> loggedRequests) {
        requests.addAll(loggedRequests);
    }

    @Override
    public List<LoggedRequest> getAllRequests() {
        return ImmutableList.copyOf(requests);
    }
}