CircularExclusiveAccessLoadBalancer.java

Go to the documentation of this file.

package com.griddynamics.jagger.invoker;

import com.griddynamics.jagger.util.Pair;
import com.griddynamics.jagger.util.Timeout;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.concurrent.TimeUnit;

public class CircularExclusiveAccessLoadBalancer<Q, E> extends ExclusiveAccessLoadBalancer<Q, E> {
    
    private final static Logger log = LoggerFactory.getLogger(CircularExclusiveAccessLoadBalancer.class);
    
    public CircularExclusiveAccessLoadBalancer(PairSupplierFactory<Q, E> pairSupplierFactory) {
        super(pairSupplierFactory);
    }
    
    private final static String POLL_TIMEOUT_NAME = "load balancer poll next timeout";
    
    private Timeout pollTimeout = new Timeout(10 * 60 * 1000, POLL_TIMEOUT_NAME); // 10 minutes by default
    
    @Override
    protected boolean isToCircleAnIteration() {
        return true;
    }
    
    protected Pair<Q, E> pollNext() {
        
        Pair<Q, E> next = null;
        long startMillis = System.currentTimeMillis();
        try {
            next = getPairQueue().poll(pollTimeout.getValue(), TimeUnit.MILLISECONDS);
        } catch (InterruptedException ignored) {
        }
        
        if (next == null) {
            throw new IllegalStateException(String.format("Didn't manage to poll the next pair. Timeout %s",
                                                          pollTimeout));
        }
        
        long endMillis = System.currentTimeMillis();
        if (endMillis - startMillis > 1000) {
            log.warn(
                    "It took {} ms to poll the next pair for the load balancer. Possible reason: not enough test data for selected load",
                    endMillis - startMillis);
        }
        
        return next;
    }
    
    public void setPollTimeout(long pollTimeout) {
        this.pollTimeout = new Timeout(pollTimeout, POLL_TIMEOUT_NAME);
    }
}