Adjust for Binding RPC codegen changes
[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
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 }

©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.