org.graylog2.inputs.BasicCache Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of graylog2-inputs Show documentation

Graylog message inputs

There is a newer version: 1.3.4

/**
 * This file is part of Graylog2.
 *
 * Graylog2 is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Graylog2 is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Graylog2.  If not, see .
 */
package org.graylog2.inputs;

import com.google.common.collect.Queues;
import org.cliffc.high_scale_lib.Counter;
import org.graylog2.plugin.Message;

import java.util.concurrent.ConcurrentLinkedQueue;

/**
 * 
 * @author Lennart Koopmann 
 */
public class BasicCache implements InputCache, OutputCache {
    
    private final ConcurrentLinkedQueue q;
    private final Counter counter;
    
    public BasicCache() {
        q = Queues.newConcurrentLinkedQueue();
        counter = new Counter();
    }
    
    @Override
    public void add(Message m) {
        q.add(m);
        counter.increment();
    }
    
    @Override
    public Message pop() {
        Message m = q.poll();
        
        // Only decrement counter if we really popped something.
        if (m != null) {
            counter.decrement();
            return m;
        } else {
            return null;
        }
    }
    
    @Override
    public int size() {
        return counter.intValue();
    }

    @Override
    public boolean isEmpty() {
        return q.isEmpty();
    }

}