Bug 7521: Convert Snapshot to store a State instance

[controller.git] / opendaylight / md-sal / sal-akka-raft / src / main / java / org / opendaylight / controller / cluster / raft / base / messages / SendInstallSnapshot.java

diff --git a/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/base/messages/SendInstallSnapshot.java b/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/base/messages/SendInstallSnapshot.java
index de33b8c95b1ad0679d6a2af3c37e880dca099efa..e8d58f253afe09a27cc79c07e306c258e1b35929 100644 (file)
--- a/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/base/messages/SendInstallSnapshot.java
+++ b/opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/base/messages/SendInstallSnapshot.java
@@ -9,8 +9,9 @@
 package org.opendaylight.controller.cluster.raft.base.messages;
 
 import com.google.common.base.Preconditions;
+import com.google.common.io.ByteSource;
 import javax.annotation.Nonnull;
-import org.opendaylight.controller.cluster.raft.Snapshot;
+import org.opendaylight.controller.cluster.raft.persisted.Snapshot;
 
 /**
  * Internal message sent from the SnapshotManager to its associated leader when a snapshot capture is complete to
@@ -18,13 +19,19 @@ import org.opendaylight.controller.cluster.raft.Snapshot;
  */
 public final class SendInstallSnapshot {
     private final Snapshot snapshot;
+    private final ByteSource snapshotBytes;
 
-    public SendInstallSnapshot(@Nonnull Snapshot snapshot) {
+    public SendInstallSnapshot(@Nonnull Snapshot snapshot, @Nonnull ByteSource snapshotBytes) {
         this.snapshot = Preconditions.checkNotNull(snapshot);
+        this.snapshotBytes = Preconditions.checkNotNull(snapshotBytes);
     }
 
     @Nonnull
     public Snapshot getSnapshot() {
         return snapshot;
     }
+
+    public ByteSource getSnapshotBytes() {
+        return snapshotBytes;
+    }
 }