Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.opendaylight.netconf.test.tool.MdsalOperationProvider Maven / Gradle / Ivy
/*
* Copyright (c) 2014 Cisco Systems, Inc. and others. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
package org.opendaylight.netconf.test.tool;
import static java.util.Objects.requireNonNull;
import com.google.common.util.concurrent.MoreExecutors;
import java.util.EnumMap;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
import org.opendaylight.mdsal.dom.api.DOMDataBroker;
import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
import org.opendaylight.mdsal.dom.api.DOMSchemaService;
import org.opendaylight.mdsal.dom.broker.SerializedDOMDataBroker;
import org.opendaylight.mdsal.dom.spi.store.DOMStore;
import org.opendaylight.mdsal.dom.store.inmemory.InMemoryDOMDataStoreFactory;
import org.opendaylight.netconf.server.api.SessionIdProvider;
import org.opendaylight.netconf.server.api.monitoring.Capability;
import org.opendaylight.netconf.server.api.monitoring.CapabilityListener;
import org.opendaylight.netconf.server.api.operations.NetconfOperation;
import org.opendaylight.netconf.server.api.operations.NetconfOperationService;
import org.opendaylight.netconf.server.api.operations.NetconfOperationServiceFactory;
import org.opendaylight.netconf.server.mdsal.CurrentSchemaContext;
import org.opendaylight.netconf.server.mdsal.TransactionProvider;
import org.opendaylight.netconf.server.mdsal.operations.Commit;
import org.opendaylight.netconf.server.mdsal.operations.DiscardChanges;
import org.opendaylight.netconf.server.mdsal.operations.EditConfig;
import org.opendaylight.netconf.server.mdsal.operations.Get;
import org.opendaylight.netconf.server.mdsal.operations.GetConfig;
import org.opendaylight.netconf.server.mdsal.operations.Lock;
import org.opendaylight.netconf.server.mdsal.operations.Unlock;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netconf.base._1._0.rev110601.SessionIdType;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.NetconfState;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.Yang;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Schemas;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema;
import org.opendaylight.yangtools.concepts.AbstractListenerRegistration;
import org.opendaylight.yangtools.concepts.ListenerRegistration;
import org.opendaylight.yangtools.concepts.Registration;
import org.opendaylight.yangtools.util.concurrent.SpecialExecutors;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeWithValue;
import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
import org.opendaylight.yangtools.yang.data.api.schema.LeafSetNode;
import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
import org.opendaylight.yangtools.yang.data.api.schema.SystemMapNode;
import org.opendaylight.yangtools.yang.data.api.schema.builder.CollectionNodeBuilder;
import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
import org.opendaylight.yangtools.yang.model.api.EffectiveModelContextListener;
import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceProvider;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
class MdsalOperationProvider implements NetconfOperationServiceFactory {
private static final Logger LOG = LoggerFactory.getLogger(MdsalOperationProvider.class);
private final Set caps;
private final EffectiveModelContext schemaContext;
private final SchemaSourceProvider sourceProvider;
MdsalOperationProvider(final SessionIdProvider idProvider,
final Set caps,
final EffectiveModelContext schemaContext,
final SchemaSourceProvider sourceProvider) {
this.caps = caps;
this.schemaContext = schemaContext;
this.sourceProvider = sourceProvider;
}
@Override
public Set getCapabilities() {
return caps;
}
@Override
public Registration registerCapabilityListener(final CapabilityListener listener) {
listener.onCapabilitiesChanged(caps, Set.of());
return () -> { };
}
@Override
public NetconfOperationService createService(final SessionIdType sessionId) {
return new MdsalOperationService(sessionId, schemaContext, caps, sourceProvider);
}
static class MdsalOperationService implements NetconfOperationService {
private final SessionIdType currentSessionId;
private final EffectiveModelContext schemaContext;
private final Set caps;
private final DOMSchemaService schemaService;
private final DOMDataBroker dataBroker;
private final SchemaSourceProvider sourceProvider;
MdsalOperationService(final SessionIdType currentSessionId, final EffectiveModelContext schemaContext,
final Set caps,
final SchemaSourceProvider sourceProvider) {
this.currentSessionId = requireNonNull(currentSessionId);
this.schemaContext = schemaContext;
this.caps = caps;
this.sourceProvider = sourceProvider;
schemaService = createSchemaService();
dataBroker = createDataStore(schemaService, currentSessionId);
}
@Override
public Set getNetconfOperations() {
ContainerNode netconf = createNetconfState();
final DOMDataTreeWriteTransaction tx = dataBroker.newWriteOnlyTransaction();
tx.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.of(NetconfState.QNAME), netconf);
try {
tx.commit().get();
LOG.debug("Netconf state updated successfully");
} catch (InterruptedException | ExecutionException e) {
LOG.warn("Unable to update netconf state", e);
}
TransactionProvider transactionProvider = new TransactionProvider(dataBroker, currentSessionId);
CurrentSchemaContext currentSchemaContext = CurrentSchemaContext.create(schemaService, sourceProvider);
final Get get = new Get(currentSessionId, currentSchemaContext, transactionProvider);
final EditConfig editConfig = new EditConfig(currentSessionId, currentSchemaContext, transactionProvider);
final GetConfig getConfig = new GetConfig(currentSessionId, currentSchemaContext, transactionProvider);
final Commit commit = new Commit(currentSessionId, transactionProvider);
final Lock lock = new Lock(currentSessionId);
final Unlock unLock = new Unlock(currentSessionId);
final DiscardChanges discardChanges = new DiscardChanges(currentSessionId, transactionProvider);
return Set.of(get, getConfig, editConfig, commit, lock, unLock, discardChanges);
}
@Override
public void close() {
// No-op
}
private ContainerNode createNetconfState() {
final DummyMonitoringService monitor = new DummyMonitoringService(caps);
final QName identifier = QName.create(Schema.QNAME, "identifier");
final QName version = QName.create(Schema.QNAME, "version");
final QName format = QName.create(Schema.QNAME, "format");
final QName location = QName.create(Schema.QNAME, "location");
final QName namespace = QName.create(Schema.QNAME, "namespace");
CollectionNodeBuilder schemaMapEntryNodeMapNodeCollectionNodeBuilder =
Builders.mapBuilder().withNodeIdentifier(new NodeIdentifier(Schema.QNAME));
LeafSetNode locationLeafSet = Builders.leafSetBuilder()
.withNodeIdentifier(new NodeIdentifier(location))
.withChild(Builders.leafSetEntryBuilder()
.withNodeIdentifier(new NodeWithValue<>(location, "NETCONF"))
.withValue("NETCONF")
.build())
.build();
Map keyValues = new HashMap<>();
for (final Schema schema : monitor.getSchemas().nonnullSchema().values()) {
keyValues.put(identifier, schema.getIdentifier());
keyValues.put(version, schema.getVersion());
keyValues.put(format, Yang.QNAME);
schemaMapEntryNodeMapNodeCollectionNodeBuilder.withChild(Builders.mapEntryBuilder()
.withNodeIdentifier(NodeIdentifierWithPredicates.of(Schema.QNAME, keyValues))
.withChild(ImmutableNodes.leafNode(identifier, schema.getIdentifier()))
.withChild(ImmutableNodes.leafNode(version, schema.getVersion()))
.withChild(ImmutableNodes.leafNode(format, Yang.QNAME))
.withChild(ImmutableNodes.leafNode(namespace, schema.getNamespace().getValue()))
.withChild(locationLeafSet)
.build());
}
return Builders.containerBuilder()
.withNodeIdentifier(new NodeIdentifier(NetconfState.QNAME))
.withChild(Builders.containerBuilder()
.withNodeIdentifier(new NodeIdentifier(Schemas.QNAME))
.withChild(schemaMapEntryNodeMapNodeCollectionNodeBuilder.build())
.build())
.build();
}
private static DOMDataBroker createDataStore(final DOMSchemaService schemaService,
final SessionIdType sessionId) {
LOG.debug("Session {}: Creating data stores for simulated device", sessionId.getValue());
final DOMStore operStore = InMemoryDOMDataStoreFactory.create("DOM-OPER", schemaService);
final DOMStore configStore = InMemoryDOMDataStoreFactory.create("DOM-CFG", schemaService);
ExecutorService listenableFutureExecutor = SpecialExecutors.newBlockingBoundedCachedThreadPool(16, 16,
"CommitFutures", MdsalOperationProvider.class);
final var datastores = new EnumMap(LogicalDatastoreType.class);
datastores.put(LogicalDatastoreType.CONFIGURATION, configStore);
datastores.put(LogicalDatastoreType.OPERATIONAL, operStore);
return new SerializedDOMDataBroker(datastores, MoreExecutors.listeningDecorator(listenableFutureExecutor));
}
private DOMSchemaService createSchemaService() {
return new DOMSchemaService() {
@Override
public EffectiveModelContext getGlobalContext() {
return schemaContext;
}
@Override
public ListenerRegistration registerSchemaContextListener(
final EffectiveModelContextListener listener) {
listener.onModelContextUpdated(getGlobalContext());
return new AbstractListenerRegistration<>(listener) {
@Override
protected void removeRegistration() {
// No-op
}
};
}
};
}
}
}
