opendaylight/netconf/config-netconf-connector/src/main/java/org/opendaylight/controller/netconf/confignetconfconnector/mapping/config/ServiceRegistryWrapper.java

   1 /*
   2  * Copyright (c) 2013 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.controller.netconf.confignetconfconnector.mapping.config;
   9
  10 import com.google.common.annotations.VisibleForTesting;
  11 import com.google.common.base.Preconditions;
  12 import com.google.common.collect.Maps;
  13 import com.google.common.collect.Sets;
  14 import org.opendaylight.controller.config.api.ServiceReferenceReadableRegistry;
  15 import org.opendaylight.yangtools.yang.common.QName;
  16
  17 import javax.management.InstanceNotFoundException;
  18 import javax.management.ObjectName;
  19 import java.util.Collection;
  20 import java.util.Collections;
  21 import java.util.Map;
  22 import java.util.Set;
  23
  24 public class ServiceRegistryWrapper {
  25
  26     private ServiceReferenceReadableRegistry configServiceRefRegistry;
  27
  28     private long suffix = 1;
  29
  30     public ServiceRegistryWrapper(ServiceReferenceReadableRegistry configServiceRefRegistry) {
  31         this.configServiceRefRegistry = configServiceRefRegistry;
  32     }
  33
  34
  35     public boolean hasRefName(String namespace, String serviceName, ObjectName on) {
  36         String qname = configServiceRefRegistry.getServiceInterfaceName(namespace, serviceName);
  37         Map<String, ObjectName> forQName = configServiceRefRegistry.getServiceMapping().get(qname);
  38         if(forQName==null){
  39             return false;
  40         }
  41         return forQName.values().contains(on);
  42     }
  43
  44     public ObjectName getByServiceAndRefName(String namespace, String serviceName, String refName) {
  45         Map<String, Map<String, String>> serviceNameToRefNameToInstance = getMappedServices().get(namespace);
  46
  47         Preconditions.checkArgument(serviceNameToRefNameToInstance != null, "No serviceInstances mapped to " + namespace);
  48
  49         Map<String, String> refNameToInstance = serviceNameToRefNameToInstance.get(serviceName);
  50         Preconditions.checkArgument(refNameToInstance != null, "No serviceInstances mapped to " + serviceName + " , "
  51                 + serviceNameToRefNameToInstance.keySet());
  52
  53         String instanceId = refNameToInstance.get(refName);
  54         Preconditions.checkArgument(instanceId != null, "No serviceInstances mapped to " + serviceName + ":"
  55                 + refName + ", " + serviceNameToRefNameToInstance.keySet());
  56
  57         Services.ServiceInstance serviceInstance = Services.ServiceInstance.fromString(instanceId);
  58         Preconditions.checkArgument(serviceInstance != null, "No serviceInstance mapped to " + refName
  59                 + " under service name " + serviceName + " , " + refNameToInstance.keySet());
  60
  61         String qNameOfService = configServiceRefRegistry.getServiceInterfaceName(namespace, serviceName);
  62         try {
  63             return configServiceRefRegistry.getServiceReference(qNameOfService, refName);
  64         } catch (InstanceNotFoundException e) {
  65             throw new IllegalArgumentException("No serviceInstance mapped to " + refName
  66                     + " under service name " + serviceName + " , " + refNameToInstance.keySet(), e);
  67
  68         }
  69     }
  70
  71     public Map<String, Map<String, Map<String, String>>> getMappedServices() {
  72         Map<String, Map<String, Map<String, String>>> retVal = Maps.newHashMap();
  73
  74         Map<String, Map<String, ObjectName>> serviceMapping = configServiceRefRegistry.getServiceMapping();
  75         for (String serviceQName : serviceMapping.keySet()){
  76             for (String refName : serviceMapping.get(serviceQName).keySet()) {
  77
  78                 ObjectName on = serviceMapping.get(serviceQName).get(refName);
  79                 Services.ServiceInstance si = Services.ServiceInstance.fromObjectName(on);
  80
  81                 QName qname = QName.create(serviceQName);
  82                 String namespace = qname.getNamespace().toString();
  83                 Map<String, Map<String, String>> serviceToRefs = retVal.get(namespace);
  84                 if(serviceToRefs==null) {
  85                     serviceToRefs = Maps.newHashMap();
  86                     retVal.put(namespace, serviceToRefs);
  87                 }
  88
  89                 String localName = qname.getLocalName();
  90                 Map<String, String> refsToSis = serviceToRefs.get(localName);
  91                 if(refsToSis==null) {
  92                     refsToSis = Maps.newHashMap();
  93                     serviceToRefs.put(localName, refsToSis);
  94                 }
  95
  96                 Preconditions.checkState(!refsToSis.containsKey(refName),
  97                         "Duplicate reference name %s for service %s:%s, now for instance %s", refName, namespace,
  98                         localName, on);
  99                 refsToSis.put(refName, si.toString());
 100             }
 101         }
 102
 103         return retVal;
 104     }
 105
 106     @VisibleForTesting
 107     public String getNewDefaultRefName(String namespace, String serviceName, String moduleName, String instanceName) {
 108         String refName;
 109         refName = "ref_" + instanceName;
 110
 111         Map<String, Map<String, String>> serviceNameToRefNameToInstance = getMappedServices().get(namespace);
 112
 113         Map<String, String> refNameToInstance;
 114         if(serviceNameToRefNameToInstance == null || !serviceNameToRefNameToInstance.containsKey(serviceName)) {
 115             refNameToInstance = Collections.emptyMap();
 116         } else {
 117             refNameToInstance = serviceNameToRefNameToInstance.get(serviceName);
 118         }
 119
 120         final Set<String> refNamesAsSet = toSet(refNameToInstance.keySet());
 121         if (refNamesAsSet.contains(refName)) {
 122             refName = findAvailableRefName(refName, refNamesAsSet);
 123         }
 124
 125         return refName;
 126     }
 127
 128
 129     private Set<String> toSet(Collection<String> values) {
 130         Set<String> refNamesAsSet = Sets.newHashSet();
 131
 132         for (String refName : values) {
 133             boolean resultAdd = refNamesAsSet.add(refName);
 134             Preconditions.checkState(resultAdd,
 135                     "Error occurred building services element, reference name {} was present twice", refName);
 136         }
 137
 138         return refNamesAsSet;
 139     }
 140
 141     private String findAvailableRefName(String refName, Set<String> refNamesAsSet) {
 142         String availableRefName = "";
 143
 144         while (true) {
 145             availableRefName = refName + "_" + suffix++;
 146             if (!refNamesAsSet.contains(availableRefName)){
 147                 return availableRefName;
 148             }
 149         }
 150     }
 151 }