InstanceIdentifierContext does not take generics
[netconf.git] / restconf / restconf-nb-bierman02 / src / main / java / org / opendaylight / netconf / sal / restconf / impl / RestconfImpl.java
index 1f3710bd6f37c42e4f2a7c374eda1a6f381900af..16bc30d218d4f2f9660a205e3323d84d22068fb4 100644 (file)
@@ -7,15 +7,19 @@
  */
 package org.opendaylight.netconf.sal.restconf.impl;
 
+import static com.google.common.base.Preconditions.checkArgument;
+import static com.google.common.base.Preconditions.checkNotNull;
+import static com.google.common.base.Preconditions.checkState;
+import static java.util.Objects.requireNonNull;
+
 import com.google.common.annotations.VisibleForTesting;
-import com.google.common.base.Preconditions;
 import com.google.common.base.Predicates;
 import com.google.common.base.Splitter;
 import com.google.common.base.Strings;
 import com.google.common.base.Throwables;
+import com.google.common.collect.ImmutableMap;
 import com.google.common.collect.ImmutableSet;
 import com.google.common.collect.Iterables;
-import com.google.common.collect.Maps;
 import com.google.common.util.concurrent.FluentFuture;
 import com.google.common.util.concurrent.Futures;
 import com.google.common.util.concurrent.ListenableFuture;
@@ -60,28 +64,31 @@ import org.opendaylight.mdsal.dom.api.DOMMountPoint;
 import org.opendaylight.mdsal.dom.api.DOMRpcImplementationNotAvailableException;
 import org.opendaylight.mdsal.dom.api.DOMRpcResult;
 import org.opendaylight.mdsal.dom.api.DOMRpcService;
+import org.opendaylight.mdsal.dom.api.DOMSchemaService;
 import org.opendaylight.mdsal.dom.spi.DefaultDOMRpcResult;
 import org.opendaylight.netconf.sal.rest.api.Draft02;
 import org.opendaylight.netconf.sal.rest.api.RestconfService;
+import org.opendaylight.netconf.sal.rest.impl.NormalizedNodeContext;
 import org.opendaylight.netconf.sal.streams.listeners.ListenerAdapter;
 import org.opendaylight.netconf.sal.streams.listeners.NotificationListenerAdapter;
 import org.opendaylight.netconf.sal.streams.listeners.Notificator;
 import org.opendaylight.netconf.sal.streams.websockets.WebSocketServer;
+import org.opendaylight.restconf.common.OperationsContent;
 import org.opendaylight.restconf.common.context.InstanceIdentifierContext;
-import org.opendaylight.restconf.common.context.NormalizedNodeContext;
 import org.opendaylight.restconf.common.errors.RestconfDocumentedException;
-import org.opendaylight.restconf.common.errors.RestconfError.ErrorTag;
-import org.opendaylight.restconf.common.errors.RestconfError.ErrorType;
 import org.opendaylight.restconf.common.patch.PatchContext;
 import org.opendaylight.restconf.common.patch.PatchStatusContext;
-import org.opendaylight.restconf.common.util.DataChangeScope;
 import org.opendaylight.restconf.common.util.OperationsResourceUtils;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.DateAndTime;
+import org.opendaylight.yang.gen.v1.urn.sal.restconf.event.subscription.rev140708.CreateDataChangeEventSubscriptionInput1.Scope;
 import org.opendaylight.yang.gen.v1.urn.sal.restconf.event.subscription.rev140708.NotificationOutputTypeGrouping.NotificationOutputType;
 import org.opendaylight.yangtools.yang.common.Empty;
+import org.opendaylight.yangtools.yang.common.ErrorTag;
+import org.opendaylight.yangtools.yang.common.ErrorType;
 import org.opendaylight.yangtools.yang.common.QName;
 import org.opendaylight.yangtools.yang.common.QNameModule;
 import org.opendaylight.yangtools.yang.common.Revision;
+import org.opendaylight.yangtools.yang.common.XMLNamespace;
 import org.opendaylight.yangtools.yang.common.YangConstants;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier;
@@ -97,13 +104,15 @@ 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.api.schema.SystemLeafSetNode;
+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.api.schema.builder.DataContainerNodeBuilder;
+import org.opendaylight.yangtools.yang.data.api.schema.builder.ListNodeBuilder;
+import org.opendaylight.yangtools.yang.data.api.schema.builder.NormalizedNodeBuilder;
 import org.opendaylight.yangtools.yang.data.api.schema.tree.ModifiedNodeDoesNotExistException;
-import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
-import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.CollectionNodeBuilder;
-import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeBuilder;
-import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.ListNodeBuilder;
-import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.NormalizedNodeBuilder;
+import org.opendaylight.yangtools.yang.data.impl.schema.SchemaAwareBuilders;
 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableContainerNodeBuilder;
 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafNodeBuilder;
 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
@@ -119,6 +128,7 @@ import org.opendaylight.yangtools.yang.model.api.RpcDefinition;
 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
