c6e2e8affb2af60bb4f88820498a0546a876439e
[controller.git] / 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 package ntfbenchmark.impl;
9
10 import com.google.common.util.concurrent.ListenableFuture;
11 import com.google.common.util.concurrent.SettableFuture;
12 import java.util.concurrent.ExecutionException;
13 import org.opendaylight.mdsal.binding.api.NotificationPublishService;
14
15 public class NtfbenchNonblockingProducer extends AbstractNtfbenchProducer {
16
17     private final SettableFuture<?> lastDeliveryFuture = SettableFuture.create();
18
19
20     public NtfbenchNonblockingProducer(final NotificationPublishService publishService, final int iterations,
21             final int payloadSize) {
22         super(publishService, iterations, payloadSize);
23     }
24
25
26     @Override
27     @SuppressWarnings("checkstyle:illegalCatch")
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 }

©2013 OpenDaylight, A Linux Foundation Collaborative Project. All Rights Reserved.
OpenDaylight is a registered trademark of The OpenDaylight Project, Inc.
Linux Foundation and OpenDaylight are registered trademarks of the Linux Foundation.
Linux is a registered trademark of Linus Torvalds.