*/
package org.opendaylight.controller.cluster.datastore.persisted;
-import com.google.common.base.Verify;
-import java.io.ByteArrayOutputStream;
-import java.io.DataInputStream;
-import java.io.DataOutputStream;
+import static com.google.common.base.Verify.verifyNotNull;
+
import java.io.IOException;
-import java.io.ObjectInputStream;
-import java.io.ObjectOutputStream;
+import java.io.ObjectInput;
+import java.io.ObjectOutput;
import java.util.Optional;
-import javax.annotation.Nonnull;
+import org.eclipse.jdt.annotation.NonNull;
import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
abstract class AbstractVersionedShardDataTreeSnapshot extends ShardDataTreeSnapshot {
private static final Logger LOG = LoggerFactory.getLogger(AbstractVersionedShardDataTreeSnapshot.class);
- static ShardDataTreeSnapshot deserialize(final DataInputStream is) throws IOException {
- final PayloadVersion version = PayloadVersion.readFrom(is);
+ @SuppressWarnings("checkstyle:FallThrough")
+ static @NonNull ShardSnapshotState versionedDeserialize(final ObjectInput in) throws IOException {
+ final PayloadVersion version = PayloadVersion.readFrom(in);
switch (version) {
- case BORON:
- // Boron snapshots use Java Serialization
- try (final ObjectInputStream ois = new ObjectInputStream(is)) {
- return (ShardDataTreeSnapshot) ois.readObject();
- } catch (ClassNotFoundException e) {
- LOG.error("Failed to serialize data tree snapshot", e);
- throw new IOException("Snapshot failed to deserialize", e);
- }
+ case SODIUM_SR1:
+ return new ShardSnapshotState(readSnapshot(in), true);
+ case MAGNESIUM:
+ return new ShardSnapshotState(readSnapshot(in), false);
case TEST_FUTURE_VERSION:
case TEST_PAST_VERSION:
// These versions are never returned and this code is effectively dead
- break;
+ default:
+ // Not included as default in above switch to ensure we get warnings when new versions are added
+ throw new IOException("Encountered unhandled version " + version);
}
+ }
- // Not included as default in above switch to ensure we get warnings when new versions are added
- throw new IOException("Encountered unhandled version" + version);
+ // Boron and Sodium snapshots use Java Serialization, but differ in stream format
+ private static @NonNull ShardDataTreeSnapshot readSnapshot(final ObjectInput in) throws IOException {
+ try {
+ return (ShardDataTreeSnapshot) in.readObject();
+ } catch (ClassNotFoundException e) {
+ LOG.error("Failed to serialize data tree snapshot", e);
+ throw new IOException("Snapshot failed to deserialize", e);
+ }
}
@Override
- public final Optional<NormalizedNode<?, ?>> getRootNode() {
- return Optional.of(Verify.verifyNotNull(rootNode(), "Snapshot %s returned non-present root node", getClass()));
+ public final Optional<NormalizedNode> getRootNode() {
+ return Optional.of(verifyNotNull(rootNode(), "Snapshot %s returned non-present root node", getClass()));
}
/**
*
* @return The root node.
*/
- abstract @Nonnull NormalizedNode<?, ?> rootNode();
+ abstract @NonNull NormalizedNode rootNode();
/**
* Return the snapshot payload version. Implementations of this method should return a constant.
*
* @return Snapshot payload version
*/
- abstract @Nonnull PayloadVersion version();
+ abstract @NonNull PayloadVersion version();
- private void versionedSerialize(final DataOutputStream dos, final PayloadVersion version) throws IOException {
+ private void versionedSerialize(final ObjectOutput out, final PayloadVersion version) throws IOException {
switch (version) {
- case BORON:
- // Boron snapshots use Java Serialization
- try (ObjectOutputStream oos = new ObjectOutputStream(dos)) {
- oos.writeObject(this);
- }
+ case SODIUM_SR1:
+ case MAGNESIUM:
+ // Sodium and Magnesium snapshots use Java Serialization, but differ in stream format
+ out.writeObject(this);
return;
case TEST_FUTURE_VERSION:
case TEST_PAST_VERSION:
break;
-
+ default:
+ throw new IOException("Invalid payload version in snapshot");
}
throw new IOException("Encountered unhandled version" + version);
}
@Override
- public final byte[] serialize() throws IOException {
- try (final ByteArrayOutputStream bos = new ByteArrayOutputStream()) {
- try (final DataOutputStream dos = new DataOutputStream(bos)) {
- final PayloadVersion version = version();
- version.writeTo(dos);
- versionedSerialize(dos, version);
- }
-
- return bos.toByteArray();
- }
+ public void serialize(final ObjectOutput out) throws IOException {
+ final PayloadVersion version = version();
+ version.writeTo(out);
+ versionedSerialize(out, version);
}
}