Add UT for SouthboundMapper and SouthboundProvider
[netvirt.git] / openstack / net-virt / src / main / java / org / opendaylight / ovsdb / openstack / netvirt / impl / ProviderNetworkManagerImpl.java
1 /*
2  * Copyright (c) 2013, 2015 Red Hat, 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
9 package org.opendaylight.ovsdb.openstack.netvirt.impl;
10
11 import java.util.HashMap;
12 import java.util.Map;
13
14 import org.opendaylight.ovsdb.openstack.netvirt.ConfigInterface;
15 import org.opendaylight.ovsdb.openstack.netvirt.api.Constants;
16 import org.opendaylight.ovsdb.openstack.netvirt.api.NetworkingProvider;
17 import org.opendaylight.ovsdb.openstack.netvirt.api.NetworkingProviderManager;
18 import org.opendaylight.ovsdb.openstack.netvirt.api.OvsdbInventoryService;
19 import org.opendaylight.ovsdb.utils.servicehelper.ServiceHelper;
20 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
21 import org.osgi.framework.BundleContext;
22 import org.osgi.framework.ServiceReference;
23 import org.slf4j.Logger;
24 import org.slf4j.LoggerFactory;
25
26 import com.google.common.base.Predicate;
27 import com.google.common.collect.Iterables;
28 import com.google.common.collect.Maps;
29
30 public class ProviderNetworkManagerImpl implements ConfigInterface, NetworkingProviderManager {
31     private static final Logger LOG = LoggerFactory.getLogger(ProviderNetworkManagerImpl.class);
32     private Map<Long, ProviderEntry> providers = Maps.newHashMap();
33     private Map<Node, NetworkingProvider> nodeToProviderMapping = Maps.newHashMap();
34     private volatile OvsdbInventoryService ovsdbInventoryService;
35
36     @Override
37     public NetworkingProvider getProvider(Node node) {
38         if (nodeToProviderMapping.get(node) != null) {
39             return nodeToProviderMapping.get(node);
40         }
41
42         final String targetVersion = Constants.OPENFLOW13;
43         Predicate<ProviderEntry> providerEntryPredicate = new Predicate<ProviderEntry>() {
44             @Override
45             public boolean apply(ProviderEntry providerEntry) {
46                 //ToDo: This should match on southboundProtocol and providerType too
47                 return providerEntry.getProperties().get(Constants.OPENFLOW_VERSION_PROPERTY).equals(targetVersion);
48             }
49         };
50
51         Iterable<ProviderEntry> matchingProviders = Iterables.filter(providers.values(), providerEntryPredicate);
52         if (!matchingProviders.iterator().hasNext()) {
53             LOG.error("No providers matching {} found", targetVersion);
54         }
55
56         // Return the first match as only have one matching provider today
57         // ToDo: Tie-breaking logic
58         NetworkingProvider provider = matchingProviders.iterator().next().getProvider();
59         nodeToProviderMapping.put(node, provider);
60         return provider;
61     }
62
63     public void providerAdded(final ServiceReference ref, final NetworkingProvider provider){
64         Map <String, String> properties = Maps.newHashMap();
65         Long pid = (Long) ref.getProperty(org.osgi.framework.Constants.SERVICE_ID);
66         properties.put(Constants.SOUTHBOUND_PROTOCOL_PROPERTY,
67                 (String) ref.getProperty(Constants.SOUTHBOUND_PROTOCOL_PROPERTY));
68         properties.put(Constants.OPENFLOW_VERSION_PROPERTY,
69                 (String) ref.getProperty(Constants.OPENFLOW_VERSION_PROPERTY));
70         properties.put(Constants.PROVIDER_TYPE_PROPERTY, (String) ref.getProperty(Constants.PROVIDER_TYPE_PROPERTY));
71         providers.put(pid, new ProviderEntry(provider, properties));
72         LOG.info("Neutron Networking Provider Registered: {}, with {} and pid={}",
73                 provider.getClass().getName(), properties.toString(), pid);
74
75         ovsdbInventoryService.providersReady();
76     }
77
78     public void providerRemoved(final ServiceReference ref){
79         Long pid = (Long)ref.getProperty(org.osgi.framework.Constants.SERVICE_ID);
80         providers.remove(pid);
81         LOG.info("Neutron Networking Provider Removed: {}", pid);
82     }
83
84     @Override
85     public void setDependencies(BundleContext bundleContext, ServiceReference serviceReference) {
86         ovsdbInventoryService =
87                 (OvsdbInventoryService) ServiceHelper.getGlobalInstance(OvsdbInventoryService.class, this);
88     }
89
90     @Override
91     public void setDependencies(Object impl) {
92
93     }
94
95     private class ProviderEntry {
96         NetworkingProvider provider;
97         Map<String, String> properties;
98
99         ProviderEntry(NetworkingProvider provider, Map<String, String> properties) {
100             this.provider = provider;
101             this.properties = properties;
102         }
103
104         public NetworkingProvider getProvider() {
105             return provider;
106         }
107
108         public Map<String, String> getProperties() {
109             return properties;
110         }
111     }
112
113 }