benchmark/ntfbenchmark/src/main/java/ntfbenchmark/impl/NtfbenchNonblockingProducer.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package ntfbenchmark.impl;
  10
  11 import com.google.common.util.concurrent.ListenableFuture;
  12 import com.google.common.util.concurrent.SettableFuture;
  13 import java.util.concurrent.ExecutionException;
  14 import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
  15
  16 public class NtfbenchNonblockingProducer extends AbstractNtfbenchProducer {
  17
  18     private final SettableFuture<?> lastDeliveryFuture = SettableFuture.create();
  19
  20
  21     public NtfbenchNonblockingProducer(final NotificationPublishService publishService, final int iterations,
  22             final int payloadSize) {
  23         super(publishService, iterations, payloadSize);
  24     }
  25
  26
  27     @Override
  28     public void run() {
  29         int ntfOk = 0;
  30         int ntfError = 0;
  31         ListenableFuture<?> lastOkFuture = null;
  32         for (int i = 0; i < this.iterations; i++) {
  33             try {
  34                 final ListenableFuture<?> result = this.publishService.offerNotification(this.ntf);
  35                 if (NotificationPublishService.REJECTED == result) {
  36                     ntfError++;
  37                 } else {
  38                     ntfOk++;
  39                     lastOkFuture = result;
  40                 }
  41             } catch (final Exception e) {
  42                 ntfError++;
  43             }
  44         }
  45
  46         this.ntfOk = ntfOk;
  47         this.ntfError = ntfError;
  48         // We wait for last notification to be delivered to listeners.
  49         if (lastOkFuture != null) {
  50             try {
  51                 lastOkFuture.get();
  52             } catch (InterruptedException | ExecutionException e) {
  53                 throw new RuntimeException(e);
  54             }
  55         }
  56     }
  57
  58 }