* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
-
package org.opendaylight.controller.cluster.datastore.node.utils.stream;
-import com.google.common.base.Preconditions;
+import static java.util.Objects.requireNonNull;
+
import com.google.common.base.Strings;
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableList.Builder;
import java.io.DataInput;
import java.io.IOException;
import java.io.StringReader;
import java.math.BigInteger;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
-import java.util.Collection;
-import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
-import java.util.Map;
import java.util.Set;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.dom.DOMSource;
+import org.eclipse.jdt.annotation.NonNull;
import org.opendaylight.controller.cluster.datastore.node.utils.QNameFactory;
+import org.opendaylight.yangtools.util.ImmutableOffsetMapTemplate;
import org.opendaylight.yangtools.yang.common.Empty;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+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.NodeWithValue;
import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.ListNodeBuilder;
-import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.NormalizedNodeAttrBuilder;
+import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.NormalizedNodeBuilder;
import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.NormalizedNodeContainerBuilder;
import org.opendaylight.yangtools.yang.model.api.SchemaPath;
import org.slf4j.Logger;
* nodes. This process goes in recursive manner, where each NodeTypes object signifies the start of the object, except
* END_NODE. If a node can have children, then that node's end is calculated based on appearance of END_NODE.
*/
-public class NormalizedNodeInputStreamReader implements NormalizedNodeDataInput {
+public class NormalizedNodeInputStreamReader extends ForwardingDataInput implements NormalizedNodeDataInput {
private static final Logger LOG = LoggerFactory.getLogger(NormalizedNodeInputStreamReader.class);
- private static final String REVISION_ARG = "?revision=";
-
- private final DataInput input;
+ private final @NonNull DataInput input;
- private final Map<Integer, String> codedStringMap = new HashMap<>();
+ private final List<String> codedStringMap = new ArrayList<>();
private QName lastLeafSetQName;
- private NormalizedNodeAttrBuilder<YangInstanceIdentifier.NodeIdentifier,
- Object, LeafNode<Object>> leafBuilder;
+ private NormalizedNodeBuilder<NodeIdentifier, Object, LeafNode<Object>> leafBuilder;
@SuppressWarnings("rawtypes")
- private NormalizedNodeAttrBuilder<NodeWithValue, Object, LeafSetEntryNode<Object>> leafSetEntryBuilder;
+ private NormalizedNodeBuilder<NodeWithValue, Object, LeafSetEntryNode<Object>> leafSetEntryBuilder;
- private final StringBuilder reusableStringBuilder = new StringBuilder(50);
-
- private boolean readSignatureMarker = true;
+ NormalizedNodeInputStreamReader(final DataInput input) {
+ this.input = requireNonNull(input);
+ }
- NormalizedNodeInputStreamReader(final DataInput input, final boolean versionChecked) {
- this.input = Preconditions.checkNotNull(input);
- readSignatureMarker = !versionChecked;
+ @Override
+ final DataInput delegate() {
+ return input;
}
@Override
public NormalizedNode<?, ?> readNormalizedNode() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
return readNormalizedNodeInternal();
}
- private void readSignatureMarkerAndVersionIfNeeded() throws IOException {
- if (readSignatureMarker) {
- readSignatureMarker = false;
-
- final byte marker = input.readByte();
- if (marker != TokenTypes.SIGNATURE_MARKER) {
- throw new InvalidNormalizedNodeStreamException(String.format(
- "Invalid signature marker: %d", marker));
- }
-
- final short version = input.readShort();
- if (version != TokenTypes.LITHIUM_VERSION) {
- throw new InvalidNormalizedNodeStreamException(String.format("Unhandled stream version %s", version));
- }
- }
- }
-
private NormalizedNode<?, ?> readNormalizedNodeInternal() throws IOException {
// each node should start with a byte
byte nodeType = input.readByte();
}
switch (nodeType) {
- case NodeTypes.AUGMENTATION_NODE :
- YangInstanceIdentifier.AugmentationIdentifier augIdentifier =
- new YangInstanceIdentifier.AugmentationIdentifier(readQNameSet());
-
+ case NodeTypes.AUGMENTATION_NODE:
+ AugmentationIdentifier augIdentifier = readAugmentationIdentifier();
LOG.trace("Reading augmentation node {} ", augIdentifier);
+ return addDataContainerChildren(Builders.augmentationBuilder().withNodeIdentifier(augIdentifier))
+ .build();
- return addDataContainerChildren(Builders.augmentationBuilder()
- .withNodeIdentifier(augIdentifier)).build();
-
- case NodeTypes.LEAF_SET_ENTRY_NODE :
- QName name = lastLeafSetQName;
- if (name == null) {
- name = readQName();
- }
-
- Object value = readObject();
- NodeWithValue<Object> leafIdentifier = new NodeWithValue<>(name, value);
-
+ case NodeTypes.LEAF_SET_ENTRY_NODE:
+ final QName name = lastLeafSetQName != null ? lastLeafSetQName : readQName();
+ final Object value = readObject();
+ final NodeWithValue<Object> leafIdentifier = new NodeWithValue<>(name, value);
LOG.trace("Reading leaf set entry node {}, value {}", leafIdentifier, value);
-
return leafSetEntryBuilder().withNodeIdentifier(leafIdentifier).withValue(value).build();
- case NodeTypes.MAP_ENTRY_NODE :
- NodeIdentifierWithPredicates entryIdentifier = new NodeIdentifierWithPredicates(
- readQName(), readKeyValueMap());
-
+ case NodeTypes.MAP_ENTRY_NODE:
+ final NodeIdentifierWithPredicates entryIdentifier = readNormalizedNodeWithPredicates();
LOG.trace("Reading map entry node {} ", entryIdentifier);
+ return addDataContainerChildren(Builders.mapEntryBuilder().withNodeIdentifier(entryIdentifier))
+ .build();
- return addDataContainerChildren(Builders.mapEntryBuilder()
- .withNodeIdentifier(entryIdentifier)).build();
-
- default :
- return readNodeIdentifierDependentNode(nodeType, new NodeIdentifier(readQName()));
+ default:
+ return readNodeIdentifierDependentNode(nodeType, readNodeIdentifier());
}
}
- private NormalizedNodeAttrBuilder<YangInstanceIdentifier.NodeIdentifier,
- Object, LeafNode<Object>> leafBuilder() {
+ private NormalizedNodeBuilder<NodeIdentifier, Object, LeafNode<Object>> leafBuilder() {
if (leafBuilder == null) {
leafBuilder = Builders.leafBuilder();
}
}
@SuppressWarnings("rawtypes")
- private NormalizedNodeAttrBuilder<NodeWithValue, Object,
- LeafSetEntryNode<Object>> leafSetEntryBuilder() {
+ private NormalizedNodeBuilder<NodeWithValue, Object, LeafSetEntryNode<Object>> leafSetEntryBuilder() {
if (leafSetEntryBuilder == null) {
leafSetEntryBuilder = Builders.leafSetEntryBuilder();
}
throws IOException {
switch (nodeType) {
- case NodeTypes.LEAF_NODE :
+ case NodeTypes.LEAF_NODE:
LOG.trace("Read leaf node {}", identifier);
// Read the object value
return leafBuilder().withNodeIdentifier(identifier).withValue(readObject()).build();
- case NodeTypes.ANY_XML_NODE :
+ case NodeTypes.ANY_XML_NODE:
LOG.trace("Read xml node");
return Builders.anyXmlBuilder().withNodeIdentifier(identifier).withValue(readDOMSource()).build();
- case NodeTypes.MAP_NODE :
+ case NodeTypes.MAP_NODE:
LOG.trace("Read map node {}", identifier);
return addDataContainerChildren(Builders.mapBuilder().withNodeIdentifier(identifier)).build();
LOG.trace("Read container node {}", identifier);
return addDataContainerChildren(Builders.containerBuilder().withNodeIdentifier(identifier)).build();
- case NodeTypes.LEAF_SET :
+ case NodeTypes.LEAF_SET:
LOG.trace("Read leaf set node {}", identifier);
return addLeafSetChildren(identifier.getNodeType(),
Builders.leafSetBuilder().withNodeIdentifier(identifier)).build();
// Read in the same sequence of writing
String localName = readCodedString();
String namespace = readCodedString();
- String revision = readCodedString();
-
- String qname;
- if (!Strings.isNullOrEmpty(revision)) {
- qname = reusableStringBuilder.append('(').append(namespace).append(REVISION_ARG).append(revision)
- .append(')').append(localName).toString();
- } else {
- qname = reusableStringBuilder.append('(').append(namespace).append(')').append(localName).toString();
- }
+ String revision = Strings.emptyToNull(readCodedString());
- reusableStringBuilder.delete(0, reusableStringBuilder.length());
- return QNameFactory.create(qname);
+ return QNameFactory.create(new QNameFactory.Key(localName, namespace, revision));
}
private String readCodedString() throws IOException {
- byte valueType = input.readByte();
- if (valueType == TokenTypes.IS_CODE_VALUE) {
- return codedStringMap.get(input.readInt());
- } else if (valueType == TokenTypes.IS_STRING_VALUE) {
- String value = input.readUTF().intern();
- codedStringMap.put(codedStringMap.size(), value);
- return value;
+ final byte valueType = input.readByte();
+ switch (valueType) {
+ case TokenTypes.IS_NULL_VALUE:
+ return null;
+ case TokenTypes.IS_CODE_VALUE:
+ final int code = input.readInt();
+ try {
+ return codedStringMap.get(code);
+ } catch (IndexOutOfBoundsException e) {
+ throw new IOException("String code " + code + " was not found", e);
+ }
+ case TokenTypes.IS_STRING_VALUE:
+ final String value = input.readUTF().intern();
+ codedStringMap.add(value);
+ return value;
+ default:
+ throw new IOException("Unhandled string value type " + valueType);
}
-
- return null;
}
private Set<QName> readQNameSet() throws IOException {
return children;
}
- private Map<QName, Object> readKeyValueMap() throws IOException {
- int count = input.readInt();
- Map<QName, Object> keyValueMap = new HashMap<>(count);
+ private AugmentationIdentifier readAugmentationIdentifier() throws IOException {
+ // FIXME: we should have a cache for these, too
+ return new AugmentationIdentifier(readQNameSet());
+ }
- for (int i = 0; i < count; i++) {
- keyValueMap.put(readQName(), readObject());
- }
+ private NodeIdentifier readNodeIdentifier() throws IOException {
+ // FIXME: we should have a cache for these, too
+ return new NodeIdentifier(readQName());
+ }
+
+ private NodeIdentifierWithPredicates readNormalizedNodeWithPredicates() throws IOException {
+ final QName qname = readQName();
+ final int count = input.readInt();
+ switch (count) {
+ case 0:
+ return new NodeIdentifierWithPredicates(qname);
+ case 1:
+ return new NodeIdentifierWithPredicates(qname, readQName(), readObject());
+ default:
+ // ImmutableList is used by ImmutableOffsetMapTemplate for lookups, hence we use that.
+ final Builder<QName> keys = ImmutableList.builderWithExpectedSize(count);
+ final Object[] values = new Object[count];
+ for (int i = 0; i < count; i++) {
+ keys.add(readQName());
+ values[i] = readObject();
+ }
- return keyValueMap;
+ return new NodeIdentifierWithPredicates(qname, ImmutableOffsetMapTemplate.ordered(keys.build())
+ .instantiateWithValues(values));
+ }
}
private Object readObject() throws IOException {
case ValueTypes.BITS_TYPE:
return readObjSet();
- case ValueTypes.BOOL_TYPE :
+ case ValueTypes.BOOL_TYPE:
return input.readBoolean();
- case ValueTypes.BYTE_TYPE :
+ case ValueTypes.BYTE_TYPE:
return input.readByte();
- case ValueTypes.INT_TYPE :
+ case ValueTypes.INT_TYPE:
return input.readInt();
- case ValueTypes.LONG_TYPE :
+ case ValueTypes.LONG_TYPE:
return input.readLong();
- case ValueTypes.QNAME_TYPE :
+ case ValueTypes.QNAME_TYPE:
return readQName();
- case ValueTypes.SHORT_TYPE :
+ case ValueTypes.SHORT_TYPE:
return input.readShort();
- case ValueTypes.STRING_TYPE :
+ case ValueTypes.STRING_TYPE:
return input.readUTF();
case ValueTypes.STRING_BYTES_TYPE:
return readStringBytes();
- case ValueTypes.BIG_DECIMAL_TYPE :
+ case ValueTypes.BIG_DECIMAL_TYPE:
return new BigDecimal(input.readUTF());
- case ValueTypes.BIG_INTEGER_TYPE :
+ case ValueTypes.BIG_INTEGER_TYPE:
return new BigInteger(input.readUTF());
- case ValueTypes.BINARY_TYPE :
+ case ValueTypes.BINARY_TYPE:
byte[] bytes = new byte[input.readInt()];
input.readFully(bytes);
return bytes;
- case ValueTypes.YANG_IDENTIFIER_TYPE :
+ case ValueTypes.YANG_IDENTIFIER_TYPE:
return readYangInstanceIdentifierInternal();
case ValueTypes.EMPTY_TYPE:
case ValueTypes.NULL_TYPE:
return Empty.getInstance();
- default :
+ default:
return null;
}
}
@Override
public SchemaPath readSchemaPath() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
-
final boolean absolute = input.readBoolean();
final int size = input.readInt();
- final Collection<QName> qnames = new ArrayList<>(size);
+
+ final Builder<QName> qnames = ImmutableList.builderWithExpectedSize(size);
for (int i = 0; i < size; ++i) {
qnames.add(readQName());
}
-
- return SchemaPath.create(qnames, absolute);
+ return SchemaPath.create(qnames.build(), absolute);
}
@Override
public YangInstanceIdentifier readYangInstanceIdentifier() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
return readYangInstanceIdentifierInternal();
}
private YangInstanceIdentifier readYangInstanceIdentifierInternal() throws IOException {
int size = input.readInt();
-
- List<PathArgument> pathArguments = new ArrayList<>(size);
-
+ final Builder<PathArgument> pathArguments = ImmutableList.builderWithExpectedSize(size);
for (int i = 0; i < size; i++) {
pathArguments.add(readPathArgument());
}
- return YangInstanceIdentifier.create(pathArguments);
+ return YangInstanceIdentifier.create(pathArguments.build());
}
private Set<String> readObjSet() throws IOException {
int type = input.readByte();
switch (type) {
-
- case PathArgumentTypes.AUGMENTATION_IDENTIFIER :
- return new YangInstanceIdentifier.AugmentationIdentifier(readQNameSet());
-
- case PathArgumentTypes.NODE_IDENTIFIER :
- return new NodeIdentifier(readQName());
-
- case PathArgumentTypes.NODE_IDENTIFIER_WITH_PREDICATES :
- return new NodeIdentifierWithPredicates(readQName(), readKeyValueMap());
-
- case PathArgumentTypes.NODE_IDENTIFIER_WITH_VALUE :
+ case PathArgumentTypes.AUGMENTATION_IDENTIFIER:
+ return readAugmentationIdentifier();
+ case PathArgumentTypes.NODE_IDENTIFIER:
+ return readNodeIdentifier();
+ case PathArgumentTypes.NODE_IDENTIFIER_WITH_PREDICATES:
+ return readNormalizedNodeWithPredicates();
+ case PathArgumentTypes.NODE_IDENTIFIER_WITH_VALUE:
return new NodeWithValue<>(readQName(), readObject());
-
- default :
+ default:
+ // FIXME: throw hard error
return null;
}
}
}
return builder;
}
-
- @Override
- public void readFully(final byte[] value) throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- input.readFully(value);
- }
-
- @Override
- public void readFully(final byte[] str, final int off, final int len) throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- input.readFully(str, off, len);
- }
-
- @Override
- public int skipBytes(final int num) throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- return input.skipBytes(num);
- }
-
- @Override
- public boolean readBoolean() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- return input.readBoolean();
- }
-
- @Override
- public byte readByte() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- return input.readByte();
- }
-
- @Override
- public int readUnsignedByte() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- return input.readUnsignedByte();
- }
-
- @Override
- public short readShort() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- return input.readShort();
- }
-
- @Override
- public int readUnsignedShort() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- return input.readUnsignedShort();
- }
-
- @Override
- public char readChar() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- return input.readChar();
- }
-
- @Override
- public int readInt() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- return input.readInt();
- }
-
- @Override
- public long readLong() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- return input.readLong();
- }
-
- @Override
- public float readFloat() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- return input.readFloat();
- }
-
- @Override
- public double readDouble() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- return input.readDouble();
- }
-
- @Override
- public String readLine() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- return input.readLine();
- }
-
- @Override
- public String readUTF() throws IOException {
- readSignatureMarkerAndVersionIfNeeded();
- return input.readUTF();
- }
}
Code Review / controller.git / blobdiff