HONEYCOMB-61: Add context data tree

For storing mapping related context necessary for the plugins
Add naming context adapter to store naming contexts in the data tree

+ Enable persistence for context data tree

Change-Id: I2ac531e80e71a48d313b065997d134da2ae7ee12
Signed-off-by: Maros Marsalek <mmarsale@cisco.com>

5 files changed, 251 insertions, 9 deletions

diff --git a/v3po/vpp-translate-utils/src/main/java/io/fd/honeycomb/v3po/translate/v3po/util/DataTreeNamingContext.java b/v3po/vpp-translate-utils/src/main/java/io/fd/honeycomb/v3po/translate/v3po/util/DataTreeNamingContext.java
new file mode 100644
index 000000000..3b20b2876
--- /dev/null
+++ b/v3po/vpp-translate-utils/src/main/java/io/fd/honeycomb/v3po/translate/v3po/util/DataTreeNamingContext.java
@@ -0,0 +1,193 @@
+/*
+ * Copyright (c) 2016 Cisco and/or its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.fd.honeycomb.v3po.translate.v3po.util;
+
+import com.google.common.collect.Lists;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+import javax.annotation.Nonnull;
+import javax.annotation.concurrent.ThreadSafe;
+import org.opendaylight.yang.gen.v1.urn.honeycomb.params.xml.ns.yang.naming.context.rev160513.Contexts;
+import org.opendaylight.yang.gen.v1.urn.honeycomb.params.xml.ns.yang.naming.context.rev160513.contexts.naming.context.Mappings;
+import org.opendaylight.yang.gen.v1.urn.honeycomb.params.xml.ns.yang.naming.context.rev160513.contexts.naming.context.mappings.Mapping;
+import org.opendaylight.yangtools.yang.common.QName;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
+import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
+import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTree;
+import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeModification;
+import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeSnapshot;
+import org.opendaylight.yangtools.yang.data.api.schema.tree.DataValidationFailedException;
+import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
+import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * Naming context keeping a mapping between int index and string name.
+ * Provides artificial names to unknown indices.
+ */
+@ThreadSafe
+public class DataTreeNamingContext extends NamingContext {
+
+    // FIXME this has to be accessed by readers/writers in a transactional manner and the transaction will only be committed once
+    // the Read or Write operation finishes successfully
+    // Context datatree has to become a first class citizen of Honeycomb, and Honeycomb needs to create and provide
+    // context read write transaction as part of context to read/write customizers
+    // This will then become just a utility writer relying on transaction provided by the infrastructure
+    // Btw. the context transaction needs to disable commit/submit when being passed to customizers
+
+    private static final Logger LOG = LoggerFactory.getLogger(DataTreeNamingContext.class);
+
+    private static final QName NAME_KEY_QNAME = QName.create(Contexts.QNAME, "name");
+    private static final QName INDEX_QNAME = QName.create(Contexts.QNAME, "index");
+
+    private final String instanceName;
+    private final DataTree contextDataTree;
+
+    private final YangInstanceIdentifier.NodeIdentifierWithPredicates namingContextNodeId;
+    private final YangInstanceIdentifier namingContextIid;
+
+    public DataTreeNamingContext(final String artificialNamePrefix, final String instanceName,
+                                 final DataTree contextDataTree) {
+        super(artificialNamePrefix);
+        this.instanceName = instanceName;
+        this.contextDataTree = contextDataTree;
+
+        namingContextNodeId = getNodeId(
+            org.opendaylight.yang.gen.v1.urn.honeycomb.params.xml.ns.yang.naming.context.rev160513.contexts.NamingContext.QNAME,
+            Collections.singletonMap(NAME_KEY_QNAME, instanceName));
+        namingContextIid = YangInstanceIdentifier.create(
+            getNodeId(Contexts.QNAME),
+            getNodeId(org.opendaylight.yang.gen.v1.urn.honeycomb.params.xml.ns.yang.naming.context.rev160513.contexts.NamingContext.QNAME),
+            namingContextNodeId);
+
+        // FIXME read current mappings and initialize map
+        mergeNewContextInDataTree(instanceName);
+    }
+
+    // TODO move the data tree aspect into a dedicated class
+    private void mergeNewContextInDataTree(final String instanceName) {
+        final DataTreeModification dataTreeModification = getModification();
+
+        final YangInstanceIdentifier.NodeIdentifier namingContextsNodeIdForMapNode = getNodeId(
+            org.opendaylight.yang.gen.v1.urn.honeycomb.params.xml.ns.yang.naming.context.rev160513.contexts.NamingContext.QNAME);
+
+        final ContainerNode newMapping = Builders.containerBuilder()
+            .withNodeIdentifier(getNodeId(Contexts.QNAME))
+            .withChild(Builders.mapBuilder()
+                .withNodeIdentifier(namingContextsNodeIdForMapNode)
+                .withChild(Builders.mapEntryBuilder()
+                    .withNodeIdentifier(namingContextNodeId)
+                    .withChild(ImmutableNodes.leafNode(NAME_KEY_QNAME, instanceName))
+                    .withChild(Builders.containerBuilder()
+                        .withNodeIdentifier(getNodeId(Mappings.QNAME))
+                        .withChild(Builders.mapBuilder()
+                            .withNodeIdentifier(getNodeId(Mapping.QNAME))
+                            .build())
+                        .build())
+                    .build())
+                .build())
+            .build();
+
+        // FIXME add logs or debug to resolve:
+//        2016-05-13 15:48:52,401 | WARN  | config-pusher    | DataTreeNamingContext            | 240 - io.fd.honeycomb.v3po.vpp-translate-utils - 1.0.0.SNAPSHOT | Unable to update context: interface-context in context data tree
+//        org.opendaylight.yangtools.yang.data.api.schema.tree.ModifiedNodeDoesNotExistException: Node /(urn:honeycomb:params:xml:ns:yang:naming:context?revision=2016-05-13)contexts/naming-context/naming-context[{(urn:honeycomb:params:xml:ns:yang:naming:context?revision=2016-05-13)name=interface-context}] does not exist. Cannot apply modification to its children.
+//        at org.opendaylight.yangtools.yang.data.impl.schema.tree.AbstractNodeContainerModificationStrategy.checkTouchApplicable(AbstractNodeContainerModificationStrategy.java:276)[55:org.opendaylight.yangtools.yang-data-impl:0.8.0.Beryllium]
+        // FIXME looks like a timing issue, did not occur when debugging
+
+        dataTreeModification.merge(YangInstanceIdentifier.create(getNodeId(Contexts.QNAME)), newMapping);
+
+        commitModification(dataTreeModification);
+    }
+
+    private void commitModification(final DataTreeModification dataTreeModification) {
+        try {
+            dataTreeModification.ready();
+            contextDataTree.validate(dataTreeModification);
+            contextDataTree.commit(contextDataTree.prepare(dataTreeModification));
+        } catch (DataValidationFailedException e) {
+            LOG.warn("Unable to update context: {} in context data tree", instanceName, e);
+            throw new IllegalStateException("Unable to update context in context data tree", e);
+        }
+    }
+
+    private DataTreeModification getModification() {
+        final DataTreeSnapshot dataTreeSnapshot = contextDataTree.takeSnapshot();
+        return dataTreeSnapshot.newModification();
+    }
+
+    private static YangInstanceIdentifier.NodeIdentifierWithPredicates getNodeId(@Nonnull final QName qName,
+                                                                                 @Nonnull final Map<QName, Object> keys) {
+        return new YangInstanceIdentifier.NodeIdentifierWithPredicates(qName, keys);
+    }
+
+    private static YangInstanceIdentifier.NodeIdentifier getNodeId(@Nonnull final QName qName) {
+        return new YangInstanceIdentifier.NodeIdentifier(qName);
+    }
+
+    public synchronized void addName(final int index, final String name) {
+        addMappingToDataTree(name, index);
+        super.addName(index, name);
+    }
+
+    private void addMappingToDataTree(final String name, final int index) {
+        final DataTreeModification dataTreeModification = getModification();
+
+        final YangInstanceIdentifier.NodeIdentifierWithPredicates mappingNodeId = getNodeId(Mapping.QNAME,
+            Collections.singletonMap(NAME_KEY_QNAME, name));
+
+        final List<YangInstanceIdentifier.PathArgument> pathArguments = namingContextIid.getPathArguments();
+        final ArrayList<YangInstanceIdentifier.PathArgument> newPathArgs = Lists.newArrayList(pathArguments);
+        newPathArgs.add(getNodeId(Mappings.QNAME));
+        newPathArgs.add(getNodeId(Mapping.QNAME));
+        newPathArgs.add(mappingNodeId);
+
+        final YangInstanceIdentifier identifier = YangInstanceIdentifier.create(newPathArgs);
+
+        final NormalizedNode<?, ?> newMapping = Builders.mapEntryBuilder()
+            .withNodeIdentifier(mappingNodeId)
+            .withChild(ImmutableNodes.leafNode(NAME_KEY_QNAME, name))
+            .withChild(ImmutableNodes.leafNode(INDEX_QNAME, index))
+            .build();
+
+        dataTreeModification.write(identifier, newMapping);
+
+        commitModification(dataTreeModification);
+    }
+
+    public synchronized int removeName(@Nonnull final String name) {
+        removeMappingFromDataTree(name);
+        return super.removeName(name);
+    }
+
+    private void removeMappingFromDataTree(final String name) {
+        final DataTreeModification dataTreeModification = getModification();
+
+        final YangInstanceIdentifier.NodeIdentifierWithPredicates mappingNodeId = getNodeId(Mapping.QNAME,
+            Collections.singletonMap(NAME_KEY_QNAME, name));
+
+        final YangInstanceIdentifier identifier = YangInstanceIdentifier.create(
+            namingContextIid.getLastPathArgument(), getNodeId(Mappings.QNAME), getNodeId(Mapping.QNAME), mappingNodeId);
+
+        dataTreeModification.delete(identifier);
+
+        commitModification(dataTreeModification);
+    }
+}
diff --git a/v3po/vpp-translate-utils/src/main/java/io/fd/honeycomb/v3po/translate/v3po/util/NamingContext.java b/v3po/vpp-translate-utils/src/main/java/io/fd/honeycomb/v3po/translate/v3po/util/NamingContext.java
index 96d908e1b..9affd0695 100644
--- a/v3po/vpp-translate-utils/src/main/java/io/fd/honeycomb/v3po/translate/v3po/util/NamingContext.java
+++ b/v3po/vpp-translate-utils/src/main/java/io/fd/honeycomb/v3po/translate/v3po/util/NamingContext.java
@@ -28,7 +28,7 @@ import org.slf4j.LoggerFactory;
  * Naming context keeping a mapping between int index and string name.
  * Provides artificial names to unknown indices.
  */
