From 64c29d4b6e6b774436927ef1c0356b1ae32b3530 Mon Sep 17 00:00:00 2001
From: Ben Johnson <benbjohnson@yahoo.com>
Date: Thu, 9 Jun 2016 14:25:41 -0600
Subject: [PATCH] add buffered snapshot writer

This commit wraps the snapshot in a `bufio.Writer`.
---
 fragment.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/fragment.go b/fragment.go
index ef9d22c77..38ed99567 100644
--- a/fragment.go
+++ b/fragment.go
@@ -2,6 +2,7 @@ package pilosa
 
 import (
 	"archive/tar"
+	"bufio"
 	"bytes"
 	"crypto/sha1"
 	"encoding/binary"
@@ -906,8 +907,11 @@ func (f *Fragment) snapshot() error {
 	defer file.Close()
 
 	// Write storage to snapshot.
-	if _, err := f.storage.WriteTo(file); err != nil {
+	bw := bufio.NewWriter(file)
+	if _, err := f.storage.WriteTo(bw); err != nil {
 		return fmt.Errorf("snapshot write to: %s", err)
+	} else if err := bw.Flush(); err != nil {
+		return fmt.Errorf("flush: %s", err)
 	}
 
 	// Close current storage.