ru.qatools.gridrouter.config.RandomHostSelectionStrategy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gridrouter-config Show documentation

There is a newer version: 1.31

package ru.qatools.gridrouter.config;

import java.util.ArrayList;
import java.util.List;

import static java.util.Collections.nCopies;
import static java.util.Collections.shuffle;

/**
 * @author Innokenty Shuvalov [email protected]
 */
public class RandomHostSelectionStrategy implements HostSelectionStrategy {

    protected  T selectRandom(List elements) {
        List copies = new ArrayList<>();
        for (T element : elements) {
            copies.addAll(nCopies(element.getCount(), element));
        }
        shuffle(copies);
        return copies.get(0);
    }

    @Override
    public Region selectRegion(List allRegions, List unvisitedRegions) {
        return selectRandom(unvisitedRegions);
    }

    @Override
    public Host selectHost(List hosts) {
        return selectRandom(hosts);
    }
}