@Test
public void testPopulator()
throws IOException, InterruptedException, ExecutionException, MutationFailedException,
TimeoutException {
PopulatorTest populator = new PopulatorTest();
tableName = ByteString.copyFrom(Bytes.toBytes(name.getMethodName()));
main(new String[] {String.valueOf(getRegionServerPort())});
}
public class PopulatorTest extends MiniClusterBase {
private static ByteString tableName = ByteString.bytesDefaultValue("testTable");
public PopulatorTest() {}
public static void main(String[] args)
throws InterruptedException, ExecutionException, TimeoutException, IOException {
int port;
if (args.length < 1) {
port = 31337;
} else {
port = Integer.parseInt(args[0]);
}
try (FakeHTable table = new FakeHTable(C5TestServerConstants.LOCALHOST, port, tableName)) {
long start = System.currentTimeMillis();
int numberOfBatches = 100;
int batchSize = 10;
if (args.length == 2) {
numberOfBatches = Integer.parseInt(args[0]);
batchSize = Integer.parseInt(args[1]);
}
compareToHBasePut(
table,
Bytes.toBytes("cf"),
Bytes.toBytes("cq"),
Bytes.toBytes("value"),
numberOfBatches,
batchSize);
long end = System.currentTimeMillis();
System.out.println("time:" + (end - start));
}
}
private static void compareToHBasePut(
final FakeHTable table,
final byte[] cf,
final byte[] cq,
final byte[] value,
final int numberOfBatches,
final int batchSize)
throws IOException {
ArrayList<Put> puts = new ArrayList<>();
long startTime = System.nanoTime();
for (int j = 1; j != numberOfBatches + 1; j++) {
for (int i = 1; i != batchSize + 1; i++) {
puts.add(new Put(Bytes.vintToBytes(i * j)).add(cf, cq, value));
}
int i = 0;
for (Put put : puts) {
i++;
if (i % 1024 == 0) {
long timeDiff = (System.nanoTime()) - startTime;
System.out.print("#(" + timeDiff + ")");
System.out.flush();
startTime = System.nanoTime();
}
if (i % (1024 * 12) == 0) {
System.out.println("");
}
table.put(put);
}
puts.clear();
}
}
@Test
public void testPopulator()
throws IOException, InterruptedException, ExecutionException, MutationFailedException,
TimeoutException {
PopulatorTest populator = new PopulatorTest();
tableName = ByteString.copyFrom(Bytes.toBytes(name.getMethodName()));
main(new String[] {String.valueOf(getRegionServerPort())});
}
}
