/* This file is part of VoltDB.
* Copyright (C) 2008-2017 VoltDB Inc.
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
* IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/
package org.voltdb.regressionsuites;
import static com.google_voltpatches.common.base.Preconditions.checkArgument;
import java.util.Iterator;
import java.util.List;
import java.util.NavigableSet;
import com.google_voltpatches.common.collect.ImmutableList;
import com.google_voltpatches.common.collect.ImmutableSortedSet;
public class InternalPortGeneratorForTest implements Iterable<Integer>{
final List<Integer> ports;
final NavigableSet<String> coordinators;
final PortGeneratorForTest generator;
public InternalPortGeneratorForTest(PortGeneratorForTest generator, int seedCount) {
checkArgument(seedCount >= 0, "clusterSize %s in less then zero", seedCount);
ImmutableList.Builder<Integer> lb = ImmutableList.builder();
ImmutableSortedSet.Builder<String> sb = ImmutableSortedSet.naturalOrder();
for (int i = 0; i < seedCount; ++i) {
int port = generator.nextInternalPort();
lb.add(port);
sb.add(":" + Integer.toString(port));
}
this.ports = lb.build();
this.coordinators = sb.build();
this.generator = generator;
}
@Override
public Iterator<Integer> iterator() {
return ports.iterator();
}
public List<Integer> getPorts() {
return ports;
}
public NavigableSet<String> getCoordinators() {
return coordinators;
}
public int nextInternalPort(int hostId) {
checkArgument(hostId >= 0, "host id %s is less then 0",hostId);
if (hostId >= ports.size()) {
return generator.nextInternalPort();
}
return ports.get(hostId);
}
}