Encapsulate OpenFlowPlugin configuration
[openflowplugin.git] / openflowplugin-impl / src / main / java / org / opendaylight / openflowplugin / impl / rpc / RpcManagerImpl.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 org.opendaylight.openflowplugin.impl.rpc;
9
10 import com.google.common.annotations.VisibleForTesting;
11 import com.google.common.collect.Iterators;
12 import java.util.Iterator;
13 import java.util.concurrent.ConcurrentHashMap;
14 import java.util.concurrent.ConcurrentMap;
15 import javax.annotation.Nonnull;
16 import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
17 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
18 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
19 import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
20 import org.opendaylight.openflowplugin.api.openflow.rpc.RpcContext;
21 import org.opendaylight.openflowplugin.api.openflow.rpc.RpcManager;
22 import org.opendaylight.openflowplugin.extension.api.core.extension.ExtensionConverterProvider;
23 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorExecutor;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.provider.config.rev160510.OpenflowProviderConfig;
25 import org.slf4j.Logger;
26 import org.slf4j.LoggerFactory;
27
28 public class RpcManagerImpl implements RpcManager {
29
30     private static final Logger LOG = LoggerFactory.getLogger(RpcManagerImpl.class);
31     private final OpenflowProviderConfig config;
32     private final RpcProviderRegistry rpcProviderRegistry;
33     private final ConcurrentMap<DeviceInfo, RpcContext> contexts = new ConcurrentHashMap<>();
34     private final ExtensionConverterProvider extensionConverterProvider;
35     private final ConvertorExecutor convertorExecutor;
36     private final NotificationPublishService notificationPublishService;
37
38
39     public RpcManagerImpl(final OpenflowProviderConfig config,
40                           final RpcProviderRegistry rpcProviderRegistry,
41                           final ExtensionConverterProvider extensionConverterProvider,
42                           final ConvertorExecutor convertorExecutor,
43                           final NotificationPublishService notificationPublishService) {
44         this.config = config;
45         this.rpcProviderRegistry = rpcProviderRegistry;
46         this.extensionConverterProvider = extensionConverterProvider;
47         this.convertorExecutor = convertorExecutor;
48         this.notificationPublishService = notificationPublishService;
49     }
50
51     @Override
52     public void close() {
53         for (final Iterator<RpcContext> iterator = Iterators.consumingIterator(contexts.values().iterator());
54                 iterator.hasNext();) {
55             iterator.next().close();
56         }
57     }
58
59     /**
60      * This method is only for testing
61      */
62     @VisibleForTesting
63     void addRecordToContexts(DeviceInfo deviceInfo, RpcContext rpcContexts) {
64         if(!contexts.containsKey(deviceInfo)) {
65             this.contexts.put(deviceInfo,rpcContexts);
66         }
67     }
68
69     public RpcContext createContext(final @Nonnull DeviceInfo deviceInfo, final @Nonnull DeviceContext deviceContext) {
70         final RpcContextImpl rpcContext = new RpcContextImpl(
71                 rpcProviderRegistry,
72                 config.getRpcRequestsQuota().getValue(),
73                 deviceContext,
74                 extensionConverterProvider,
75                 convertorExecutor,
76                 notificationPublishService,
77                 config.isIsStatisticsRpcEnabled());
78
79         contexts.put(deviceInfo, rpcContext);
80         return rpcContext;
81     }
82
83     @Override
84     public void onDeviceRemoved(final DeviceInfo deviceInfo) {
85         contexts.remove(deviceInfo);
86         if (LOG.isDebugEnabled()) {
87             LOG.debug("Rpc context removed for node {}", deviceInfo.getLOGValue());
88         }
89     }
90 }