use of org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.sal.clustering.it.car.rev140818.cars.CarEntry in project controller by opendaylight.
the class CarProvider method stressTest.
@Override
public Future<RpcResult<Void>> stressTest(final StressTestInput input) {
final int inputRate;
final long inputCount;
// If rate is not provided, or given as zero, then just return.
if (input.getRate() == null || input.getRate() == 0) {
LOG_PURCHASE_CAR.info("Exiting stress test as no rate is given.");
return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withError(ErrorType.PROTOCOL, "invalid rate").build());
}
inputRate = input.getRate();
if (input.getCount() != null) {
inputCount = input.getCount();
} else {
inputCount = 0;
}
LOG_PURCHASE_CAR.info("Stress test starting : rate: {} count: {}", inputRate, inputCount);
stopThread();
// clear counters
succcessCounter.set(0);
failureCounter.set(0);
WriteTransaction tx = dataProvider.newWriteOnlyTransaction();
InstanceIdentifier<Cars> carsId = InstanceIdentifier.<Cars>builder(Cars.class).build();
tx.merge(LogicalDatastoreType.CONFIGURATION, carsId, new CarsBuilder().build());
try {
tx.submit().checkedGet(5, TimeUnit.SECONDS);
} catch (TransactionCommitFailedException | TimeoutException e) {
LOG_PURCHASE_CAR.error("Put Cars failed", e);
return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
}
stopThread = false;
final long sleep = TimeUnit.NANOSECONDS.convert(1000, TimeUnit.MILLISECONDS) / inputRate;
final Stopwatch sw = Stopwatch.createUnstarted();
testThread = new Thread(() -> {
sw.start();
AtomicLong count = new AtomicLong();
while (!stopThread) {
long id = count.incrementAndGet();
WriteTransaction tx1 = dataProvider.newWriteOnlyTransaction();
CarEntry car = new CarEntryBuilder().setId(new CarId("car" + id)).build();
tx1.put(LogicalDatastoreType.CONFIGURATION, InstanceIdentifier.<Cars>builder(Cars.class).child(CarEntry.class, car.getKey()).build(), car);
CheckedFuture<Void, TransactionCommitFailedException> future = tx1.submit();
Futures.addCallback(future, new FutureCallback<Void>() {
@Override
public void onSuccess(final Void result) {
// Transaction succeeded
succcessCounter.getAndIncrement();
}
@Override
public void onFailure(final Throwable ex) {
// Transaction failed
failureCounter.getAndIncrement();
LOG_CAR_PROVIDER.error("Put Cars failed", ex);
}
}, MoreExecutors.directExecutor());
try {
TimeUnit.NANOSECONDS.sleep(sleep);
} catch (InterruptedException e) {
break;
}
if (count.get() % 1000 == 0) {
LOG_PURCHASE_CAR.info("Cars created {}, time: {}", count.get(), sw.elapsed(TimeUnit.SECONDS));
}
// Check if a count is specified in input and we have created that many cars.
if (inputCount != 0 && count.get() >= inputCount) {
stopThread = true;
}
}
LOG_PURCHASE_CAR.info("Stress test thread stopping after creating {} cars.", count.get());
});
testThread.start();
return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
}
Aggregations