io.split.client.impressions.InMemoryImpressionsStorage Maven / Gradle / Ivy

Go to download
package io.split.client.impressions;

import io.split.client.dtos.KeyImpression;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingQueue;

public class InMemoryImpressionsStorage implements ImpressionsStorage {

    private static final Logger _log = LoggerFactory.getLogger(InMemoryImpressionsStorage.class);

    private final BlockingQueue _queue;

    public InMemoryImpressionsStorage(int queueSize) {
        _queue = new LinkedBlockingQueue<>(queueSize);
    }

    @Override
    public List pop(int count) {
        ArrayList popped = new ArrayList<>();
        _queue.drainTo(popped, count);
        return popped;
    }

    @Override
    public List pop() {
        ArrayList popped = new ArrayList<>();
        _queue.drainTo(popped);
        return popped;
    }

    @Override
    public boolean isFull() {
        return _queue.remainingCapacity() == 0;
    }

    private boolean put(KeyImpression imp) {
        try {
            return _queue.offer(imp);
        } catch (ClassCastException | NullPointerException | IllegalArgumentException e) {
            _log.warn("Unable to send impression to ImpressionsManager", e);
            return false;
        }
    }

    @Override
    public long put(List imps) {
        return imps.stream().reduce(0, (accum, current) -> {
            if(this.put(current)) {
                return accum + 1;
            }
            return accum;
        }, Integer::sum).longValue();
    }
}