From 994333e720c88a27ad3eee69e028dc1fe70db574 Mon Sep 17 00:00:00 2001
From: Xiang Li <xiangli.cs@gmail.com>
Date: Tue, 16 Feb 2016 16:05:20 -0800
Subject: [PATCH] *: record the number of bytes of snapshot sent/received

---
 etcdserver/snapshot_merge.go | 5 ++++-
 snap/db.go                   | 6 +++++-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/etcdserver/snapshot_merge.go b/etcdserver/snapshot_merge.go
index c09a4051a..51599bcf5 100644
--- a/etcdserver/snapshot_merge.go
+++ b/etcdserver/snapshot_merge.go
@@ -60,7 +60,10 @@ func (s *EtcdServer) createMergedSnapshotMessage(m raftpb.Message, snapi uint64,
 func newSnapshotReaderCloser(snapshot backend.Snapshot) io.ReadCloser {
 	pr, pw := io.Pipe()
 	go func() {
-		_, err := snapshot.WriteTo(pw)
+		n, err := snapshot.WriteTo(pw)
+		if err == nil {
+			plog.Infof("wrote database snapshot out [total bytes: %d]", n)
+		}
 		pw.CloseWithError(err)
 		snapshot.Close()
 	}()
diff --git a/snap/db.go b/snap/db.go
index 0d825714a..ca68837cb 100644
--- a/snap/db.go
+++ b/snap/db.go
@@ -31,7 +31,8 @@ func (s *Snapshotter) SaveDBFrom(r io.Reader, id uint64) error {
 	if err != nil {
 		return err
 	}
-	_, err = io.Copy(f, r)
+	var n int64
+	n, err = io.Copy(f, r)
 	if err == nil {
 		err = f.Sync()
 	}
@@ -50,6 +51,9 @@ func (s *Snapshotter) SaveDBFrom(r io.Reader, id uint64) error {
 		os.Remove(f.Name())
 		return err
 	}
+
+	plog.Infof("saved database snapshot to disk [total bytes: %d]", n)
+
 	return nil
 }