+import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier.Absolute;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -135,11 +145,10 @@ public final class RestconfImpl implements RestconfService {
 
     private static final Logger LOG = LoggerFactory.getLogger(RestconfImpl.class);
 
-    private static final DataChangeScope DEFAULT_SCOPE = DataChangeScope.BASE;
-
     private static final LogicalDatastoreType DEFAULT_DATASTORE = LogicalDatastoreType.CONFIGURATION;
 
-    private static final URI NAMESPACE_EVENT_SUBSCRIPTION_AUGMENT = URI.create("urn:sal:restconf:event:subscription");
+    private static final XMLNamespace NAMESPACE_EVENT_SUBSCRIPTION_AUGMENT =
+        XMLNamespace.of("urn:sal:restconf:event:subscription");
 
     private static final String DATASTORE_PARAM_NAME = "datastore";
 
@@ -151,7 +160,7 @@ public final class RestconfImpl implements RestconfService {
 
     private static final String NETCONF_BASE_PAYLOAD_NAME = "data";
 
-    private static final QName NETCONF_BASE_QNAME = QName.create(QNameModule.create(URI.create(NETCONF_BASE)),
+    private static final QName NETCONF_BASE_QNAME = QName.create(QNameModule.create(XMLNamespace.of(NETCONF_BASE)),
         NETCONF_BASE_PAYLOAD_NAME).intern();
 
     private static final QNameModule SAL_REMOTE_AUGMENT = QNameModule.create(NAMESPACE_EVENT_SUBSCRIPTION_AUGMENT,
@@ -199,21 +208,22 @@ public final class RestconfImpl implements RestconfService {
     }
 
     @Override
+    @Deprecated
     public NormalizedNodeContext getModules(final UriInfo uriInfo) {
         final MapNode allModuleMap = makeModuleMapNode(controllerContext.getAllModules());
 
-        final EffectiveModelContext schemaContext = this.controllerContext.getGlobalSchema();
+        final EffectiveModelContext schemaContext = controllerContext.getGlobalSchema();
 
         final Module restconfModule = getRestconfModule();
-        final DataSchemaNode modulesSchemaNode = this.controllerContext.getRestconfModuleRestConfSchemaNode(
+        final DataSchemaNode modulesSchemaNode = controllerContext.getRestconfModuleRestConfSchemaNode(
                 restconfModule, Draft02.RestConfModule.MODULES_CONTAINER_SCHEMA_NODE);
-        Preconditions.checkState(modulesSchemaNode instanceof ContainerSchemaNode);
+        checkState(modulesSchemaNode instanceof ContainerSchemaNode);
 
         final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> moduleContainerBuilder =
-                Builders.containerBuilder((ContainerSchemaNode) modulesSchemaNode);
+                SchemaAwareBuilders.containerBuilder((ContainerSchemaNode) modulesSchemaNode);
         moduleContainerBuilder.withChild(allModuleMap);
 
-        return new NormalizedNodeContext(new InstanceIdentifierContext<>(null, modulesSchemaNode, null, schemaContext),
+        return new NormalizedNodeContext(new InstanceIdentifierContext(null, modulesSchemaNode, null, schemaContext),
                 moduleContainerBuilder.build(), QueryParametersParser.parseWriterParameters(uriInfo));
     }
 
@@ -221,8 +231,8 @@ public final class RestconfImpl implements RestconfService {
      * Valid only for mount point.
      */
     @Override
+    @Deprecated
     public NormalizedNodeContext getModules(final String identifier, final UriInfo uriInfo) {
-        Preconditions.checkNotNull(identifier);
         if (!identifier.contains(ControllerContext.MOUNT)) {
             final String errMsg = "URI has bad format. If modules behind mount point should be showed,"
                     + " URI has to end with " + ControllerContext.MOUNT;
@@ -230,43 +240,43 @@ public final class RestconfImpl implements RestconfService {
             throw new RestconfDocumentedException(errMsg, ErrorType.PROTOCOL, ErrorTag.INVALID_VALUE);
         }
 
-        final InstanceIdentifierContext<?> mountPointIdentifier =
-                this.controllerContext.toMountPointIdentifier(identifier);
+        final InstanceIdentifierContext mountPointIdentifier =
+                controllerContext.toMountPointIdentifier(identifier);
         final DOMMountPoint mountPoint = mountPointIdentifier.getMountPoint();
         final MapNode mountPointModulesMap = makeModuleMapNode(controllerContext.getAllModules(mountPoint));
 
         final Module restconfModule = getRestconfModule();
-        final DataSchemaNode modulesSchemaNode = this.controllerContext.getRestconfModuleRestConfSchemaNode(
+        final DataSchemaNode modulesSchemaNode = controllerContext.getRestconfModuleRestConfSchemaNode(
                 restconfModule, Draft02.RestConfModule.MODULES_CONTAINER_SCHEMA_NODE);
-        Preconditions.checkState(modulesSchemaNode instanceof ContainerSchemaNode);
+        checkState(modulesSchemaNode instanceof ContainerSchemaNode);
 
         final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> moduleContainerBuilder =
-                Builders.containerBuilder((ContainerSchemaNode) modulesSchemaNode);
+                SchemaAwareBuilders.containerBuilder((ContainerSchemaNode) modulesSchemaNode);
         moduleContainerBuilder.withChild(mountPointModulesMap);
 
         return new NormalizedNodeContext(
-                new InstanceIdentifierContext<>(null, modulesSchemaNode, mountPoint,
-                        this.controllerContext.getGlobalSchema()),
+                new InstanceIdentifierContext(null, modulesSchemaNode, mountPoint,
+                        controllerContext.getGlobalSchema()),
                 moduleContainerBuilder.build(), QueryParametersParser.parseWriterParameters(uriInfo));
     }
 
     @Override
+    @Deprecated
     public NormalizedNodeContext getModule(final String identifier, final UriInfo uriInfo) {
-        Preconditions.checkNotNull(identifier);
-        final Entry<String, Revision> nameRev = getModuleNameAndRevision(identifier);
+        final Entry<String, Revision> nameRev = getModuleNameAndRevision(requireNonNull(identifier));
         Module module = null;
         DOMMountPoint mountPoint = null;
         final EffectiveModelContext schemaContext;
         if (identifier.contains(ControllerContext.MOUNT)) {
-            final InstanceIdentifierContext<?> mountPointIdentifier =
-                    this.controllerContext.toMountPointIdentifier(identifier);
+            final InstanceIdentifierContext mountPointIdentifier =
+                    controllerContext.toMountPointIdentifier(identifier);
             mountPoint = mountPointIdentifier.getMountPoint();
-            module = this.controllerContext.findModuleByNameAndRevision(mountPoint, nameRev.getKey(),
+            module = controllerContext.findModuleByNameAndRevision(mountPoint, nameRev.getKey(),
                 nameRev.getValue());
-            schemaContext = mountPoint.getEffectiveModelContext();
+            schemaContext = modelContext(mountPoint);
         } else {
-            module = this.controllerContext.findModuleByNameAndRevision(nameRev.getKey(), nameRev.getValue());
-            schemaContext = this.controllerContext.getGlobalSchema();
+            module = controllerContext.findModuleByNameAndRevision(nameRev.getKey(), nameRev.getValue());
+            schemaContext = controllerContext.getGlobalSchema();
         }
 
         if (module == null) {
@@ -279,50 +289,59 @@ public final class RestconfImpl implements RestconfService {
         final Set<Module> modules = Collections.singleton(module);
         final MapNode moduleMap = makeModuleMapNode(modules);
 
-        final DataSchemaNode moduleSchemaNode = this.controllerContext
+        final DataSchemaNode moduleSchemaNode = controllerContext
                 .getRestconfModuleRestConfSchemaNode(restconfModule, Draft02.RestConfModule.MODULE_LIST_SCHEMA_NODE);
-        Preconditions.checkState(moduleSchemaNode instanceof ListSchemaNode);
+        checkState(moduleSchemaNode instanceof ListSchemaNode);
 
         return new NormalizedNodeContext(
-                new InstanceIdentifierContext<>(null, moduleSchemaNode, mountPoint, schemaContext), moduleMap,
+                new InstanceIdentifierContext(null, moduleSchemaNode, mountPoint, schemaContext), moduleMap,
                 QueryParametersParser.parseWriterParameters(uriInfo));
     }
 
     @Override
+    @Deprecated
     public NormalizedNodeContext getAvailableStreams(final UriInfo uriInfo) {
-        final EffectiveModelContext schemaContext = this.controllerContext.getGlobalSchema();
+        final EffectiveModelContext schemaContext = controllerContext.getGlobalSchema();
         final Set<String> availableStreams = Notificator.getStreamNames();
         final Module restconfModule = getRestconfModule();
-        final DataSchemaNode streamSchemaNode = this.controllerContext
+        final DataSchemaNode streamSchemaNode = controllerContext
                 .getRestconfModuleRestConfSchemaNode(restconfModule, Draft02.RestConfModule.STREAM_LIST_SCHEMA_NODE);
-        Preconditions.checkState(streamSchemaNode instanceof ListSchemaNode);
+        checkState(streamSchemaNode instanceof ListSchemaNode);
 
-        final CollectionNodeBuilder<MapEntryNode, MapNode> listStreamsBuilder =
-                Builders.mapBuilder((ListSchemaNode) streamSchemaNode);
+        final CollectionNodeBuilder<MapEntryNode, SystemMapNode> listStreamsBuilder =
+                SchemaAwareBuilders.mapBuilder((ListSchemaNode) streamSchemaNode);
 
         for (final String streamName : availableStreams) {
             listStreamsBuilder.withChild(toStreamEntryNode(streamName, streamSchemaNode));
         }
 
-        final DataSchemaNode streamsContainerSchemaNode = this.controllerContext.getRestconfModuleRestConfSchemaNode(
+        final DataSchemaNode streamsContainerSchemaNode = controllerContext.getRestconfModuleRestConfSchemaNode(
                 restconfModule, Draft02.RestConfModule.STREAMS_CONTAINER_SCHEMA_NODE);
-        Preconditions.checkState(streamsContainerSchemaNode instanceof ContainerSchemaNode);
+        checkState(streamsContainerSchemaNode instanceof ContainerSchemaNode);
 
         final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> streamsContainerBuilder =
-                Builders.containerBuilder((ContainerSchemaNode) streamsContainerSchemaNode);
+                SchemaAwareBuilders.containerBuilder((ContainerSchemaNode) streamsContainerSchemaNode);
         streamsContainerBuilder.withChild(listStreamsBuilder.build());
 
         return new NormalizedNodeContext(
-                new InstanceIdentifierContext<>(null, streamsContainerSchemaNode, null, schemaContext),
+                new InstanceIdentifierContext(null, streamsContainerSchemaNode, null, schemaContext),
                 streamsContainerBuilder.build(), QueryParametersParser.parseWriterParameters(uriInfo));
     }
 
     @Override
-    public NormalizedNodeContext getOperations(final UriInfo uriInfo) {
-        return OperationsResourceUtils.contextForModelContext(controllerContext.getGlobalSchema(), null);
+    @Deprecated
+    public String getOperationsJSON() {
+        return OperationsContent.JSON.bodyFor(controllerContext.getGlobalSchema());
     }
 
     @Override
+    @Deprecated
+    public String getOperationsXML() {
+        return OperationsContent.XML.bodyFor(controllerContext.getGlobalSchema());
+    }
+
+    @Override
+    @Deprecated
     public NormalizedNodeContext getOperations(final String identifier, final UriInfo uriInfo) {
         if (!identifier.contains(ControllerContext.MOUNT)) {
             final String errMsg = "URI has bad format. If operations behind mount point should be showed, URI has to "
@@ -331,14 +350,14 @@ public final class RestconfImpl implements RestconfService {
             throw new RestconfDocumentedException(errMsg, ErrorType.PROTOCOL, ErrorTag.INVALID_VALUE);
         }
 
-        final InstanceIdentifierContext<?> mountPointIdentifier =
-                this.controllerContext.toMountPointIdentifier(identifier);
+        final InstanceIdentifierContext mountPointIdentifier = controllerContext.toMountPointIdentifier(identifier);
         final DOMMountPoint mountPoint = mountPointIdentifier.getMountPoint();
-        return OperationsResourceUtils.contextForModelContext(mountPoint.getSchemaContext(), mountPoint);
+        final var entry = OperationsResourceUtils.contextForModelContext(modelContext(mountPoint), mountPoint);
+        return new NormalizedNodeContext(entry.getKey(), entry.getValue());
     }
 
     private Module getRestconfModule() {
-        final Module restconfModule = this.controllerContext.getRestconfModule();
+        final Module restconfModule = controllerContext.getRestconfModule();
         if (restconfModule == null) {
             LOG.debug("ietf-restconf module was not found.");
             throw new RestconfDocumentedException("ietf-restconf module was not found.", ErrorType.APPLICATION,
@@ -391,19 +410,19 @@ public final class RestconfImpl implements RestconfService {
         final SchemaNode schema = payload.getInstanceIdentifierContext().getSchemaNode();
         final ListenableFuture<? extends DOMRpcResult> response;
         final DOMMountPoint mountPoint = payload.getInstanceIdentifierContext().getMountPoint();
-        final NormalizedNode<?, ?> input =  nonnullInput(schema, payload.getData());
+        final NormalizedNode input =  nonnullInput(schema, payload.getData());
         final EffectiveModelContext schemaContext;
 
         if (mountPoint != null) {
             final Optional<DOMRpcService> mountRpcServices = mountPoint.getService(DOMRpcService.class);
-            if (!mountRpcServices.isPresent()) {
+            if (mountRpcServices.isEmpty()) {
                 LOG.debug("Error: Rpc service is missing.");
                 throw new RestconfDocumentedException("Rpc service is missing.");
             }
-            schemaContext = mountPoint.getEffectiveModelContext();
-            response = mountRpcServices.get().invokeRpc(schema.getPath(), input);
+            schemaContext = modelContext(mountPoint);
+            response = mountRpcServices.get().invokeRpc(schema.getQName(), input);
         } else {
-            final URI namespace = schema.getQName().getNamespace();
+            final XMLNamespace namespace = schema.getQName().getNamespace();
             if (namespace.toString().equals(SAL_REMOTE_NAMESPACE)) {
                 if (identifier.contains(CREATE_DATA_SUBSCR)) {
                     response = invokeSalRemoteRpcSubscribeRPC(payload);
@@ -415,15 +434,15 @@ public final class RestconfImpl implements RestconfService {
                     throw new RestconfDocumentedException(msg, ErrorType.RPC, ErrorTag.OPERATION_NOT_SUPPORTED);
                 }
             } else {
-                response = this.broker.invokeRpc(schema.getPath(), input);
+                response = broker.invokeRpc(schema.getQName(), input);
             }
-            schemaContext = this.controllerContext.getGlobalSchema();
+            schemaContext = controllerContext.getGlobalSchema();
         }
 
         final DOMRpcResult result = checkRpcResponse(response);
 
         RpcDefinition resultNodeSchema = null;
-        final NormalizedNode<?, ?> resultData;
+        final NormalizedNode resultData;
         if (result != null && result.getResult() != null) {
             resultData = result.getResult();
             resultNodeSchema = (RpcDefinition) payload.getInstanceIdentifierContext().getSchemaNode();
@@ -431,11 +450,11 @@ public final class RestconfImpl implements RestconfService {
             resultData = null;
         }
 
-        if (resultData != null && ((ContainerNode) resultData).getValue().isEmpty()) {
+        if (resultData != null && ((ContainerNode) resultData).isEmpty()) {
             throw new WebApplicationException(Response.Status.NO_CONTENT);
         } else {
             return new NormalizedNodeContext(
-                    new InstanceIdentifierContext<>(null, resultNodeSchema, mountPoint, schemaContext),
+                    new InstanceIdentifierContext(null, resultNodeSchema, mountPoint, schemaContext),
                     resultData, QueryParametersParser.parseWriterParameters(uriInfo));
         }
     }
@@ -448,9 +467,9 @@ public final class RestconfImpl implements RestconfService {
         final EffectiveModelContext schemaContext;
         if (identifier.contains(ControllerContext.MOUNT)) {
             // mounted RPC call - look up mount instance.
-            final InstanceIdentifierContext<?> mountPointId = this.controllerContext.toMountPointIdentifier(identifier);
+            final InstanceIdentifierContext mountPointId = controllerContext.toMountPointIdentifier(identifier);
             mountPoint = mountPointId.getMountPoint();
-            schemaContext = mountPoint.getEffectiveModelContext();
+            schemaContext = modelContext(mountPoint);
             final int startOfRemoteRpcName =
                     identifier.lastIndexOf(ControllerContext.MOUNT) + ControllerContext.MOUNT.length() + 1;
             final String remoteRpcName = identifier.substring(startOfRemoteRpcName);
@@ -459,7 +478,7 @@ public final class RestconfImpl implements RestconfService {
         } else if (identifier.indexOf('/') == CHAR_NOT_FOUND) {
             identifierEncoded = identifier;
             mountPoint = null;
-            schemaContext = this.controllerContext.getGlobalSchema();
+            schemaContext = controllerContext.getGlobalSchema();
         } else {
             LOG.debug("Identifier {} cannot contain slash character (/).", identifier);
             throw new RestconfDocumentedException(String.format("Identifier %n%s%ncan\'t contain slash character (/).%n"
@@ -470,9 +489,9 @@ public final class RestconfImpl implements RestconfService {
         final String identifierDecoded = ControllerContext.urlPathArgDecode(identifierEncoded);
         final RpcDefinition rpc;
         if (mountPoint == null) {
-            rpc = this.controllerContext.getRpcDefinition(identifierDecoded);
+            rpc = controllerContext.getRpcDefinition(identifierDecoded);
         } else {
-            rpc = findRpc(mountPoint.getSchemaContext(), identifierDecoded);
+            rpc = findRpc(modelContext(mountPoint), identifierDecoded);
         }
 
         if (rpc == null) {
@@ -490,16 +509,16 @@ public final class RestconfImpl implements RestconfService {
         final ListenableFuture<? extends DOMRpcResult> response;
         if (mountPoint != null) {
             final Optional<DOMRpcService> mountRpcServices = mountPoint.getService(DOMRpcService.class);
-            if (!mountRpcServices.isPresent()) {
+            if (mountRpcServices.isEmpty()) {
                 throw new RestconfDocumentedException("Rpc service is missing.");
             }
-            response = mountRpcServices.get().invokeRpc(rpc.getPath(), input);
+            response = mountRpcServices.get().invokeRpc(rpc.getQName(), input);
         } else {
-            response = this.broker.invokeRpc(rpc.getPath(), input);
+            response = broker.invokeRpc(rpc.getQName(), input);
         }
 
-        final NormalizedNode<?, ?> result = checkRpcResponse(response).getResult();
-        if (result != null && ((ContainerNode) result).getValue().isEmpty()) {
+        final NormalizedNode result = checkRpcResponse(response).getResult();
+        if (result != null && ((ContainerNode) result).isEmpty()) {
             throw new WebApplicationException(Response.Status.NO_CONTENT);
         }
 
@@ -510,11 +529,11 @@ public final class RestconfImpl implements RestconfService {
         //
         //        This is legacy code, so if anybody cares to do that refactor, feel free to contribute, but I am not
         //        doing that work.
-        return new NormalizedNodeContext(new InstanceIdentifierContext<>(null, rpc, mountPoint, schemaContext), result,
+        return new NormalizedNodeContext(new InstanceIdentifierContext(null, rpc, mountPoint, schemaContext), result,
             QueryParametersParser.parseWriterParameters(uriInfo));
     }
 
-    private static @NonNull NormalizedNode<?, ?> nonnullInput(final SchemaNode rpc, final NormalizedNode<?, ?> input) {
+    private static @NonNull NormalizedNode nonnullInput(final SchemaNode rpc, final NormalizedNode input) {
         return input != null ? input : defaultInput(rpc.getQName());
     }
 
@@ -577,10 +596,9 @@ public final class RestconfImpl implements RestconfService {
     private ListenableFuture<DOMRpcResult> invokeSalRemoteRpcSubscribeRPC(final NormalizedNodeContext payload) {
         final ContainerNode value = (ContainerNode) payload.getData();
         final QName rpcQName = payload.getInstanceIdentifierContext().getSchemaNode().getQName();
-        final Optional<DataContainerChild<? extends PathArgument, ?>> path = value.getChild(
-            new NodeIdentifier(QName.create(payload.getInstanceIdentifierContext().getSchemaNode().getQName(),
-                "path")));
-        final Object pathValue = path.isPresent() ? path.get().getValue() : null;
+        final Optional<DataContainerChild> path =
+            value.findChildByArg(new NodeIdentifier(QName.create(rpcQName, "path")));
+        final Object pathValue = path.isPresent() ? path.get().body() : null;
 
         if (!(pathValue instanceof YangInstanceIdentifier)) {
             LOG.debug("Instance identifier {} was not normalized correctly", rpcQName);
@@ -593,14 +611,14 @@ public final class RestconfImpl implements RestconfService {
         NotificationOutputType outputType = null;
         if (!pathIdentifier.isEmpty()) {
             final String fullRestconfIdentifier =
-                    DATA_SUBSCR + this.controllerContext.toFullRestconfIdentifier(pathIdentifier, null);
+                    DATA_SUBSCR + controllerContext.toFullRestconfIdentifier(pathIdentifier, null);
 
             LogicalDatastoreType datastore =
                     parseEnumTypeParameter(value, LogicalDatastoreType.class, DATASTORE_PARAM_NAME);
             datastore = datastore == null ? DEFAULT_DATASTORE : datastore;
 
-            DataChangeScope scope = parseEnumTypeParameter(value, DataChangeScope.class, SCOPE_PARAM_NAME);
-            scope = scope == null ? DEFAULT_SCOPE : scope;
+            Scope scope = parseEnumTypeParameter(value, Scope.class, SCOPE_PARAM_NAME);
+            scope = scope == null ? Scope.BASE : scope;
 
             outputType = parseEnumTypeParameter(value, NotificationOutputType.class, OUTPUT_TYPE_PARAM_NAME);
             outputType = outputType == null ? NotificationOutputType.XML : outputType;
@@ -669,6 +687,8 @@ public final class RestconfImpl implements RestconfService {
                     break;
             }
         }
+
+        // TODO: this flag is always ignored
         boolean tagged = false;
         if (withDefaUsed) {
             if ("report-all-tagged".equals(withDefa)) {
@@ -680,32 +700,31 @@ public final class RestconfImpl implements RestconfService {
             }
         }
 
-        final InstanceIdentifierContext<?> iiWithData = this.controllerContext.toInstanceIdentifier(identifier);
+        final InstanceIdentifierContext iiWithData = controllerContext.toInstanceIdentifier(identifier);
         final DOMMountPoint mountPoint = iiWithData.getMountPoint();
-        NormalizedNode<?, ?> data = null;
+        NormalizedNode data = null;
         final YangInstanceIdentifier normalizedII = iiWithData.getInstanceIdentifier();
         if (mountPoint != null) {
-            data = this.broker.readConfigurationData(mountPoint, normalizedII, withDefa);
+            data = broker.readConfigurationData(mountPoint, normalizedII, withDefa);
         } else {
-            data = this.broker.readConfigurationData(normalizedII, withDefa);
+            data = broker.readConfigurationData(normalizedII, withDefa);
         }
         if (data == null) {
             throw dataMissing(identifier);
         }
-        return new NormalizedNodeContext(iiWithData, data,
-                QueryParametersParser.parseWriterParameters(uriInfo, tagged));
+        return new NormalizedNodeContext(iiWithData, data, QueryParametersParser.parseWriterParameters(uriInfo));
     }
 
     @Override
     public NormalizedNodeContext readOperationalData(final String identifier, final UriInfo uriInfo) {
-        final InstanceIdentifierContext<?> iiWithData = this.controllerContext.toInstanceIdentifier(identifier);
+        final InstanceIdentifierContext iiWithData = controllerContext.toInstanceIdentifier(identifier);
         final DOMMountPoint mountPoint = iiWithData.getMountPoint();
-        NormalizedNode<?, ?> data = null;
+        NormalizedNode data = null;
         final YangInstanceIdentifier normalizedII = iiWithData.getInstanceIdentifier();
         if (mountPoint != null) {
-            data = this.broker.readOperationalData(mountPoint, normalizedII);
+            data = broker.readOperationalData(mountPoint, normalizedII);
         } else {
-            data = this.broker.readOperationalData(normalizedII);
+            data = broker.readOperationalData(normalizedII);
         }
         if (data == null) {
             throw dataMissing(identifier);
@@ -759,9 +778,9 @@ public final class RestconfImpl implements RestconfService {
                     "Point parameter can be used only with 'after' or 'before' values of Insert parameter.");
         }
 
-        Preconditions.checkNotNull(identifier);
+        requireNonNull(identifier);
 
-        final InstanceIdentifierContext<?> iiWithData = payload.getInstanceIdentifierContext();
+        final InstanceIdentifierContext iiWithData = payload.getInstanceIdentifierContext();
 
         validateInput(iiWithData.getSchemaNode(), payload);
         validateTopLevelNodeName(payload, iiWithData.getInstanceIdentifier());
@@ -791,10 +810,10 @@ public final class RestconfImpl implements RestconfService {
         while (true) {
             if (mountPoint != null) {
 
-                result = this.broker.commitMountPointDataPut(mountPoint, normalizedII, payload.getData(), insert,
+                result = broker.commitMountPointDataPut(mountPoint, normalizedII, payload.getData(), insert,
                         point);
             } else {
-                result = this.broker.commitConfigurationDataPut(this.controllerContext.getGlobalSchema(), normalizedII,
+                result = broker.commitConfigurationDataPut(controllerContext.getGlobalSchema(), normalizedII,
                         payload.getData(), insert, point);
             }
 
@@ -827,12 +846,12 @@ public final class RestconfImpl implements RestconfService {
     private static void validateTopLevelNodeName(final NormalizedNodeContext node,
             final YangInstanceIdentifier identifier) {
 
-        final String payloadName = node.getData().getNodeType().getLocalName();
+        final String payloadName = node.getData().getIdentifier().getNodeType().getLocalName();
 
         // no arguments
         if (identifier.isEmpty()) {
             // no "data" payload
-            if (!node.getData().getNodeType().equals(NETCONF_BASE_QNAME)) {
+            if (!node.getData().getIdentifier().getNodeType().equals(NETCONF_BASE_QNAME)) {
                 throw new RestconfDocumentedException("Instance identifier has to contain at least one path argument",
                         ErrorType.PROTOCOL, ErrorTag.MALFORMED_MESSAGE);
             }
@@ -856,11 +875,11 @@ public final class RestconfImpl implements RestconfService {
      *
      */
     private static void validateListKeysEqualityInPayloadAndUri(final NormalizedNodeContext payload) {
-        Preconditions.checkArgument(payload != null);
-        final InstanceIdentifierContext<?> iiWithData = payload.getInstanceIdentifierContext();
+        checkArgument(payload != null);
+        final InstanceIdentifierContext iiWithData = payload.getInstanceIdentifierContext();
         final PathArgument lastPathArgument = iiWithData.getInstanceIdentifier().getLastPathArgument();
         final SchemaNode schemaNode = iiWithData.getSchemaNode();
-        final NormalizedNode<?, ?> data = payload.getData();
+        final NormalizedNode data = payload.getData();
         if (schemaNode instanceof ListSchemaNode) {
             final List<QName> keyDefinitions = ((ListSchemaNode) schemaNode).getKeyDefinition();
             if (lastPathArgument instanceof NodeIdentifierWithPredicates && data instanceof MapEntryNode) {
@@ -874,7 +893,7 @@ public final class RestconfImpl implements RestconfService {
     public static void isEqualUriAndPayloadKeyValues(final Map<QName, Object> uriKeyValues, final MapEntryNode payload,
             final List<QName> keyDefinitions) {
 
-        final Map<QName, Object> mutableCopyUriKeyValues = Maps.newHashMap(uriKeyValues);
+        final Map<QName, Object> mutableCopyUriKeyValues = new HashMap<>(uriKeyValues);
         for (final QName keyDefinition : keyDefinitions) {
             final Object uriKeyValue = RestconfDocumentedException.throwIfNull(
                 // should be caught during parsing URI to InstanceIdentifier
@@ -904,7 +923,7 @@ public final class RestconfImpl implements RestconfService {
             throw new RestconfDocumentedException("Input is required.", ErrorType.PROTOCOL, ErrorTag.MALFORMED_MESSAGE);
         }
         final DOMMountPoint mountPoint = payload.getInstanceIdentifierContext().getMountPoint();
-        final InstanceIdentifierContext<?> iiWithData = payload.getInstanceIdentifierContext();
+        final InstanceIdentifierContext iiWithData = payload.getInstanceIdentifierContext();
         final YangInstanceIdentifier normalizedII = iiWithData.getInstanceIdentifier();
 
         boolean insertUsed = false;
@@ -947,10 +966,10 @@ public final class RestconfImpl implements RestconfService {
 
         FluentFuture<? extends CommitInfo> future;
         if (mountPoint != null) {
-            future = this.broker.commitConfigurationDataPost(mountPoint, normalizedII, payload.getData(), insert,
+            future = broker.commitConfigurationDataPost(mountPoint, normalizedII, payload.getData(), insert,
                     point);
         } else {
-            future = this.broker.commitConfigurationDataPost(this.controllerContext.getGlobalSchema(), normalizedII,
+            future = broker.commitConfigurationDataPost(controllerContext.getGlobalSchema(), normalizedII,
                     payload.getData(), insert, point);
         }
 
@@ -987,7 +1006,7 @@ public final class RestconfImpl implements RestconfService {
         final UriBuilder uriBuilder = uriInfo.getBaseUriBuilder();
         uriBuilder.path("config");
         try {
-            uriBuilder.path(this.controllerContext.toFullRestconfIdentifier(normalizedII, mountPoint));
+            uriBuilder.path(controllerContext.toFullRestconfIdentifier(normalizedII, mountPoint));
         } catch (final Exception e) {
             LOG.info("Location for instance identifier {} was not created", normalizedII, e);
             return null;
@@ -997,15 +1016,15 @@ public final class RestconfImpl implements RestconfService {
 
     @Override
     public Response deleteConfigurationData(final String identifier) {
-        final InstanceIdentifierContext<?> iiWithData = this.controllerContext.toInstanceIdentifier(identifier);
+        final InstanceIdentifierContext iiWithData = controllerContext.toInstanceIdentifier(identifier);
         final DOMMountPoint mountPoint = iiWithData.getMountPoint();
         final YangInstanceIdentifier normalizedII = iiWithData.getInstanceIdentifier();
 
         final FluentFuture<? extends CommitInfo> future;
         if (mountPoint != null) {
-            future = this.broker.commitConfigurationDataDelete(mountPoint, normalizedII);
+            future = broker.commitConfigurationDataDelete(mountPoint, normalizedII);
         } else {
-            future = this.broker.commitConfigurationDataDelete(normalizedII);
+            future = broker.commitConfigurationDataDelete(normalizedII);
         }
 
         try {
@@ -1037,7 +1056,7 @@ public final class RestconfImpl implements RestconfService {
      * <ul>
      * <li>datastore - default CONFIGURATION (other values of
      * {@link LogicalDatastoreType} enum type)</li>
-     * <li>scope - default BASE (other values of {@link DataChangeScope})</li>
+     * <li>scope - default BASE (other values of {@link Scope})</li>
      * </ul>
      */
     @Override
@@ -1112,17 +1131,14 @@ public final class RestconfImpl implements RestconfService {
 
         if (response != null) {
             // prepare node with value of location
-            final InstanceIdentifierContext<?> iid = prepareIIDSubsStreamOutput();
+            final InstanceIdentifierContext iid = prepareIIDSubsStreamOutput();
             final NormalizedNodeBuilder<NodeIdentifier, Object, LeafNode<Object>> builder =
                     ImmutableLeafNodeBuilder.create().withValue(response.toString());
             builder.withNodeIdentifier(
                     NodeIdentifier.create(QName.create("subscribe:to:notification", "2016-10-28", "location")));
 
             // prepare new header with location
-            final Map<String, Object> headers = new HashMap<>();
-            headers.put("Location", response);
-
-            return new NormalizedNodeContext(iid, builder.build(), headers);
+            return new NormalizedNodeContext(iid, builder.build(), ImmutableMap.of("Location", response));
         }
 
         final String msg = "Bad type of notification of sal-remote";
@@ -1148,7 +1164,7 @@ public final class RestconfImpl implements RestconfService {
      * @return {@link InstanceIdentifierContext} of location leaf for
      *         notification
      */
-    private InstanceIdentifierContext<?> prepareIIDSubsStreamOutput() {
+    private InstanceIdentifierContext prepareIIDSubsStreamOutput() {
         final QName qnameBase = QName.create("subscribe:to:notification", "2016-10-28", "notifi");
         final EffectiveModelContext schemaCtx = controllerContext.getGlobalSchema();
         final DataSchemaNode location = ((ContainerSchemaNode) schemaCtx
@@ -1158,8 +1174,7 @@ public final class RestconfImpl implements RestconfService {
         path.add(NodeIdentifier.create(qnameBase));
         path.add(NodeIdentifier.create(QName.create(qnameBase, "location")));
 
-        return new InstanceIdentifierContext<SchemaNode>(YangInstanceIdentifier.create(path), location, null,
-                schemaCtx);
+        return new InstanceIdentifierContext(YangInstanceIdentifier.create(path), location, null, schemaCtx);
     }
 
     /**
@@ -1190,7 +1205,7 @@ public final class RestconfImpl implements RestconfService {
         }
 
         for (final NotificationListenerAdapter listener : listeners) {
-            this.broker.registerToListenNotification(listener);
+            broker.registerToListenNotification(listener);
             listener.setQueryParams(start, Optional.ofNullable(stop), Optional.ofNullable(filter), false, false);
         }
 
@@ -1251,14 +1266,13 @@ public final class RestconfImpl implements RestconfService {
             throw new RestconfDocumentedException("Stream name doesn't contains datastore value (pattern /datastore=)",
                     ErrorType.APPLICATION, ErrorTag.MISSING_ATTRIBUTE);
         }
-        final DataChangeScope scope =
-                parserURIEnumParameter(DataChangeScope.class, paramToValues.get(SCOPE_PARAM_NAME));
+        final Scope scope = parserURIEnumParameter(Scope.class, paramToValues.get(SCOPE_PARAM_NAME));
         if (scope == null) {
             throw new RestconfDocumentedException("Stream name doesn't contains datastore value (pattern /scope=)",
                     ErrorType.APPLICATION, ErrorTag.MISSING_ATTRIBUTE);
         }
 
-        this.broker.registerToListenDataChanges(datastore, scope, listener);
+        broker.registerToListenDataChanges(datastore, scope, listener);
 
         final UriBuilder uriBuilder = uriInfo.getAbsolutePathBuilder();
 
@@ -1279,7 +1293,7 @@ public final class RestconfImpl implements RestconfService {
         }
 
         try {
-            return this.broker.patchConfigurationDataWithinTransaction(context);
+            return broker.patchConfigurationDataWithinTransaction(context);
         } catch (final Exception e) {
             LOG.debug("Patch transaction failed", e);
             throw new RestconfDocumentedException(e.getMessage(), e);
@@ -1294,7 +1308,7 @@ public final class RestconfImpl implements RestconfService {
         }
 
         try {
-            return this.broker.patchConfigurationDataWithinTransaction(context);
+            return broker.patchConfigurationDataWithinTransaction(context);
         } catch (final Exception e) {
             LOG.debug("Patch transaction failed", e);
             throw new RestconfDocumentedException(e.getMessage(), e);
@@ -1311,21 +1325,20 @@ public final class RestconfImpl implements RestconfService {
      */
     private static <T> T parseEnumTypeParameter(final ContainerNode value, final Class<T> classDescriptor,
             final String paramName) {
-        final Optional<DataContainerChild<? extends PathArgument, ?>> optAugNode = value.getChild(
-            SAL_REMOTE_AUG_IDENTIFIER);
-        if (!optAugNode.isPresent()) {
+        final Optional<DataContainerChild> optAugNode = value.findChildByArg(SAL_REMOTE_AUG_IDENTIFIER);
+        if (optAugNode.isEmpty()) {
             return null;
         }
-        final DataContainerChild<? extends PathArgument, ?> augNode = optAugNode.get();
+        final DataContainerChild augNode = optAugNode.get();
         if (!(augNode instanceof AugmentationNode)) {
             return null;
         }
-        final Optional<DataContainerChild<? extends PathArgument, ?>> enumNode = ((AugmentationNode) augNode).getChild(
+        final Optional<DataContainerChild> enumNode = ((AugmentationNode) augNode).findChildByArg(
                 new NodeIdentifier(QName.create(SAL_REMOTE_AUGMENT, paramName)));
-        if (!enumNode.isPresent()) {
+        if (enumNode.isEmpty()) {
             return null;
         }
-        final Object rawValue = enumNode.get().getValue();
+        final Object rawValue = enumNode.get().body();
         if (!(rawValue instanceof String)) {
             return null;
         }
@@ -1372,14 +1385,14 @@ public final class RestconfImpl implements RestconfService {
     }
 
     private MapNode makeModuleMapNode(final Collection<? extends Module> modules) {
-        Preconditions.checkNotNull(modules);
+        requireNonNull(modules);
         final Module restconfModule = getRestconfModule();
-        final DataSchemaNode moduleSchemaNode = this.controllerContext
+        final DataSchemaNode moduleSchemaNode = controllerContext
                 .getRestconfModuleRestConfSchemaNode(restconfModule, Draft02.RestConfModule.MODULE_LIST_SCHEMA_NODE);
-        Preconditions.checkState(moduleSchemaNode instanceof ListSchemaNode);
+        checkState(moduleSchemaNode instanceof ListSchemaNode);
 
-        final CollectionNodeBuilder<MapEntryNode, MapNode> listModuleBuilder =
-                Builders.mapBuilder((ListSchemaNode) moduleSchemaNode);
+        final CollectionNodeBuilder<MapEntryNode, SystemMapNode> listModuleBuilder =
+                SchemaAwareBuilders.mapBuilder((ListSchemaNode) moduleSchemaNode);
 
         for (final Module module : modules) {
             listModuleBuilder.withChild(toModuleEntryNode(module, moduleSchemaNode));
@@ -1387,45 +1400,45 @@ public final class RestconfImpl implements RestconfService {
         return listModuleBuilder.build();
     }
 
-    private MapEntryNode toModuleEntryNode(final Module module, final DataSchemaNode moduleSchemaNode) {
-        Preconditions.checkArgument(moduleSchemaNode instanceof ListSchemaNode,
+    private static MapEntryNode toModuleEntryNode(final Module module, final DataSchemaNode moduleSchemaNode) {
+        checkArgument(moduleSchemaNode instanceof ListSchemaNode,
                 "moduleSchemaNode has to be of type ListSchemaNode");
         final ListSchemaNode listModuleSchemaNode = (ListSchemaNode) moduleSchemaNode;
         final DataContainerNodeBuilder<NodeIdentifierWithPredicates, MapEntryNode> moduleNodeValues =
-                Builders.mapEntryBuilder(listModuleSchemaNode);
+                SchemaAwareBuilders.mapEntryBuilder(listModuleSchemaNode);
 
         List<DataSchemaNode> instanceDataChildrenByName =
                 ControllerContext.findInstanceDataChildrenByName(listModuleSchemaNode, "name");
         final DataSchemaNode nameSchemaNode = Iterables.getFirst(instanceDataChildrenByName, null);
-        Preconditions.checkState(nameSchemaNode instanceof LeafSchemaNode);
-        moduleNodeValues
-                .withChild(Builders.leafBuilder((LeafSchemaNode) nameSchemaNode).withValue(module.getName()).build());
+        checkState(nameSchemaNode instanceof LeafSchemaNode);
+        moduleNodeValues.withChild(
+            SchemaAwareBuilders.leafBuilder((LeafSchemaNode) nameSchemaNode).withValue(module.getName()).build());
 
         final QNameModule qNameModule = module.getQNameModule();
 
         instanceDataChildrenByName =
                 ControllerContext.findInstanceDataChildrenByName(listModuleSchemaNode, "revision");
         final DataSchemaNode revisionSchemaNode = Iterables.getFirst(instanceDataChildrenByName, null);
-        Preconditions.checkState(revisionSchemaNode instanceof LeafSchemaNode);
+        checkState(revisionSchemaNode instanceof LeafSchemaNode);
         final Optional<Revision> revision = qNameModule.getRevision();
-        moduleNodeValues.withChild(Builders.leafBuilder((LeafSchemaNode) revisionSchemaNode)
+        moduleNodeValues.withChild(SchemaAwareBuilders.leafBuilder((LeafSchemaNode) revisionSchemaNode)
                 .withValue(revision.map(Revision::toString).orElse("")).build());
 
         instanceDataChildrenByName =
                 ControllerContext.findInstanceDataChildrenByName(listModuleSchemaNode, "namespace");
         final DataSchemaNode namespaceSchemaNode = Iterables.getFirst(instanceDataChildrenByName, null);
-        Preconditions.checkState(namespaceSchemaNode instanceof LeafSchemaNode);
-        moduleNodeValues.withChild(Builders.leafBuilder((LeafSchemaNode) namespaceSchemaNode)
+        checkState(namespaceSchemaNode instanceof LeafSchemaNode);
+        moduleNodeValues.withChild(SchemaAwareBuilders.leafBuilder((LeafSchemaNode) namespaceSchemaNode)
                 .withValue(qNameModule.getNamespace().toString()).build());
 
         instanceDataChildrenByName =
                 ControllerContext.findInstanceDataChildrenByName(listModuleSchemaNode, "feature");
         final DataSchemaNode featureSchemaNode = Iterables.getFirst(instanceDataChildrenByName, null);
-        Preconditions.checkState(featureSchemaNode instanceof LeafListSchemaNode);
-        final ListNodeBuilder<Object, LeafSetEntryNode<Object>> featuresBuilder =
-                Builders.leafSetBuilder((LeafListSchemaNode) featureSchemaNode);
+        checkState(featureSchemaNode instanceof LeafListSchemaNode);
+        final ListNodeBuilder<Object, SystemLeafSetNode<Object>> featuresBuilder =
+                SchemaAwareBuilders.leafSetBuilder((LeafListSchemaNode) featureSchemaNode);
         for (final FeatureDefinition feature : module.getFeatures()) {
-            featuresBuilder.withChild(Builders.leafSetEntryBuilder((LeafListSchemaNode) featureSchemaNode)
+            featuresBuilder.withChild(SchemaAwareBuilders.leafSetEntryBuilder((LeafListSchemaNode) featureSchemaNode)
                     .withValue(feature.getQName().getLocalName()).build());
         }
         moduleNodeValues.withChild(featuresBuilder.build());
@@ -1434,44 +1447,46 @@ public final class RestconfImpl implements RestconfService {
     }
 
     protected MapEntryNode toStreamEntryNode(final String streamName, final DataSchemaNode streamSchemaNode) {
-        Preconditions.checkArgument(streamSchemaNode instanceof ListSchemaNode,
+        checkArgument(streamSchemaNode instanceof ListSchemaNode,
                 "streamSchemaNode has to be of type ListSchemaNode");
         final ListSchemaNode listStreamSchemaNode = (ListSchemaNode) streamSchemaNode;
         final DataContainerNodeBuilder<NodeIdentifierWithPredicates, MapEntryNode> streamNodeValues =
-                Builders.mapEntryBuilder(listStreamSchemaNode);
+                SchemaAwareBuilders.mapEntryBuilder(listStreamSchemaNode);
 
         List<DataSchemaNode> instanceDataChildrenByName =
                 ControllerContext.findInstanceDataChildrenByName(listStreamSchemaNode, "name");
         final DataSchemaNode nameSchemaNode = Iterables.getFirst(instanceDataChildrenByName, null);
-        Preconditions.checkState(nameSchemaNode instanceof LeafSchemaNode);
-        streamNodeValues.withChild(Builders.leafBuilder((LeafSchemaNode) nameSchemaNode).withValue(streamName).build());
+        checkState(nameSchemaNode instanceof LeafSchemaNode);
+        streamNodeValues.withChild(
+            SchemaAwareBuilders.leafBuilder((LeafSchemaNode) nameSchemaNode).withValue(streamName).build());
 
         instanceDataChildrenByName =
                 ControllerContext.findInstanceDataChildrenByName(listStreamSchemaNode, "description");
         final DataSchemaNode descriptionSchemaNode = Iterables.getFirst(instanceDataChildrenByName, null);
-        Preconditions.checkState(descriptionSchemaNode instanceof LeafSchemaNode);
-        streamNodeValues.withChild(
-                Builders.leafBuilder((LeafSchemaNode) nameSchemaNode).withValue("DESCRIPTION_PLACEHOLDER").build());
+        checkState(descriptionSchemaNode instanceof LeafSchemaNode);
+        streamNodeValues.withChild(SchemaAwareBuilders.leafBuilder((LeafSchemaNode) nameSchemaNode)
+            .withValue("DESCRIPTION_PLACEHOLDER")
+            .build());
 
         instanceDataChildrenByName =
                 ControllerContext.findInstanceDataChildrenByName(listStreamSchemaNode, "replay-support");
         final DataSchemaNode replaySupportSchemaNode = Iterables.getFirst(instanceDataChildrenByName, null);
-        Preconditions.checkState(replaySupportSchemaNode instanceof LeafSchemaNode);
-        streamNodeValues.withChild(Builders.leafBuilder((LeafSchemaNode) replaySupportSchemaNode)
+        checkState(replaySupportSchemaNode instanceof LeafSchemaNode);
+        streamNodeValues.withChild(SchemaAwareBuilders.leafBuilder((LeafSchemaNode) replaySupportSchemaNode)
                 .withValue(Boolean.TRUE).build());
 
         instanceDataChildrenByName =
                 ControllerContext.findInstanceDataChildrenByName(listStreamSchemaNode, "replay-log-creation-time");
         final DataSchemaNode replayLogCreationTimeSchemaNode = Iterables.getFirst(instanceDataChildrenByName, null);
-        Preconditions.checkState(replayLogCreationTimeSchemaNode instanceof LeafSchemaNode);
+        checkState(replayLogCreationTimeSchemaNode instanceof LeafSchemaNode);
         streamNodeValues.withChild(
-                Builders.leafBuilder((LeafSchemaNode) replayLogCreationTimeSchemaNode).withValue("").build());
+            SchemaAwareBuilders.leafBuilder((LeafSchemaNode) replayLogCreationTimeSchemaNode).withValue("").build());
 
         instanceDataChildrenByName = ControllerContext.findInstanceDataChildrenByName(listStreamSchemaNode, "events");
         final DataSchemaNode eventsSchemaNode = Iterables.getFirst(instanceDataChildrenByName, null);
-        Preconditions.checkState(eventsSchemaNode instanceof LeafSchemaNode);
+        checkState(eventsSchemaNode instanceof LeafSchemaNode);
         streamNodeValues.withChild(
-                Builders.leafBuilder((LeafSchemaNode) eventsSchemaNode).withValue(Empty.getInstance()).build());
+            SchemaAwareBuilders.leafBuilder((LeafSchemaNode) eventsSchemaNode).withValue(Empty.value()).build());
 
         return streamNodeValues.build();
     }
@@ -1487,25 +1502,25 @@ public final class RestconfImpl implements RestconfService {
         final ContainerNode data = (ContainerNode) payload.getData();
         LeafSetNode leafSet = null;
         String outputType = "XML";
-        for (final DataContainerChild<? extends PathArgument, ?> dataChild : data.getValue()) {
+        for (final DataContainerChild dataChild : data.body()) {
             if (dataChild instanceof LeafSetNode) {
                 leafSet = (LeafSetNode) dataChild;
             } else if (dataChild instanceof AugmentationNode) {
-                outputType = (String) ((AugmentationNode) dataChild).getValue().iterator().next().getValue();
+                outputType = (String) ((AugmentationNode) dataChild).body().iterator().next().body();
             }
         }
 
-        final Collection<LeafSetEntryNode> entryNodes = leafSet.getValue();
+        final Collection<LeafSetEntryNode<?>> entryNodes = leafSet.body();
         final List<SchemaPath> paths = new ArrayList<>();
         String streamName = CREATE_NOTIFICATION_STREAM + "/";
 
         StringBuilder streamNameBuilder = new StringBuilder(streamName);
-        final Iterator<LeafSetEntryNode> iterator = entryNodes.iterator();
+        final Iterator<LeafSetEntryNode<?>> iterator = entryNodes.iterator();
         while (iterator.hasNext()) {
-            final QName valueQName = QName.create((String) iterator.next().getValue());
-            final Module module = controllerContext.findModuleByNamespace(valueQName.getModule().getNamespace());
-            Preconditions.checkNotNull(module,
-                    "Module for namespace " + valueQName.getModule().getNamespace() + " does not exist");
+            final QName valueQName = QName.create((String) iterator.next().body());
+            final XMLNamespace namespace = valueQName.getModule().getNamespace();
+            final Module module = controllerContext.findModuleByNamespace(namespace);
+            checkNotNull(module, "Module for namespace %s does not exist", namespace);
             NotificationDefinition notifiDef = null;
             for (final NotificationDefinition notification : module.getNotifications()) {
                 if (notification.getQName().equals(valueQName)) {
@@ -1514,9 +1529,8 @@ public final class RestconfImpl implements RestconfService {
                 }
             }
             final String moduleName = module.getName();
-            Preconditions.checkNotNull(notifiDef,
-                    "Notification " + valueQName + "doesn't exist in module " + moduleName);
-            paths.add(notifiDef.getPath());
+            checkNotNull(notifiDef, "Notification %s does not exist in module %s", valueQName, moduleName);
+            paths.add(SchemaPath.of(Absolute.of(notifiDef.getQName())));
             streamNameBuilder.append(moduleName).append(':').append(valueQName.getLocalName());
             if (iterator.hasNext()) {
                 streamNameBuilder.append(',');
@@ -1539,4 +1553,10 @@ public final class RestconfImpl implements RestconfService {
 
         return Futures.immediateFuture(new DefaultDOMRpcResult(output));
     }
+
+    private static EffectiveModelContext modelContext(final DOMMountPoint mountPoint) {
+        return mountPoint.getService(DOMSchemaService.class)
+            .flatMap(svc -> Optional.ofNullable(svc.getGlobalContext()))
+            .orElse(null);
+    }
 }