2 * Copyright (c) 2013 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.controller.netconf.confignetconfconnector.mapping.config;
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;
17 import javax.management.InstanceNotFoundException;
18 import javax.management.ObjectName;
19 import java.util.Collection;
20 import java.util.Collections;
24 public class ServiceRegistryWrapper {
26 private ServiceReferenceReadableRegistry configServiceRefRegistry;
28 private long suffix = 1;
30 public ServiceRegistryWrapper(ServiceReferenceReadableRegistry configServiceRefRegistry) {
31 this.configServiceRefRegistry = configServiceRefRegistry;
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);
41 return forQName.values().contains(on);
44 public ObjectName getByServiceAndRefName(String namespace, String serviceName, String refName) {
45 Map<String, Map<String, String>> serviceNameToRefNameToInstance = getMappedServices().get(namespace);
47 Preconditions.checkArgument(serviceNameToRefNameToInstance != null, "No serviceInstances mapped to " + namespace);
49 Map<String, String> refNameToInstance = serviceNameToRefNameToInstance.get(serviceName);
50 Preconditions.checkArgument(refNameToInstance != null, "No serviceInstances mapped to " + serviceName + " , "
51 + serviceNameToRefNameToInstance.keySet());
53 String instanceId = refNameToInstance.get(refName);
54 Preconditions.checkArgument(instanceId != null, "No serviceInstances mapped to " + serviceName + ":"
55 + refName + ", " + serviceNameToRefNameToInstance.keySet());
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());
61 String qNameOfService = configServiceRefRegistry.getServiceInterfaceName(namespace, serviceName);
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);
71 public Map<String, Map<String, Map<String, String>>> getMappedServices() {
72 Map<String, Map<String, Map<String, String>>> retVal = Maps.newHashMap();
74 Map<String, Map<String, ObjectName>> serviceMapping = configServiceRefRegistry.getServiceMapping();
75 for (String serviceQName : serviceMapping.keySet()){
76 for (String refName : serviceMapping.get(serviceQName).keySet()) {
78 ObjectName on = serviceMapping.get(serviceQName).get(refName);
79 Services.ServiceInstance si = Services.ServiceInstance.fromObjectName(on);
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);
89 String localName = qname.getLocalName();
90 Map<String, String> refsToSis = serviceToRefs.get(localName);
92 refsToSis = Maps.newHashMap();
93 serviceToRefs.put(localName, refsToSis);
96 Preconditions.checkState(!refsToSis.containsKey(refName),
97 "Duplicate reference name %s for service %s:%s, now for instance %s", refName, namespace,
99 refsToSis.put(refName, si.toString());
107 public String getNewDefaultRefName(String namespace, String serviceName, String moduleName, String instanceName) {
109 refName = "ref_" + instanceName;
111 Map<String, Map<String, String>> serviceNameToRefNameToInstance = getMappedServices().get(namespace);
113 Map<String, String> refNameToInstance;
114 if(serviceNameToRefNameToInstance == null || !serviceNameToRefNameToInstance.containsKey(serviceName)) {
115 refNameToInstance = Collections.emptyMap();
117 refNameToInstance = serviceNameToRefNameToInstance.get(serviceName);
120 final Set<String> refNamesAsSet = toSet(refNameToInstance.keySet());
121 if (refNamesAsSet.contains(refName)) {
122 refName = findAvailableRefName(refName, refNamesAsSet);
129 private Set<String> toSet(Collection<String> values) {
130 Set<String> refNamesAsSet = Sets.newHashSet();
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);
138 return refNamesAsSet;
141 private String findAvailableRefName(String refName, Set<String> refNamesAsSet) {
142 String availableRefName = "";
145 availableRefName = refName + "_" + suffix++;
146 if (!refNamesAsSet.contains(availableRefName)){
147 return availableRefName;