X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?p=controller.git;a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsamples%2Fclustering-test-app%2Fprovider%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fclustering%2Fit%2Fprovider%2FPurchaseCarProvider.java;h=055dc6ac62acbfc5d017217abff5ee2ec119409b;hp=74a0aa68ed91dcac50dbb3714421ffd77f70efe8;hb=91eca0854c2ebe344b69bb0e0a1f992827267f01;hpb=4019c9fd2ad99628dd790acc1ad4c104f48b6428

diff --git a/opendaylight/md-sal/samples/clustering-test-app/provider/src/main/java/org/opendaylight/controller/clustering/it/provider/PurchaseCarProvider.java b/opendaylight/md-sal/samples/clustering-test-app/provider/src/main/java/org/opendaylight/controller/clustering/it/provider/PurchaseCarProvider.java
index 74a0aa68ed..055dc6ac62 100644
--- a/opendaylight/md-sal/samples/clustering-test-app/provider/src/main/java/org/opendaylight/controller/clustering/it/provider/PurchaseCarProvider.java
+++ b/opendaylight/md-sal/samples/clustering-test-app/provider/src/main/java/org/opendaylight/controller/clustering/it/provider/PurchaseCarProvider.java
@@ -5,12 +5,17 @@
  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
  * and is available at http://www.eclipse.org/legal/epl-v10.html
  */
-
 package org.opendaylight.controller.clustering.it.provider;
 
-import com.google.common.util.concurrent.SettableFuture;
-import org.opendaylight.controller.sal.binding.api.NotificationProviderService;
+import static java.util.Objects.requireNonNull;
+
+import com.google.common.util.concurrent.Futures;
+import com.google.common.util.concurrent.ListenableFuture;
+import com.google.common.util.concurrent.MoreExecutors;
+import org.opendaylight.mdsal.binding.api.NotificationPublishService;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.sal.clustering.it.car.purchase.rev140818.BuyCarInput;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.sal.clustering.it.car.purchase.rev140818.BuyCarOutput;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.sal.clustering.it.car.purchase.rev140818.BuyCarOutputBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.sal.clustering.it.car.purchase.rev140818.CarBoughtBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.sal.clustering.it.car.purchase.rev140818.CarPurchaseService;
 import org.opendaylight.yangtools.yang.common.RpcResult;
@@ -18,35 +23,29 @@ import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-import java.util.concurrent.Future;
-
-
-public class PurchaseCarProvider implements CarPurchaseService, AutoCloseable{
-
-  private static final Logger log = LoggerFactory.getLogger(PurchaseCarProvider.class);
-
-  private NotificationProviderService notificationProvider;
+public class PurchaseCarProvider implements CarPurchaseService, AutoCloseable {
+    private static final Logger LOG = LoggerFactory.getLogger(PurchaseCarProvider.class);
 
+    private final NotificationPublishService notificationProvider;
 
-  public void setNotificationProvider(final NotificationProviderService salService) {
-    this.notificationProvider = salService;
-  }
+    public PurchaseCarProvider(final NotificationPublishService notificationProvider) {
+        this.notificationProvider = requireNonNull(notificationProvider);
+    }
 
+    @Override
+    public ListenableFuture<RpcResult<BuyCarOutput>> buyCar(final BuyCarInput input) {
+        LOG.info("Routed RPC buyCar : generating notification for buying car [{}]", input);
 
-  @Override
-  public Future<RpcResult<Void>> buyCar(BuyCarInput input) {
-    log.info("Routed RPC buyCar : generating notification for buying car [{}]", input);
-    SettableFuture<RpcResult<Void>> futureResult = SettableFuture.create();
-    CarBoughtBuilder carBoughtBuilder = new CarBoughtBuilder();
-    carBoughtBuilder.setCarId(input.getCarId());
-    carBoughtBuilder.setPersonId(input.getPersonId());
-    notificationProvider.publish(carBoughtBuilder.build());
-    futureResult.set(RpcResultBuilder.<Void>success().build());
-    return futureResult;
-  }
+        return Futures.transform(notificationProvider.offerNotification(new CarBoughtBuilder()
+            .setCarId(input.getCarId())
+            .setPersonId(input.getPersonId())
+            .build()),
+            result -> RpcResultBuilder.success(new BuyCarOutputBuilder().build()).build(),
+            MoreExecutors.directExecutor());
+    }
 
-  @Override
-  public void close() throws Exception {
+    @Override
+    public void close() {
 
-  }
+    }
 }