From 4c311aa1a789a14cf4cc17b75e83283695cc3ba5 Mon Sep 17 00:00:00 2001
From: Travis <travis@pilosa.com>
Date: Wed, 25 Mar 2020 15:37:42 -0500
Subject: [PATCH] write to temp partition file. use io.Copy

---
 boltdb/translate.go | 31 +++++++++++++++++++++++--------
 1 file changed, 23 insertions(+), 8 deletions(-)

diff --git a/boltdb/translate.go b/boltdb/translate.go
index 48462d5f9..cb8088674 100644
--- a/boltdb/translate.go
+++ b/boltdb/translate.go
@@ -18,7 +18,6 @@ import (
 	"bytes"
 	"context"
 	"io"
-	"io/ioutil"
 	"os"
 	"path/filepath"
 	"sync"
@@ -35,6 +34,11 @@ var (
 	ErrTranslateStoreClosed = errors.New("boltdb: translate store closing")
 )
 
+const (
+	// snapshotExt is the file extension used for an in-process snapshot.
+	snapshotExt = ".snapshotting"
+)
+
 // OpenTranslateStore opens and initializes a boltdb translation store.
 func OpenTranslateStore(path, index, field string, partitionID, partitionN int) (pilosa.TranslateStore, error) {
 	s := NewTranslateStore(index, field, partitionID, partitionN)
@@ -356,21 +360,32 @@ func (s *TranslateStore) WriteTo(w io.Writer) (int64, error) {
 }
 
 // ReadFrom reads the content and overwrites the existing store.
-func (s *TranslateStore) ReadFrom(r io.Reader) (int64, error) {
+func (s *TranslateStore) ReadFrom(r io.Reader) (n int64, err error) {
 	// Close store.
 	if err := s.Close(); err != nil {
 		return 0, errors.Wrap(err, "closing store")
 	}
 
-	buf := bytes.NewBuffer(nil)
-	n, err := buf.ReadFrom(r)
+	// Create a temporary file to snapshot to.
+	snapshotPath := s.Path + snapshotExt
+	file, err := os.Create(snapshotPath)
 	if err != nil {
-		return n, errors.Wrap(err, "reading from reader")
+		return n, errors.Wrap(err, "creating snapshot file")
 	}
 
-	// Overwrite the store file.
-	if err := ioutil.WriteFile(s.Path, buf.Bytes(), 0666); err != nil {
-		return n, errors.Wrap(err, "writing file")
+	// Write payload to snapshot.
+	if n, err = io.Copy(file, r); err != nil {
+		file.Close()
+		return n, errors.Wrap(err, "snapshot write to")
+	}
+
+	// we close the file here so we don't still have it open when trying
+	// to open it in a moment.
+	file.Close()
+
+	// Move snapshot to data file location.
+	if err := os.Rename(snapshotPath, s.Path); err != nil {
+		return n, errors.Wrap(err, "renaming snapshot")
 	}
 
 	// Re-open the store.