import java.util.ArrayDeque;
import java.util.Collection;
import java.util.Deque;
+import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier;
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.PathArgument;
import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
+import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
import org.opendaylight.yangtools.yang.data.api.schema.LeafSetEntryNode;
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.OrderedMapNode;
import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListEntryNode;
import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListNode;
+import org.opendaylight.yangtools.yang.data.api.schema.YangModeledAnyXmlNode;
import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
+import org.opendaylight.yangtools.yang.data.api.schema.stream.SchemaAwareNormalizedNodeStreamWriter;
import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.CollectionNodeBuilder;
import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeAttrBuilder;
import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeBuilder;
import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafSetNodeBuilder;
import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableMapEntryNodeBuilder;
import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableMapNodeBuilder;
+import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableOrderedLeafSetNodeBuilder;
import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableOrderedMapNodeBuilder;
import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableUnkeyedListEntryNodeBuilder;
import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableUnkeyedListNodeBuilder;
+import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableYangModeledAnyXmlNodeBuilder;
+import org.opendaylight.yangtools.yang.data.util.LeafInterner;
+import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
+import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
+import org.opendaylight.yangtools.yang.model.api.YangModeledAnyXmlSchemaNode;
/**
*
*
*
*/
-public class ImmutableNormalizedNodeStreamWriter implements NormalizedNodeStreamWriter {
+public class ImmutableNormalizedNodeStreamWriter implements SchemaAwareNormalizedNodeStreamWriter {
@SuppressWarnings("rawtypes")
private final Deque<NormalizedNodeContainerBuilder> builders = new ArrayDeque<>();
+ private DataSchemaNode nextSchema;
@SuppressWarnings("rawtypes")
protected ImmutableNormalizedNodeStreamWriter(final NormalizedNodeContainerBuilder topLevelBuilder) {
* @param builder Builder to which data will be written.
* @return {@link NormalizedNodeStreamWriter} which writes data
*/
- public static final NormalizedNodeStreamWriter from(final NormalizedNodeContainerBuilder<?, ?, ?, ?> builder) {
+ public static NormalizedNodeStreamWriter from(final NormalizedNodeContainerBuilder<?, ?, ?, ?> builder) {
return new ImmutableNormalizedNodeStreamWriter(builder);
}
/**
- *
* Creates a {@link NormalizedNodeStreamWriter} which creates one instance of top
* level {@link NormalizedNode} (type of NormalizedNode) is determined by first
* start event.
* @param result {@link NormalizedNodeResult} object which will hold result value.
* @return {@link NormalizedNodeStreamWriter} which will write item to supplied result holder.
*/
- public static final NormalizedNodeStreamWriter from(final NormalizedNodeResult result) {
+ public static NormalizedNodeStreamWriter from(final NormalizedNodeResult result) {
return new ImmutableNormalizedNodeStreamWriter(result);
}
-
@SuppressWarnings("rawtypes")
private NormalizedNodeContainerBuilder getCurrent() {
return builders.peek();
@SuppressWarnings("rawtypes")
private void enter(final NormalizedNodeContainerBuilder next) {
builders.push(next);
+ nextSchema = null;
}
@SuppressWarnings("unchecked")
Preconditions.checkState(current != null, "Reached top level node, which could not be closed in this writer.");
final NormalizedNode<PathArgument, ?> product = finishedBuilder.build();
current.addChild(product);
+ nextSchema = null;
}
@Override
public void leafNode(final NodeIdentifier name, final Object value) {
checkDataNodeContainer();
- writeChild(ImmutableNodes.leafNode(name, value));
+
+ final LeafNode<Object> sample = ImmutableNodes.leafNode(name, value);
+ final LeafNode<Object> node;
+ if (nextSchema instanceof LeafSchemaNode) {
+ node = LeafInterner.forSchema((LeafSchemaNode)nextSchema).intern(sample);
+ } else {
+ node = sample;
+ }
+
+ writeChild(node);
+ nextSchema = null;
}
@Override
public void startLeafSet(final NodeIdentifier name, final int childSizeHint) {
checkDataNodeContainer();
final ListNodeBuilder<Object, LeafSetEntryNode<Object>> builder = UNKNOWN_SIZE == childSizeHint ?
- ImmutableLeafSetNodeBuilder.create() : ImmutableLeafSetNodeBuilder.create(childSizeHint);
+ InterningLeafSetNodeBuilder.create(nextSchema) :
+ InterningLeafSetNodeBuilder.create(nextSchema, childSizeHint);
builder.withNodeIdentifier(name);
enter(builder);
}
@Override
- public void leafSetEntryNode(final Object value) {
- Preconditions.checkArgument(getCurrent() instanceof ImmutableLeafSetNodeBuilder<?>);
- @SuppressWarnings("unchecked")
- final ListNodeBuilder<Object, LeafSetEntryNode<Object>> builder = ((ImmutableLeafSetNodeBuilder<Object>) getCurrent());
- builder.withChildValue(value);
+ public void leafSetEntryNode(final QName name, final Object value) {
+ if (getCurrent() instanceof ImmutableOrderedLeafSetNodeBuilder) {
+ @SuppressWarnings("unchecked")
+ ListNodeBuilder<Object, LeafSetEntryNode<Object>> builder = ((ImmutableOrderedLeafSetNodeBuilder<Object>) getCurrent());
+ builder.withChildValue(value);
+ } else if (getCurrent() instanceof ImmutableLeafSetNodeBuilder) {
+ @SuppressWarnings("unchecked")
+ ListNodeBuilder<Object, LeafSetEntryNode<Object>> builder = ((ImmutableLeafSetNodeBuilder<Object>) getCurrent());
+ builder.withChildValue(value);
+ } else {
+ throw new IllegalArgumentException("LeafSetEntryNode is not valid for parent " + getCurrent());
+ }
+
+ nextSchema = null;
+ }
+
+ @Override
+ public void startOrderedLeafSet(final NodeIdentifier name, final int childSizeHint) {
+ checkDataNodeContainer();
+ final ListNodeBuilder<Object, LeafSetEntryNode<Object>> builder = Builders.orderedLeafSetBuilder();
+ builder.withNodeIdentifier(name);
+ enter(builder);
}
@Override
public void anyxmlNode(final NodeIdentifier name, final Object value) {
checkDataNodeContainer();
+ nextSchema = null;
}
@Override
enter(builder.withNodeIdentifier(name));
}
+ @Override
+ public void startYangModeledAnyXmlNode(final NodeIdentifier name, final int childSizeHint) {
+ checkDataNodeContainer();
+
+ Preconditions.checkArgument(nextSchema instanceof YangModeledAnyXmlSchemaNode,
+ "Schema of this node should be instance of YangModeledAnyXmlSchemaNode");
+ final DataContainerNodeAttrBuilder<NodeIdentifier, YangModeledAnyXmlNode> builder = UNKNOWN_SIZE == childSizeHint ? ImmutableYangModeledAnyXmlNodeBuilder
+ .create((YangModeledAnyXmlSchemaNode) nextSchema) : ImmutableYangModeledAnyXmlNodeBuilder.create(
+ (YangModeledAnyXmlSchemaNode) nextSchema, childSizeHint);
+ enter(builder.withNodeIdentifier(name));
+ }
+
@Override
public void startUnkeyedList(final NodeIdentifier name, final int childSizeHint) {
checkDataNodeContainer();
@Override
public void startUnkeyedListItem(final NodeIdentifier name, final int childSizeHint) {
- Preconditions.checkArgument(getCurrent() instanceof ImmutableUnkeyedListNodeBuilder);
-
+ Preconditions.checkArgument((getCurrent() instanceof NormalizedNodeResultBuilder)
+ || getCurrent() instanceof ImmutableUnkeyedListNodeBuilder);
final DataContainerNodeAttrBuilder<NodeIdentifier, UnkeyedListEntryNode> builder = UNKNOWN_SIZE == childSizeHint ?
ImmutableUnkeyedListEntryNodeBuilder.create() : ImmutableUnkeyedListEntryNodeBuilder.create(childSizeHint);
enter(builder.withNodeIdentifier(name));
@Override
public void startMapEntryNode(final NodeIdentifierWithPredicates identifier, final int childSizeHint) {
- if(!(getCurrent() instanceof NormalizedNodeResultBuilder)) {
+ if (!(getCurrent() instanceof NormalizedNodeResultBuilder)) {
Preconditions.checkArgument(getCurrent() instanceof ImmutableMapNodeBuilder || getCurrent() instanceof ImmutableOrderedMapNodeBuilder);
}
public void close() {
// no-op
}
+
+ @Override
+ public void nextDataSchemaNode(final DataSchemaNode schema) {
+ nextSchema = Preconditions.checkNotNull(schema);
+ }
}