--- /dev/null
+package org.opendaylight.controller.md.sal.dom.broker.impl.compat;
+
+import static com.google.common.base.Preconditions.checkNotNull;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.Future;
+
+import org.opendaylight.controller.md.sal.common.api.TransactionStatus;
+import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
+import org.opendaylight.controller.md.sal.common.impl.util.compat.DataNormalizationOperation;
+import org.opendaylight.controller.md.sal.common.impl.util.compat.DataNormalizer;
+import org.opendaylight.controller.md.sal.dom.api.DOMDataReadTransaction;
+import org.opendaylight.controller.md.sal.dom.api.DOMDataReadWriteTransaction;
+import org.opendaylight.controller.sal.core.api.data.DataModificationTransaction;
+import org.opendaylight.yangtools.concepts.Delegator;
+import org.opendaylight.yangtools.concepts.ListenerRegistration;
+import org.opendaylight.yangtools.yang.common.RpcResult;
+import org.opendaylight.yangtools.yang.data.api.CompositeNode;
+import org.opendaylight.yangtools.yang.data.api.InstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.InstanceIdentifier.AugmentationIdentifier;
+import org.opendaylight.yangtools.yang.data.api.InstanceIdentifier.NodeIdentifier;
+import org.opendaylight.yangtools.yang.data.api.InstanceIdentifier.PathArgument;
+import org.opendaylight.yangtools.yang.data.api.schema.AugmentationNode;
+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.MapNode;
+import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
+import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.google.common.base.Optional;
+import com.google.common.base.Preconditions;
+import com.google.common.collect.Iterables;
+import com.google.common.util.concurrent.ListenableFuture;
+
+public abstract class BackwardsCompatibleTransaction<T extends DOMDataReadTransaction> implements
+ DataModificationTransaction, Delegator<T> {
+
+ private static final Logger LOG = LoggerFactory.getLogger(BackwardsCompatibleTransaction.class);
+
+ private final T asyncTx;
+ private final DataNormalizer normalizer;
+
+ protected BackwardsCompatibleTransaction(final T asyncTx, final DataNormalizer normalizer) {
+ super();
+ this.asyncTx = asyncTx;
+ this.normalizer = normalizer;
+ }
+
+ public static BackwardsCompatibleTransaction<?> readOnlyTransaction(final DOMDataReadTransaction readTx,
+ final DataNormalizer normalizer) {
+
+ return new BackwardsCompatibleTransaction<DOMDataReadTransaction>(readTx, normalizer) {
+
+ @Override
+ public TransactionStatus getStatus() {
+ return TransactionStatus.NEW;
+ }
+
+ @Override
+ public Future<RpcResult<TransactionStatus>> commit() {
+ getDelegate().close();
+ return null;
+ }
+ };
+ }
+
+ public static BackwardsCompatibleTransaction<?> readWriteTransaction(final DOMDataReadWriteTransaction rwTx,
+ final DataNormalizer normalizer) {
+ return new ReadWriteTransaction(rwTx, normalizer);
+ }
+
+ protected DataNormalizer getNormalizer() {
+ return normalizer;
+ }
+
+ @Override
+ public T getDelegate() {
+ return asyncTx;
+ };
+
+ @Override
+ public CompositeNode readConfigurationData(final InstanceIdentifier legacyPath) {
+
+ InstanceIdentifier normalizedPath = normalizer.toNormalized(legacyPath);
+
+ ListenableFuture<Optional<NormalizedNode<?, ?>>> normalizedData = asyncTx.read(
+ LogicalDatastoreType.CONFIGURATION, normalizedPath);
+
+ try {
+ return normalizer.toLegacy(normalizedPath, normalizedData.get().orNull());
+ } catch (InterruptedException | ExecutionException e) {
+ return null;
+ }
+ }
+
+ @Override
+ public CompositeNode readOperationalData(final InstanceIdentifier legacyPath) {
+ InstanceIdentifier normalizedPath = normalizer.toNormalized(legacyPath);
+
+ ListenableFuture<Optional<NormalizedNode<?, ?>>> normalizedData = asyncTx.read(
+ LogicalDatastoreType.OPERATIONAL, normalizedPath);
+
+ try {
+ return normalizer.toLegacy(normalizedPath, normalizedData.get().orNull());
+ } catch (InterruptedException | ExecutionException e) {
+ return null;
+ }
+ }
+
+ @Override
+ public ListenerRegistration<DataTransactionListener> registerListener(final DataTransactionListener listener) {
+ throw new UnsupportedOperationException();
+ }
+
+ @Override
+ public Map<InstanceIdentifier, CompositeNode> getCreatedConfigurationData() {
+ return Collections.emptyMap();
+ }
+
+ @Override
+ public Map<InstanceIdentifier, CompositeNode> getCreatedOperationalData() {
+ return Collections.emptyMap();
+ }
+
+ @Override
+ public Map<InstanceIdentifier, CompositeNode> getOriginalConfigurationData() {
+ return Collections.emptyMap();
+ }
+
+ @Override
+ public Map<InstanceIdentifier, CompositeNode> getOriginalOperationalData() {
+ return Collections.emptyMap();
+ }
+
+ @Override
+ public Set<InstanceIdentifier> getRemovedConfigurationData() {
+ return Collections.emptySet();
+ }
+
+ @Override
+ public Set<InstanceIdentifier> getRemovedOperationalData() {
+ return Collections.emptySet();
+ }
+
+ @Override
+ public Map<InstanceIdentifier, CompositeNode> getUpdatedConfigurationData() {
+ return Collections.emptyMap();
+ }
+
+ @Override
+ public Map<InstanceIdentifier, CompositeNode> getUpdatedOperationalData() {
+ return Collections.emptyMap();
+ }
+
+ @Override
+ public void putConfigurationData(final InstanceIdentifier path, final CompositeNode data) {
+ throw new UnsupportedOperationException();
+ }
+
+ @Override
+ public void putOperationalData(final InstanceIdentifier path, final CompositeNode data) {
+ throw new UnsupportedOperationException();
+ }
+
+ @Override
+ public void removeConfigurationData(final InstanceIdentifier path) {
+ throw new UnsupportedOperationException();
+ }
+
+ @Override
+ public void removeOperationalData(final InstanceIdentifier path) {
+ throw new UnsupportedOperationException();
+ }
+
+ @Override
+ public Object getIdentifier() {
+ return asyncTx.getIdentifier();
+ }
+
+ private static final class ReadWriteTransaction extends BackwardsCompatibleTransaction<DOMDataReadWriteTransaction> {
+
+ private TransactionStatus status = TransactionStatus.NEW;
+
+ protected ReadWriteTransaction(final DOMDataReadWriteTransaction asyncTx, final DataNormalizer normalizer) {
+ super(asyncTx, normalizer);
+ }
+
+ @Override
+ public TransactionStatus getStatus() {
+ return status;
+ }
+
+ @Override
+ public Future<RpcResult<TransactionStatus>> commit() {
+ Preconditions.checkState(status == TransactionStatus.NEW);
+ status = TransactionStatus.SUBMITED;
+ return getDelegate().commit();
+ }
+
+ @Override
+ public void putConfigurationData(final InstanceIdentifier legacyPath, final CompositeNode legacyData) {
+ checkNotNull(legacyPath, "Path MUST NOT be null.");
+ checkNotNull(legacyData, "Data for path %s MUST NOT be null",legacyData);
+ Entry<InstanceIdentifier, NormalizedNode<?, ?>> normalizedData = getNormalizer().toNormalized(legacyPath, legacyData);
+ putWithEnsuredParents(LogicalDatastoreType.CONFIGURATION, normalizedData.getKey(), normalizedData.getValue());
+ }
+
+ @Override
+ public void putOperationalData(final InstanceIdentifier legacyPath, final CompositeNode legacyData) {
+ checkNotNull(legacyPath, "Path MUST NOT be null.");
+ checkNotNull(legacyData, "Data for path %s MUST NOT be null",legacyData);
+ Entry<InstanceIdentifier, NormalizedNode<?, ?>> normalizedData = getNormalizer().toNormalized(legacyPath, legacyData);
+ putWithEnsuredParents(LogicalDatastoreType.OPERATIONAL, normalizedData.getKey(), normalizedData.getValue());
+ }
+
+ private void putWithEnsuredParents(final LogicalDatastoreType store, final InstanceIdentifier normalizedPath,
+ final NormalizedNode<?, ?> normalizedData) {
+
+ LOG.trace("write {}:{} ",store,normalizedPath);
+ try {
+ List<PathArgument> currentArguments = new ArrayList<>();
+ DataNormalizationOperation<?> currentOp = getNormalizer().getRootOperation();
+ Iterator<PathArgument> iterator = normalizedPath.getPath().iterator();
+ while(iterator.hasNext()) {
+ PathArgument currentArg = iterator.next();
+ currentOp = currentOp.getChild(currentArg);
+ currentArguments.add(currentArg);
+ InstanceIdentifier currentPath = new InstanceIdentifier(currentArguments);
+ boolean isPresent = getDelegate().read(store, currentPath).get().isPresent();
+ if(isPresent == false && iterator.hasNext()) {
+ getDelegate().put(store, currentPath, currentOp.createDefault(currentArg));
+ }
+ }
+ } catch (InterruptedException | ExecutionException e) {
+ LOG.error("Exception durring read.",e);
+ }
+
+ getDelegate().put(store, normalizedPath, normalizedData);
+ }
+
+ private boolean isAugmentationChild(final InstanceIdentifier normalizedPath) {
+ List<PathArgument> parentArgs = parentPath(normalizedPath).getPath();
+ if(parentArgs.isEmpty()) {
+ return false;
+ }
+ return Iterables.getLast(parentArgs) instanceof AugmentationIdentifier;
+ }
+
+ private void ensureParentNode(final LogicalDatastoreType store, final InstanceIdentifier normalizedPath,
+ final NormalizedNode<?, ?> normalizedData) {
+ InstanceIdentifier parentPath = parentPath(normalizedPath);
+ PathArgument parentType = Iterables.getLast(parentPath.getPath());
+ if(parentType instanceof AugmentationIdentifier) {
+ AugmentationNode node = Builders.augmentationBuilder()
+ .withNodeIdentifier((AugmentationIdentifier) parentType)
+ .build();
+ getDelegate().put(store, parentPath, node);
+ }
+ if(normalizedData instanceof MapEntryNode) {
+ MapNode mapNode = Builders.mapBuilder().withNodeIdentifier(new NodeIdentifier(normalizedData.getNodeType())).build();
+ getDelegate().put(store, parentPath, mapNode);
+ } else if (normalizedData instanceof LeafSetNode<?>){
+ LeafSetNode<Object> leafNode = Builders.leafSetBuilder().withNodeIdentifier(new NodeIdentifier(normalizedData.getNodeType())).build();
+ getDelegate().put(store, parentPath, leafNode);
+ }
+
+
+ }
+
+ private InstanceIdentifier parentPath(final InstanceIdentifier normalizedPath) {
+ List<PathArgument> childArgs = normalizedPath.getPath();
+ return new InstanceIdentifier(childArgs.subList(0, childArgs.size() -1));
+ }
+
+ private boolean parentNodeDoesNotExists(final LogicalDatastoreType store, final InstanceIdentifier normalizedPath) {
+ try {
+ return !getDelegate().read(store, parentPath(normalizedPath)).get().isPresent();
+ } catch (InterruptedException | ExecutionException e) {
+ throw new IllegalStateException(e);
+ }
+ }
+
+ @Override
+ public void removeConfigurationData(final InstanceIdentifier legacyPath) {
+ checkNotNull(legacyPath, "Path MUST NOT be null.");
+ getDelegate().delete(LogicalDatastoreType.CONFIGURATION, getNormalizer().toNormalized(legacyPath));
+ }
+
+ @Override
+ public void removeOperationalData(final InstanceIdentifier legacyPath) {
+ checkNotNull(legacyPath, "Path MUST NOT be null.");
+ getDelegate().delete(LogicalDatastoreType.OPERATIONAL, getNormalizer().toNormalized(legacyPath));
+ }
+ }
+}