2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.openflowplugin.impl.rpc;
10 import com.google.common.util.concurrent.FutureCallback;
11 import com.google.common.util.concurrent.Futures;
12 import com.google.common.util.concurrent.ListenableFuture;
13 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ProviderContext;
14 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.RoutedRpcRegistration;
15 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
16 import org.opendaylight.openflowplugin.api.openflow.device.RequestContext;
17 import org.opendaylight.openflowplugin.api.openflow.rpc.RpcContext;
18 import org.opendaylight.yangtools.yang.binding.DataObject;
19 import org.opendaylight.yangtools.yang.binding.RpcService;
20 import org.opendaylight.yangtools.yang.common.RpcError;
21 import org.opendaylight.yangtools.yang.common.RpcResult;
22 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
23 import java.util.ArrayList;
24 import java.util.Collections;
25 import java.util.List;
26 import java.util.concurrent.Future;
28 public class RpcContextImpl implements RpcContext {
30 final ProviderContext providerContext;
32 // TODO: add private Sal salBroker
33 private final List<RequestContext> requestContexts = new ArrayList<>();
34 private DeviceContext deviceContext;
35 private final List<RoutedRpcRegistration> rpcRegistrations = new ArrayList<>();
36 private final List<RequestContext> synchronizedRequestsList = Collections.synchronizedList(new ArrayList<RequestContext>());
38 private int maxRequestsPerDevice;
40 public RpcContextImpl(final ProviderContext providerContext, final DeviceContext deviceContext) {
41 this.providerContext = providerContext;
42 this.deviceContext = deviceContext;
46 * @see org.opendaylight.openflowplugin.api.openflow.rpc.RpcContext#registerRpcServiceImplementation(java.lang.Class,
47 * org.opendaylight.yangtools.yang.binding.RpcService)
50 public <S extends RpcService> void registerRpcServiceImplementation(final Class<S> serviceClass,
51 final S serviceInstance) {
52 rpcRegistrations.add(providerContext.addRoutedRpcImplementation(serviceClass, serviceInstance));
56 public <T extends DataObject> Future<RpcResult<T>> addNewRequest(final DataObject data) {
57 final Future<RpcResult<T>> rpcResultFuture;
58 final RequestContext requestContext = new RequestContextImpl(this);
60 if (synchronizedRequestsList.size() < maxRequestsPerDevice) {
61 synchronizedRequestsList.add(requestContext);
62 rpcResultFuture = requestContext.createRequestFuture(data);
64 ListenableFuture<RpcResult<Void>> resultFutureFromDevice = sendRequestToDevice(data);
65 Futures.addCallback(resultFutureFromDevice, new FutureCallback<Object>() {
67 public void onSuccess(final Object o) {
68 requestContext.requestSucceeded();
72 public void onFailure(final Throwable throwable) {
73 requestContext.requestFailed(throwable.getCause().getMessage());
77 rpcResultFuture = Futures.immediateFuture(RpcResultBuilder.<T>failed().withError(RpcError.ErrorType.APPLICATION, "", "Request queue full.").build());
81 return rpcResultFuture;
84 private ListenableFuture<RpcResult<Void>> sendRequestToDevice(final DataObject data) {
85 //TODO : send data to device
90 * Unregisters all services.
92 * @see java.lang.AutoCloseable#close()
95 public void close() throws Exception {
96 for (final RoutedRpcRegistration rpcRegistration : rpcRegistrations) {
97 rpcRegistration.close();
102 * @see org.opendaylight.openflowplugin.api.openflow.rpc.RpcContext#setRequestContextQuota(int)
105 public void setRequestContextQuota(final int maxRequestsPerDevice) {
106 this.maxRequestsPerDevice = maxRequestsPerDevice;
110 public void forgetRequestContext(final RequestContext requestContext) {
111 requestContexts.remove(requestContext);
114 public boolean isRequestContextCapacityEmpty() {
115 return requestContexts.size() <= maxRequestsPerDevice;