Expose DataTreeSnapshot's SchemaContext
[yangtools.git] / yang / yang-data-impl / src / main / java / org / opendaylight / yangtools / yang / data / impl / schema / tree / InMemoryDataTreeSnapshot.java
1 /*
2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.yangtools.yang.data.impl.schema.tree;
9
10 import static com.google.common.base.Preconditions.checkArgument;
11 import static java.util.Objects.requireNonNull;
12
13 import java.util.Optional;
14 import javax.annotation.Nonnull;
15 import org.eclipse.jdt.annotation.NonNull;
16 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
17 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
18 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNodeContainer;
19 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNodes;
20 import org.opendaylight.yangtools.yang.data.api.schema.tree.CursorAwareDataTreeSnapshot;
21 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeSnapshotCursor;
22 import org.opendaylight.yangtools.yang.data.api.schema.tree.spi.TreeNode;
23 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
24
25 final class InMemoryDataTreeSnapshot extends AbstractCursorAware implements CursorAwareDataTreeSnapshot {
26     private final @NonNull RootModificationApplyOperation applyOper;
27     private final @NonNull SchemaContext schemaContext;
28     private final @NonNull TreeNode rootNode;
29
30     InMemoryDataTreeSnapshot(final SchemaContext schemaContext, final TreeNode rootNode,
31             final RootModificationApplyOperation applyOper) {
32         this.schemaContext = requireNonNull(schemaContext);
33         this.rootNode = requireNonNull(rootNode);
34         this.applyOper = requireNonNull(applyOper);
35     }
36
37     TreeNode getRootNode() {
38         return rootNode;
39     }
40
41     @Override
42     public SchemaContext getSchemaContext() {
43         return schemaContext;
44     }
45
46     @Override
47     public Optional<NormalizedNode<?, ?>> readNode(final YangInstanceIdentifier path) {
48         return NormalizedNodes.findNode(rootNode.getData(), path);
49     }
50
51     @Override
52     public InMemoryDataTreeModification newModification() {
53         return new InMemoryDataTreeModification(this, applyOper);
54     }
55
56     @Override
57     public DataTreeSnapshotCursor createCursor(@Nonnull final YangInstanceIdentifier path) {
58         final Optional<NormalizedNode<?, ?>> maybeRoot = NormalizedNodes.findNode(rootNode.getData(), path);
59         if (!maybeRoot.isPresent()) {
60             return null;
61         }
62
63         final NormalizedNode<?, ?> root = maybeRoot.get();
64         checkArgument(root instanceof NormalizedNodeContainer, "Child %s is not a container", path);
65         return openCursor(new InMemoryDataTreeSnapshotCursor(this, path, (NormalizedNodeContainer<?, ?, ?>)root));
66     }
67
68     @Override
69     public String toString() {
70         return rootNode.getSubtreeVersion().toString();
71     }
72 }