-public final class NamingContext implements AutoCloseable {
+public class NamingContext implements AutoCloseable {
 
     private static final Logger LOG = LoggerFactory.getLogger(NamingContext.class);
 
@@ -49,17 +49,14 @@ public final class NamingContext implements AutoCloseable {
         return nameMapping.inverse().get(index);
     }
 
-    @Nonnull
     public synchronized boolean containsName(final int index) {
         return nameMapping.inverse().containsKey(index);
     }
 
-    @Nonnull
     public synchronized void addName(final int index, final String name) {
         nameMapping.put(name, index);
     }
 
-    @Nonnull
     public synchronized int removeName(final String name) {
         return nameMapping.remove(name);
     }
@@ -71,19 +68,17 @@ public final class NamingContext implements AutoCloseable {
      * @return integer index value matching supplied name
      * @throws IllegalArgumentException if name was not found
      */
-    @Nonnull
     public synchronized int getIndex(String name) {
         checkArgument(nameMapping.containsKey(name), "Name %s not found. Known names: %s",
                 name, nameMapping);
         return nameMapping.get(name);
     }
 
-    @Nonnull
     public synchronized boolean containsIndex(String interfaceName) {
         return nameMapping.containsKey(interfaceName);
     }
 
-    public String getArtificialName(final int index) {
+    private String getArtificialName(final int index) {
         return artificialNamePrefix + index;
     }
 
diff --git a/v3po/vpp-translate-utils/src/main/java/org/opendaylight/yang/gen/v1/urn/honeycomb/params/xml/ns/yang/vpp/util/rev160406/NamingContextImplModule.java b/v3po/vpp-translate-utils/src/main/java/org/opendaylight/yang/gen/v1/urn/honeycomb/params/xml/ns/yang/vpp/util/rev160406/NamingContextImplModule.java
index 81cecb14b..50b577a25 100644
--- a/v3po/vpp-translate-utils/src/main/java/org/opendaylight/yang/gen/v1/urn/honeycomb/params/xml/ns/yang/vpp/util/rev160406/NamingContextImplModule.java
+++ b/v3po/vpp-translate-utils/src/main/java/org/opendaylight/yang/gen/v1/urn/honeycomb/params/xml/ns/yang/vpp/util/rev160406/NamingContextImplModule.java
@@ -1,6 +1,6 @@
 package org.opendaylight.yang.gen.v1.urn.honeycomb.params.xml.ns.yang.vpp.util.rev160406;
 
-import io.fd.honeycomb.v3po.translate.v3po.util.NamingContext;
+import io.fd.honeycomb.v3po.translate.v3po.util.DataTreeNamingContext;
 
 public class NamingContextImplModule extends org.opendaylight.yang.gen.v1.urn.honeycomb.params.xml.ns.yang.vpp.util.rev160406.AbstractNamingContextImplModule {
     public NamingContextImplModule(org.opendaylight.controller.config.api.ModuleIdentifier identifier, org.opendaylight.controller.config.api.DependencyResolver dependencyResolver) {
@@ -18,7 +18,10 @@ public class NamingContextImplModule extends org.opendaylight.yang.gen.v1.urn.ho
 
     @Override
     public java.lang.AutoCloseable createInstance() {
-        return new NamingContext(getArtificialNamePrefix());
+        return new DataTreeNamingContext(
+            getArtificialNamePrefix(),
+            getIdentifier().getInstanceName(),
+            getContextDataTreeDependency());
     }
 
 }
diff --git a/v3po/vpp-translate-utils/src/main/yang/naming-context.yang b/v3po/vpp-translate-utils/src/main/yang/naming-context.yang
new file mode 100644
index 000000000..0bcfc099b
--- /dev/null
+++ b/v3po/vpp-translate-utils/src/main/yang/naming-context.yang
@@ -0,0 +1,42 @@
+module naming-context {
+    yang-version 1;
+    namespace "urn:honeycomb:params:xml:ns:yang:naming:context";
+    prefix "vpp-u";
+
+    description
+        "This module contains data definition for naming mapping context";
+
+    revision "2016-05-13" {
+        description
+            "Initial revision.";
+    }
+
+    // TODO This should become part of v3po-api as operational-only data
+    container contexts {
+
+        list naming-context {
+
+            key "name";
+
+            leaf name {
+                type string;
+            }
+
+            container mappings {
+                list mapping {
+
+                    key "name";
+
+                    leaf name {
+                        type string;
+                    }
+
+                    leaf index {
+                        type int32;
+                    }
+                }
+            }
+        }
+    }
+
+}
\ No newline at end of file
diff --git a/v3po/vpp-translate-utils/src/main/yang/vpp-util.yang b/v3po/vpp-translate-utils/src/main/yang/vpp-util.yang
index 01e759dc2..bd903848c 100644
--- a/v3po/vpp-translate-utils/src/main/yang/vpp-util.yang
+++ b/v3po/vpp-translate-utils/src/main/yang/vpp-util.yang
@@ -4,6 +4,7 @@ module vpp-util {
     prefix "vpp-u";
 
     import config { prefix config; revision-date 2013-04-05; }
+    import data-api { prefix dapi; revision-date 2016-04-11; }
 
     description
         "This module contains utilities for vpp readers/writers";
@@ -31,6 +32,14 @@ module vpp-util {
                 type string;
             }
 
+            container context-data-tree {
+                uses config:service-ref {
+                    refine type {
+                        mandatory true;
+                        config:required-identity dapi:data-tree;
+                    }
+                }
+            }
         }
     }