private final BiMap<URI, String> prefixes = HashBiMap.create();
private final NamespaceContext context;
- RandomPrefix() {
- this.context = null;
- }
-
RandomPrefix(final NamespaceContext context) {
- this.context = Preconditions.checkNotNull(context);
+ this.context = context;
}
Iterable<Entry<URI, String>> getPrefixes() {
return prefix;
}
+ if (context != null) {
+ prefix = context.getPrefix(namespace.toString());
+ if (prefix != null) {
+ return prefix;
+ }
+ }
+
do {
prefix = encode(counter);
counter++;
import java.net.URI;
import java.util.Map.Entry;
import javax.annotation.Nonnull;
+import javax.xml.namespace.NamespaceContext;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.data.util.AbstractStringInstanceIdentifierCodec;
import org.opendaylight.yangtools.yang.data.util.DataSchemaContextTree;
import org.opendaylight.yangtools.yang.model.api.SchemaContext;
final class RandomPrefixInstanceIdentifierSerializer extends AbstractStringInstanceIdentifierCodec {
- private final RandomPrefix prefixes = new RandomPrefix();
private final DataSchemaContextTree schemaTree;
+ private final RandomPrefix prefixes;
-
- RandomPrefixInstanceIdentifierSerializer(SchemaContext ctx) {
- schemaTree = DataSchemaContextTree.from(ctx);
+ RandomPrefixInstanceIdentifierSerializer(final SchemaContext schemaContext, final NamespaceContext nsContext) {
+ schemaTree = DataSchemaContextTree.from(schemaContext);
+ prefixes = new RandomPrefix(nsContext);
}
Iterable<Entry<URI, String>> getPrefixes() {
*/
package org.opendaylight.yangtools.yang.data.codec.xml;
-import com.google.common.base.Strings;
import java.io.IOException;
import java.util.Collections;
import java.util.Map;
-import java.util.Map.Entry;
import javax.annotation.Nonnull;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;
return new SchemaAwareXMLStreamNormalizedNodeStreamWriter(writer, context, path);
}
- @Override
- protected void writeAttributes(@Nonnull final Map<QName, String> attributes) throws IOException {
- for (final Entry<QName, String> qNameStringEntry : attributes.entrySet()) {
- try {
- final String namespace = qNameStringEntry.getKey().getNamespace().toString();
-
- if (Strings.isNullOrEmpty(namespace)) {
- writer.writeAttribute(qNameStringEntry.getKey().getLocalName(), qNameStringEntry.getValue());
- } else {
- writer.writeAttribute(namespace, qNameStringEntry.getKey().getLocalName(), qNameStringEntry.getValue());
- }
- } catch (final XMLStreamException e) {
- throw new IOException("Unable to emit attribute " + qNameStringEntry, e);
- }
- }
- }
-
@Override
protected void writeValue(final XMLStreamWriter xmlWriter, final QName qname, @Nonnull final Object value,
final SchemaNode schemaNode) throws IOException, XMLStreamException {
}
@Override
- public void leafNode(NodeIdentifier name, Object value, Map<QName, String> attributes) throws IOException {
+ public void leafNode(final NodeIdentifier name, final Object value, final Map<QName, String> attributes) throws IOException {
final LeafSchemaNode schema = tracker.leafNode(name);
writeElement(schema.getQName(), value, attributes, schema);
}
@Override
void writeInstanceIdentifier(final XMLStreamWriter writer, final YangInstanceIdentifier value)
throws XMLStreamException {
- RandomPrefixInstanceIdentifierSerializer iiCodec = new RandomPrefixInstanceIdentifierSerializer(schemaContext);
+ RandomPrefixInstanceIdentifierSerializer iiCodec = new RandomPrefixInstanceIdentifierSerializer(schemaContext,
+ writer.getNamespaceContext());
String serializedValue = iiCodec.serialize(value);
for (Entry<URI, String> e : iiCodec.getPrefixes()) {
*/
package org.opendaylight.yangtools.yang.data.codec.xml;
-import com.google.common.base.Strings;
import java.io.IOException;
import java.util.ArrayDeque;
import java.util.Collections;
import java.util.Deque;
import java.util.Map;
-import java.util.Map.Entry;
import javax.annotation.Nonnull;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;
}
private final Deque<ContainerType> containerTypeStack = new ArrayDeque<>();
- private final RandomPrefix randomPrefix;
- private SchemalessXMLStreamNormalizedNodeStreamWriter(XMLStreamWriter writer) {
+ private SchemalessXMLStreamNormalizedNodeStreamWriter(final XMLStreamWriter writer) {
super(writer);
- randomPrefix = new RandomPrefix();
}
- static NormalizedNodeStreamWriter newInstance(XMLStreamWriter writer) {
+ static NormalizedNodeStreamWriter newInstance(final XMLStreamWriter writer) {
return new SchemalessXMLStreamNormalizedNodeStreamWriter(writer);
}
@Override
- public void leafNode(NodeIdentifier name, Object value, Map<QName, String> attributes) throws IOException {
+ public void leafNode(final NodeIdentifier name, final Object value, final Map<QName, String> attributes) throws IOException {
writeElement(name.getNodeType(), value, attributes, null);
}
@Override
- public void leafSetEntryNode(QName name, Object value, Map<QName, String> attributes) throws IOException {
+ public void leafSetEntryNode(final QName name, final Object value, final Map<QName, String> attributes) throws IOException {
writeElement(name, value, attributes, null);
}
@Override
- public void leafNode(NodeIdentifier name, Object value) throws IOException {
+ public void leafNode(final NodeIdentifier name, final Object value) throws IOException {
writeElement(name.getNodeType(), value, Collections.emptyMap(), null);
}
@Override
- public void leafSetEntryNode(QName name, Object value) throws IOException {
+ public void leafSetEntryNode(final QName name, final Object value) throws IOException {
writeElement(name, value, Collections.emptyMap(), null);
}
@Override
- public void startLeafSet(NodeIdentifier name, int childSizeHint) throws IOException {
+ public void startLeafSet(final NodeIdentifier name, final int childSizeHint) throws IOException {
containerTypeStack.push(ContainerType.LEAF_SET);
}
@Override
- public void startOrderedLeafSet(NodeIdentifier name, int childSizeHint)
+ public void startOrderedLeafSet(final NodeIdentifier name, final int childSizeHint)
throws IOException, IllegalArgumentException {
containerTypeStack.push(ContainerType.LEAF_SET);
}
@Override
- public void startContainerNode(NodeIdentifier name, int childSizeHint) throws IOException {
+ public void startContainerNode(final NodeIdentifier name, final int childSizeHint) throws IOException {
containerTypeStack.push(ContainerType.CONTAINER);
startElement(name.getNodeType());
}
@Override
- public void startChoiceNode(NodeIdentifier name, int childSizeHint) throws IOException {
+ public void startChoiceNode(final NodeIdentifier name, final int childSizeHint) throws IOException {
containerTypeStack.push(ContainerType.CHOICE);
}
@Override
- public void startAugmentationNode(AugmentationIdentifier identifier) throws IOException {
+ public void startAugmentationNode(final AugmentationIdentifier identifier) throws IOException {
containerTypeStack.push(ContainerType.AUGMENTATION);
}
@Override
- public void anyxmlNode(NodeIdentifier name, Object value) throws IOException {
+ public void anyxmlNode(final NodeIdentifier name, final Object value) throws IOException {
anyxmlNode(name.getNodeType(), value);
}
@Override
- public void startYangModeledAnyXmlNode(NodeIdentifier name, int childSizeHint) throws IOException {
+ public void startYangModeledAnyXmlNode(final NodeIdentifier name, final int childSizeHint) throws IOException {
containerTypeStack.push(ContainerType.ANY_XML);
startElement(name.getNodeType());
}
@Override
- protected void writeAttributes(@Nonnull final Map<QName, String> attributes) throws IOException {
- for (final Entry<QName, String> qNameStringEntry : attributes.entrySet()) {
- try {
- final String namespace = qNameStringEntry.getKey().getNamespace().toString();
-
- if (Strings.isNullOrEmpty(namespace)) {
- writer.writeAttribute(qNameStringEntry.getKey().getLocalName(), qNameStringEntry.getValue());
- } else {
- final String prefix = randomPrefix.encodePrefix(qNameStringEntry.getKey().getNamespace());
- writer.writeAttribute(prefix, namespace, qNameStringEntry.getKey().getLocalName(), qNameStringEntry
- .getValue());
- }
- } catch (final XMLStreamException e) {
- throw new IOException("Unable to emit attribute " + qNameStringEntry, e);
- }
- }
- }
-
- @Override
- protected void writeValue(XMLStreamWriter xmlWriter, QName qname, @Nonnull Object value, Object context)
+ protected void writeValue(final XMLStreamWriter xmlWriter, final QName qname, @Nonnull final Object value, final Object context)
throws XMLStreamException {
xmlWriter.writeCharacters(value.toString());
}
@Override
- protected void startList(NodeIdentifier name) {
+ protected void startList(final NodeIdentifier name) {
containerTypeStack.push(ContainerType.LIST);
}
@Override
- protected void startListItem(PathArgument name) throws IOException {
+ protected void startListItem(final PathArgument name) throws IOException {
containerTypeStack.push(ContainerType.LIST_ITEM);
startElement(name.getNodeType());
}
@Override
- protected void endNode(XMLStreamWriter xmlWriter) throws IOException, XMLStreamException {
+ protected void endNode(final XMLStreamWriter xmlWriter) throws IOException, XMLStreamException {
ContainerType type = containerTypeStack.pop();
switch (type) {
case CONTAINER:
package org.opendaylight.yangtools.yang.data.codec.xml;
import com.google.common.base.Preconditions;
+import com.google.common.base.Strings;
import java.io.IOException;
import java.io.StringWriter;
+import java.net.URI;
import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import javax.xml.XMLConstants;
TRANSFORMER_FACTORY = f;
}
+ private static final Set<String> BROKEN_NAMESPACES = ConcurrentHashMap.newKeySet();
+
+ private final RandomPrefix prefixes;
final XMLStreamWriter writer;
XMLStreamNormalizedNodeStreamWriter(final XMLStreamWriter writer) {
this.writer = Preconditions.checkNotNull(writer);
+ this.prefixes = new RandomPrefix(writer.getNamespaceContext());
}
/**
* @return A new {@link NormalizedNodeStreamWriter}
*/
public static NormalizedNodeStreamWriter create(final XMLStreamWriter writer, final SchemaContext context) {
- return create( writer, context, SchemaPath.ROOT);
+ return create(writer, context, SchemaPath.ROOT);
}
/**
return SchemalessXMLStreamNormalizedNodeStreamWriter.newInstance(writer);
}
- abstract void writeAttributes(@Nonnull final Map<QName, String> attributes) throws IOException;
-
abstract void writeValue(final XMLStreamWriter xmlWriter, final QName qname,
@Nonnull final Object value, T context) throws IOException, XMLStreamException;
abstract void endNode(XMLStreamWriter xmlWriter) throws IOException, XMLStreamException;
+ private void writeAttributes(@Nonnull final Map<QName, String> attributes) throws IOException {
+ for (final Entry<QName, String> qNameStringEntry : attributes.entrySet()) {
+ try {
+ final QName qname = qNameStringEntry.getKey();
+ final String namespace = qname.getNamespace().toString();
+
+ if (!Strings.isNullOrEmpty(namespace)) {
+ final String prefix = getPrefix(qname.getNamespace(), namespace);
+ writer.writeAttribute(prefix, namespace, qname.getLocalName(), qNameStringEntry.getValue());
+ } else {
+ writer.writeAttribute(qname.getLocalName(), qNameStringEntry.getValue());
+ }
+ } catch (final XMLStreamException e) {
+ throw new IOException("Unable to emit attribute " + qNameStringEntry, e);
+ }
+ }
+ }
+
+ private String getPrefix(final URI uri, final String str) throws XMLStreamException {
+ final String prefix = writer.getPrefix(str);
+ if (prefix != null) {
+ return prefix;
+ }
+
+ // This is needed to recover from attributes emitted while the namespace was not declared. Ordinarily
+ // attribute namespaces would be bound in the writer, so the resulting XML is efficient, but we cannot rely
+ // on that having been done.
+ if (BROKEN_NAMESPACES.add(str)) {
+ LOG.info("Namespace {} was not bound, please fix the caller", str, new Throwable());
+ }
+
+ return prefixes.encodePrefix(uri);
+ }
+
private void writeStartElement(final QName qname) throws XMLStreamException {
String ns = qname.getNamespace().toString();
writer.writeStartElement(XMLConstants.DEFAULT_NS_PREFIX, qname.getLocalName(), ns);
import static org.junit.Assert.assertThat;
import com.google.common.collect.Iterables;
-import com.google.common.collect.Lists;
-import com.google.common.collect.Sets;
import java.net.URI;
+import java.util.ArrayList;
import java.util.Date;
+import java.util.HashSet;
import java.util.List;
import org.hamcrest.CoreMatchers;
import org.junit.Test;
@Test
public void testEncodeDecode() throws Exception {
- final List<String> allGenerated = Lists.newArrayList();
+ final List<String> allGenerated = new ArrayList<>(MAX_COUNTER);
for (int i = 0; i < MAX_COUNTER; i++) {
final String encoded = RandomPrefix.encode(i);
assertEquals(RandomPrefix.decode(encoded), i);
assertEquals(allGenerated.size(), MAX_COUNTER);
assertEquals("dPT", allGenerated.get(MAX_COUNTER - 1));
assertEquals("a", allGenerated.get(0));
- assertEquals(allGenerated.size(), Sets.newHashSet(allGenerated).size());
+ assertEquals(allGenerated.size(), new HashSet<>(allGenerated).size());
}
@Test
public void testQNameWithPrefix() throws Exception {
- final RandomPrefix a = new RandomPrefix();
+ final RandomPrefix a = new RandomPrefix(null);
- final List<String> allGenerated = Lists.newArrayList();
+ final List<String> allGenerated = new ArrayList<>();
for (int i = 0; i < MAX_COUNTER; i++) {
final String prefix = RandomPrefix.encode(i);
final URI uri = new URI("localhost:" + prefix);
@Test
public void test2QNames1Namespace() throws Exception {
- final RandomPrefix a = new RandomPrefix();
+ final RandomPrefix a = new RandomPrefix(null);
final URI uri = URI.create("localhost");
final QName qName = QName.create(QNameModule.create(uri, new Date()), "local-name");
@Test
public void testQNameNoPrefix() throws Exception {
- final RandomPrefix a = new RandomPrefix();
+ final RandomPrefix a = new RandomPrefix(null);
final URI uri = URI.create("localhost");
QName qName = QName.create(uri, new Date(), "local-name");
name = getAttrQName("namespace2", "2012-12-12", "attr", Optional.absent());
final Map.Entry<QName, String> attributeEntryNoPrefix = new AbstractMap.SimpleEntry<>(name, "value");
- final RandomPrefix randomPrefix = new RandomPrefix();
+ final RandomPrefix randomPrefix = new RandomPrefix(null);
XMLStreamWriterUtils.writeAttribute(writer, attributeEntry, randomPrefix);
XMLStreamWriterUtils.writeAttribute(writer, attributeEntryNoPrefix, randomPrefix);