Adjust for Binding RPC codegen changes
[controller.git] / benchmark / rpcbenchmark / src / main / java / rpcbenchmark / impl / RpcbenchmarkProvider.java
1 /*
2  * Copyright (c) 2015 Cisco Systems Inc. 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 rpcbenchmark.impl;
9
10 import com.google.common.util.concurrent.ListenableFuture;
11 import java.util.ArrayList;
12 import java.util.List;
13 import java.util.concurrent.ExecutorService;
14 import java.util.concurrent.Executors;
15 import java.util.concurrent.TimeUnit;
16 import java.util.concurrent.atomic.AtomicReference;
17 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.RoutedRpcRegistration;
18 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
19 import org.opendaylight.yang.gen.v1.rpcbench.payload.rev150702.NodeContext;
20 import org.opendaylight.yang.gen.v1.rpcbench.payload.rev150702.RpcbenchPayloadService;
21 import org.opendaylight.yang.gen.v1.rpcbench.payload.rev150702.RpcbenchRpcRoutes;
22 import org.opendaylight.yang.gen.v1.rpcbench.payload.rev150702.rpcbench.rpc.routes.RpcRoute;
23 import org.opendaylight.yang.gen.v1.rpcbench.payload.rev150702.rpcbench.rpc.routes.RpcRouteKey;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rpcbenchmark.rev150702.RpcbenchmarkService;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rpcbenchmark.rev150702.StartTestInput;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rpcbenchmark.rev150702.StartTestOutput;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rpcbenchmark.rev150702.StartTestOutputBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rpcbenchmark.rev150702.TestStatusInput;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rpcbenchmark.rev150702.TestStatusOutput;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rpcbenchmark.rev150702.TestStatusOutput.ExecStatus;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rpcbenchmark.rev150702.TestStatusOutputBuilder;
32 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
33 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
34 import org.opendaylight.yangtools.yang.common.RpcResult;
35 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
36 import org.slf4j.Logger;
37 import org.slf4j.LoggerFactory;
38
39 public class RpcbenchmarkProvider implements AutoCloseable, RpcbenchmarkService {
40
41     private static final Logger LOG = LoggerFactory.getLogger(RpcbenchmarkProvider.class);
42     private static final int testTimeout = 5;
43
44     private final GlobalBindingRTCServer globalServer;
45     private final AtomicReference<ExecStatus> execStatus = new AtomicReference<>(ExecStatus.Idle);
46     private final RpcProviderRegistry providerRegistry;
47
48     public RpcbenchmarkProvider(final RpcProviderRegistry providerRegistry, final GlobalBindingRTCServer globalServer) {
49         this.providerRegistry = providerRegistry;
50         this.globalServer = globalServer;
51     }
52
53     public void init() {
54         LOG.info("RpcbenchmarkProvider initiated");
55     }
56
57     @Override
58     public void close() {
59         LOG.info("RpcbenchmarkProvider closed");
60     }
61
62     @Override
63     public ListenableFuture<RpcResult<StartTestOutput>> startTest(final StartTestInput input) {
64         LOG.debug("startTest {}", input);
65
66         final RTCClient client;
67         final List<RoutedRpcRegistration<?>> rpcRegs = new ArrayList<>();
68
69         switch (input.getOperation()) {
70         case ROUTEDRTC:
71             List<InstanceIdentifier<?>> routeIid = new ArrayList<>();
72             for (int i = 0; i < input.getNumServers().intValue(); i++) {
73                 GlobalBindingRTCServer server = new GlobalBindingRTCServer();
74                 RoutedRpcRegistration<RpcbenchPayloadService> routedReg =
75                         providerRegistry.addRoutedRpcImplementation(RpcbenchPayloadService.class, server);
76
77                 KeyedInstanceIdentifier<RpcRoute, RpcRouteKey> iid =
78                         InstanceIdentifier
79                             .create(RpcbenchRpcRoutes.class)
80                             .child(RpcRoute.class, new RpcRouteKey(Integer.toString(i)));
81                 routeIid.add(iid);
82                 routedReg.registerPath(NodeContext.class, iid);
83                 rpcRegs.add(routedReg);
84             }
85
86             client = new RoutedBindingRTClient(providerRegistry, input.getPayloadSize().intValue(), routeIid);
87             break;
88
89         case GLOBALRTC:
90             client = new GlobalBindingRTCClient(providerRegistry, input.getPayloadSize().intValue());
91             break;
92
93         default:
94             LOG.error("Unsupported server/client type {}", input.getOperation());
95             throw new IllegalArgumentException("Unsupported server/client type" + input.getOperation());
96         }
97
98         try {
99             ExecutorService executor = Executors.newFixedThreadPool(input.getNumClients().intValue());
100
101             final Runnable testRun = () -> client.runTest(input.getIterations().intValue());
102
103             LOG.info("Test Started");
104             long startTime = System.nanoTime();
105
106             for (int i = 0; i < input.getNumClients().intValue(); i++ ) {
107                 executor.submit(testRun);
108             }
109
110             executor.shutdown();
111             try {
112                 executor.awaitTermination(testTimeout, TimeUnit.MINUTES);
113             } catch (final InterruptedException e) {
114                 LOG.error("Out of time: test did not finish within the {} min deadline ", testTimeout);
115             }
116
117             long endTime = System.nanoTime();
118             LOG.info("Test Done");
119
120             long elapsedTime = endTime - startTime;
121
122             StartTestOutput output = new StartTestOutputBuilder()
123                                             .setRate((long)0)
124                                             .setGlobalRtcClientError(client.getRpcError())
125                                             .setGlobalRtcClientOk(client.getRpcOk())
126                                             .setExecTime(TimeUnit.NANOSECONDS.toMillis(elapsedTime))
127                                             .setRate((client.getRpcOk() + client.getRpcError()) * 1000000000 / elapsedTime)
128                                             .build();
129             return RpcResultBuilder.success(output).buildFuture();
130         } finally {
131             for (RoutedRpcRegistration<?> routedRpcRegistration : rpcRegs) {
132                 routedRpcRegistration.close();
133             }
134         }
135     }
136
137     @Override
138     public ListenableFuture<RpcResult<TestStatusOutput>> testStatus(final TestStatusInput input) {
139         LOG.info("testStatus");
140         TestStatusOutput output = new TestStatusOutputBuilder()
141                                         .setGlobalServerCnt((long)globalServer.getNumRpcs())
142                                         .setExecStatus(execStatus.get())
143                                         .build();
144         return RpcResultBuilder.success(output).buildFuture();
145     }
146
147 }

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