+ StopStressTestOutputBuilder stopStressTestOutput;
+ stopStressTestOutput = new StopStressTestOutputBuilder()
+ .setSuccessCount(succcessCounter.longValue())
+ .setFailureCount(failureCounter.longValue());
+
+ final StopStressTestOutput result = stopStressTestOutput.build();
+ LOG_PURCHASE_CAR.info("Executed Stop Stress test; No. of cars created {}; "
+ + "No. of cars failed {}; ", succcessCounter, failureCounter);
+ // clear counters
+ succcessCounter.set(0);
+ failureCounter.set(0);
+ return Futures.immediateFuture(RpcResultBuilder.<StopStressTestOutput>success(result).build());
+ }
+
+
+ @Override
+ public ListenableFuture<RpcResult<RegisterOwnershipOutput>> registerOwnership(final RegisterOwnershipInput input) {
+ if (registeredListener.compareAndSet(false, true)) {
+ ownershipService.registerListener(ENTITY_TYPE, ownershipListener);
+ }
+
+ Entity entity = new Entity(ENTITY_TYPE, input.getCarId());
+ try {
+ ownershipService.registerCandidate(entity);
+ } catch (CandidateAlreadyRegisteredException e) {
+ return RpcResultBuilder.<RegisterOwnershipOutput>failed().withError(ErrorType.APPLICATION,
+ "Could not register for car " + input.getCarId(), e).buildFuture();
+ }
+
+ return RpcResultBuilder.success(new RegisterOwnershipOutputBuilder().build()).buildFuture();
+ }
+
+ @Override
+ public ListenableFuture<RpcResult<UnregisterOwnershipOutput>> unregisterOwnership(
+ final UnregisterOwnershipInput input) {
+ return RpcResultBuilder.success(new UnregisterOwnershipOutputBuilder().build()).buildFuture();
+ }
+
+ private static class CarEntityOwnershipListener implements EntityOwnershipListener {
+ @Override
+ public void ownershipChanged(final EntityOwnershipChange ownershipChange) {
+ LOG_CAR_PROVIDER.info("ownershipChanged: {}", ownershipChange);
+ }
+ }
+
+ @Override
+ public ListenableFuture<RpcResult<RegisterLoggingDtclOutput>> registerLoggingDtcl(
+ final RegisterLoggingDtclInput input) {
+ LOG_CAR_PROVIDER.info("Registering a new CarDataTreeChangeListener");
+ final ListenerRegistration<CarDataTreeChangeListener> carsDtclRegistration =
+ dataProvider.registerDataTreeChangeListener(CARS_DTID, new CarDataTreeChangeListener());
+
+ carsDtclRegistrations.add(carsDtclRegistration);
+ return RpcResultBuilder.success(new RegisterLoggingDtclOutputBuilder().build()).buildFuture();
+ }
+
+ @Override
+ public ListenableFuture<RpcResult<UnregisterLoggingDtclsOutput>> unregisterLoggingDtcls(
+ final UnregisterLoggingDtclsInput input) {
+ LOG_CAR_PROVIDER.info("Unregistering the CarDataTreeChangeListener(s)");
+ synchronized (carsDtclRegistrations) {
+ int numListeners = 0;
+ for (ListenerRegistration<CarDataTreeChangeListener> carsDtclRegistration : carsDtclRegistrations) {
+ carsDtclRegistration.close();
+ numListeners++;
+ }
+ carsDtclRegistrations.clear();
+ LOG_CAR_PROVIDER.info("Unregistered {} CaraDataTreeChangeListener(s)", numListeners);
+ }
+ return RpcResultBuilder.success(new UnregisterLoggingDtclsOutputBuilder().build()).buildFuture();
+ }
+
+ @Override
+ @SuppressWarnings("checkstyle:IllegalCatch")
+ public ListenableFuture<RpcResult<UnregisterCommitCohortOutput>> unregisterCommitCohort(
+ final UnregisterCommitCohortInput input) {
+ closeCommitCohortRegistration();
+
+ return RpcResultBuilder.success(new UnregisterCommitCohortOutputBuilder().build()).buildFuture();
+ }
+
+ private void closeCommitCohortRegistration() {
+ final DOMDataTreeCommitCohortRegistration<CarEntryDataTreeCommitCohort> reg = commitCohortReg.getAndSet(null);
+ if (reg != null) {
+ reg.close();
+ LOG_CAR_PROVIDER.info("Unregistered commit cohort");
+ }