BUG-4626: create AbstractNormalizedNodeDataOutput

[controller.git] / opendaylight / md-sal / sal-clustering-commons / src / main / java / org / opendaylight / controller / cluster / datastore / node / utils / stream / NormalizedNodeInputOutput.java

diff --git a/opendaylight/md-sal/sal-clustering-commons/src/main/java/org/opendaylight/controller/cluster/datastore/node/utils/stream/NormalizedNodeInputOutput.java b/opendaylight/md-sal/sal-clustering-commons/src/main/java/org/opendaylight/controller/cluster/datastore/node/utils/stream/NormalizedNodeInputOutput.java
index cb84ef881095725fa8a7ec0d9e3f7f79380c21a4..5b93b274e0b757766e750efa8c5f8a129e9f6eed 100644 (file)
--- a/opendaylight/md-sal/sal-clustering-commons/src/main/java/org/opendaylight/controller/cluster/datastore/node/utils/stream/NormalizedNodeInputOutput.java
+++ b/opendaylight/md-sal/sal-clustering-commons/src/main/java/org/opendaylight/controller/cluster/datastore/node/utils/stream/NormalizedNodeInputOutput.java
@@ -7,18 +7,31 @@
  */
 package org.opendaylight.controller.cluster.datastore.node.utils.stream;
 
+import com.google.common.annotations.Beta;
 import java.io.DataInput;
 import java.io.DataOutput;
 import java.io.IOException;
 import javax.annotation.Nonnull;
 
+@Beta
 public final class NormalizedNodeInputOutput {
     private NormalizedNodeInputOutput() {
         throw new UnsupportedOperationException();
     }
 
     public static NormalizedNodeDataInput newDataInput(@Nonnull final DataInput input) throws IOException {
-        return new NormalizedNodeInputStreamReader(input);
+        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();
+        switch (version) {
+            case TokenTypes.LITHIUM_VERSION:
+                return new NormalizedNodeInputStreamReader(input, true);
+            default:
+                throw new InvalidNormalizedNodeStreamException(String.format("Unhandled stream version %s", version));
+        }
     }
 
     public static NormalizedNodeDataOutput newDataOutput(@Nonnull final DataOutput output) throws IOException {