2 * Copyright (c) 2018 Orange and/or its affiliates.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at:
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
16 package io.fd.honeycomb.transportpce.device.read;
18 import com.google.common.base.Function;
19 import com.google.common.base.Optional;
20 import com.google.common.collect.ArrayListMultimap;
21 import com.google.common.collect.Collections2;
22 import com.google.common.collect.Lists;
23 import com.google.common.collect.Sets;
24 import java.util.Collections;
26 import javax.annotation.Nonnull;
27 import javax.annotation.Nullable;
28 import org.opendaylight.controller.config.util.capability.Capability;
29 import org.opendaylight.netconf.api.monitoring.NetconfManagementSession;
30 import org.opendaylight.netconf.api.monitoring.NetconfMonitoringService;
31 import org.opendaylight.netconf.api.monitoring.SessionEvent;
32 import org.opendaylight.netconf.api.monitoring.SessionListener;
33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.Yang;
35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Capabilities;
36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.CapabilitiesBuilder;
37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Schemas;
38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.SchemasBuilder;
39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Sessions;
40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.SessionsBuilder;
41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema;
42 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema.Location;
43 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema.Location.Enumeration;
44 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.SchemaBuilder;
45 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.SchemaKey;
47 public class DummyMonitoringService implements NetconfMonitoringService {
49 private static final Sessions EMPTY_SESSIONS = new SessionsBuilder().setSession(Collections.emptyList()).build();
50 private static final Function<Capability, Uri> CAPABILITY_URI_FUNCTION =
51 capability -> new Uri(capability.getCapabilityUri());
53 private static final Function<Capability, Schema> CAPABILITY_SCHEMA_FUNCTION = new Function<Capability, Schema>() {
56 public Schema apply(@Nonnull final Capability capability) {
57 return new SchemaBuilder()
58 .setIdentifier(capability.getModuleName().get())
59 .setNamespace(new Uri(capability.getModuleNamespace().get()))
60 .setFormat(Yang.class)
61 .setVersion(capability.getRevision().get())
62 .setLocation(Collections.singletonList(new Location(Enumeration.NETCONF)))
63 .setKey(new SchemaKey(Yang.class, capability.getModuleName().get(),
64 capability.getRevision().get()))
69 private final Capabilities capabilities;
70 private final ArrayListMultimap<String, Capability> capabilityMultiMap;
71 private final Schemas schemas;
73 public DummyMonitoringService(final Set<Capability> capabilities) {
75 this.capabilities = new CapabilitiesBuilder().setCapability(
76 Lists.newArrayList(Collections2.transform(capabilities, CAPABILITY_URI_FUNCTION))).build();
78 Set<Capability> moduleCapabilities = Sets.newHashSet();
79 this.capabilityMultiMap = ArrayListMultimap.create();
80 for (Capability cap : capabilities) {
81 if (cap.getModuleName().isPresent()) {
82 capabilityMultiMap.put(cap.getModuleName().get(), cap);
83 moduleCapabilities.add(cap);
87 this.schemas = new SchemasBuilder().setSchema(
88 Lists.newArrayList(Collections2.transform(moduleCapabilities, CAPABILITY_SCHEMA_FUNCTION))).build();
92 public Sessions getSessions() {
93 return EMPTY_SESSIONS;
97 public SessionListener getSessionListener() {
98 return new SessionListener() {
100 public void onSessionUp(final NetconfManagementSession session) {
105 public void onSessionDown(final NetconfManagementSession session) {
110 public void onSessionEvent(final SessionEvent event) {
117 public Schemas getSchemas() {
122 public String getSchemaForCapability(final String moduleName, final Optional<String> revision) {
124 for (Capability capability : capabilityMultiMap.get(moduleName)) {
125 if (capability.getRevision().get().equals(revision.get())) {
126 return capability.getCapabilitySchema().get();
129 throw new IllegalArgumentException(
130 "Module with name: " + moduleName + " and revision: " + revision + " does not exist");
134 public Capabilities getCapabilities() {
139 public AutoCloseable registerCapabilitiesListener(final CapabilitiesListener listener) {
144 public AutoCloseable registerSessionsListener(final SessionsListener listener) {