codec/yang-data-codec-xml/src/main/java/org/opendaylight/yangtools/yang/data/codec/xml/SchemalessXMLStreamNormalizedNodeStreamWriter.java

   1 /*
   2  * Copyright (c) 2016 Brocade Communications 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.codec.xml;
   9
  10 import static com.google.common.base.Preconditions.checkState;
  11
  12 import java.io.IOException;
  13 import java.util.ArrayDeque;
  14 import java.util.Deque;
  15 import javax.xml.stream.XMLStreamWriter;
  16 import javax.xml.transform.dom.DOMSource;
  17 import org.opendaylight.yangtools.yang.common.QName;
  18 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  19 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeWithValue;
  20 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
  21
  22 final class SchemalessXMLStreamNormalizedNodeStreamWriter extends XMLStreamNormalizedNodeStreamWriter<Object> {
  23     private enum NodeType {
  24         CONTAINER,
  25         LEAF_SET,
  26         LIST,
  27         LIST_ITEM,
  28         YANG_MODELED_ANY_XML,
  29         CHOICE,
  30         AUGMENTATION,
  31         SCALAR,
  32         ANY_XML,
  33         ANYDATA,
  34     }
  35
  36     private final Deque<NodeType> nodeTypeStack = new ArrayDeque<>();
  37
  38     SchemalessXMLStreamNormalizedNodeStreamWriter(final XMLStreamWriter writer) {
  39         super(PreferredPrefixes.empty(), writer);
  40     }
  41
  42     @Override
  43     public void startLeafNode(final NodeIdentifier name) throws IOException {
  44         nodeTypeStack.push(NodeType.SCALAR);
  45         startElement(name.getNodeType());
  46     }
  47
  48     @Override
  49     public void startLeafSetEntryNode(final NodeWithValue<?> name) throws IOException {
  50         nodeTypeStack.push(NodeType.SCALAR);
  51         startElement(name.getNodeType());
  52     }
  53
  54     @Override
  55     public void startLeafSet(final NodeIdentifier name, final int childSizeHint) throws IOException {
  56         nodeTypeStack.push(NodeType.LEAF_SET);
  57     }
  58
  59     @Override
  60     public void startOrderedLeafSet(final NodeIdentifier name, final int childSizeHint) throws IOException {
  61         nodeTypeStack.push(NodeType.LEAF_SET);
  62     }
  63
  64     @Override
  65     public void startContainerNode(final NodeIdentifier name, final int childSizeHint) throws IOException {
  66         nodeTypeStack.push(NodeType.CONTAINER);
  67         startElement(name.getNodeType());
  68     }
  69
  70     @Override
  71     public void startChoiceNode(final NodeIdentifier name, final int childSizeHint) throws IOException {
  72         nodeTypeStack.push(NodeType.CHOICE);
  73     }
  74
  75     @Override
  76     public boolean startAnyxmlNode(final NodeIdentifier name, final Class<?> objectModel) throws IOException {
  77         if (DOMSource.class.isAssignableFrom(objectModel)) {
  78             nodeTypeStack.push(NodeType.ANY_XML);
  79             startElement(name.getNodeType());
  80             return true;
  81         }
  82         return false;
  83     }
  84
  85     @Override
  86     String encodeValue(final ValueWriter xmlWriter, final Object value, final Object context) {
  87         return value.toString();
  88     }
  89
  90     @Override
  91     String encodeAnnotationValue(final ValueWriter xmlWriter, final QName qname, final Object value) {
  92         return value.toString();
  93     }
  94
  95     @Override
  96     void startList(final NodeIdentifier name) {
  97         nodeTypeStack.push(NodeType.LIST);
  98     }
  99
 100     @Override
 101     void startListItem(final PathArgument name) throws IOException {
 102         nodeTypeStack.push(NodeType.LIST_ITEM);
 103         startElement(name.getNodeType());
 104     }
 105
 106     @Override
 107     public void scalarValue(final Object value) throws IOException {
 108         final NodeType type = nodeTypeStack.peek();
 109         if (type == NodeType.SCALAR) {
 110             writeValue(value, null);
 111         } else if (type == NodeType.ANYDATA) {
 112             anydataValue(value);
 113         } else {
 114             throw new IllegalStateException("Unexpected scalar " + value + " in type " + type);
 115         }
 116     }
 117
 118     @Override
 119     public void domSourceValue(final DOMSource value) throws IOException {
 120         final NodeType type = nodeTypeStack.peek();
 121         checkState(type == NodeType.ANY_XML, "Unexpected DOMSource %s in %s", value, type);
 122         anyxmlValue(value);
 123     }
 124
 125     @Override
 126     public void endNode() throws IOException {
 127         NodeType type = nodeTypeStack.pop();
 128         switch (type) {
 129             case CONTAINER:
 130             case LIST_ITEM:
 131             case SCALAR:
 132             case ANY_XML:
 133             case ANYDATA:
 134                 endElement();
 135                 break;
 136             default:
 137                 break;
 138         }
 139     }
 140
 141     @Override
 142     void startAnydata(final NodeIdentifier name) {
 143         nodeTypeStack.push(NodeType.ANYDATA);
 144     }
 145 }