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

   1 /*
   2  * Copyright (c) 2017 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.codec.xml;
   9
  10 import static com.google.common.base.Verify.verifyNotNull;
  11 import static java.util.Objects.requireNonNull;
  12
  13 import java.net.URI;
  14 import java.util.Map.Entry;
  15 import javax.xml.stream.XMLStreamException;
  16 import javax.xml.stream.XMLStreamWriter;
  17 import org.eclipse.jdt.annotation.NonNull;
  18 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  19 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  20 import org.opendaylight.yangtools.yang.model.api.SchemaContextProvider;
  21 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
  22 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
  23 import org.opendaylight.yangtools.yang.model.api.type.LeafrefTypeDefinition;
  24 import org.opendaylight.yangtools.yang.model.util.SchemaContextUtil;
  25
  26 final class SchemaAwareXMLStreamWriterUtils extends XMLStreamWriterUtils implements SchemaContextProvider {
  27     private final @NonNull SchemaContext schemaContext;
  28
  29     SchemaAwareXMLStreamWriterUtils(final SchemaContext schemaContext) {
  30         this.schemaContext = requireNonNull(schemaContext);
  31     }
  32
  33     @Override
  34     TypeDefinition<?> getBaseTypeForLeafRef(final SchemaNode schemaNode, final LeafrefTypeDefinition type) {
  35         final TypeDefinition<?> ret = SchemaContextUtil.getBaseTypeForLeafRef(type, schemaContext, schemaNode);
  36         return verifyNotNull(ret, "Unable to find base type for leafref node '%s'.", schemaNode.getPath());
  37     }
  38
  39     @Override
  40     void writeInstanceIdentifier(final XMLStreamWriter writer, final YangInstanceIdentifier value)
  41             throws XMLStreamException {
  42         RandomPrefixInstanceIdentifierSerializer iiCodec = new RandomPrefixInstanceIdentifierSerializer(schemaContext,
  43             writer.getNamespaceContext());
  44         String serializedValue = iiCodec.serialize(value);
  45
  46         for (Entry<URI, String> e : iiCodec.getPrefixes()) {
  47             writer.writeNamespace(e.getValue(), e.getKey().toString());
  48         }
  49
  50         writer.writeCharacters(serializedValue);
  51     }
  52
  53     @Override
  54     public SchemaContext getSchemaContext() {
  55         return schemaContext;
  56     }
  57 }