common: WriteFileAtomic use tempfile in current dir

This commit is contained in:
Ethan Buchman 2017-09-22 13:20:13 -04:00
parent 2130c329eb
commit 3d98504c4c
1 changed files with 3 additions and 1 deletions

View File

@ -7,6 +7,7 @@ import (
"io/ioutil"
"os"
"os/signal"
"path/filepath"
"strings"
)
@ -101,7 +102,8 @@ func MustWriteFile(filePath string, contents []byte, mode os.FileMode) {
// WriteFileAtomic writes newBytes to temp and atomically moves to filePath
// when everything else succeeds.
func WriteFileAtomic(filePath string, newBytes []byte, mode os.FileMode) error {
f, err := ioutil.TempFile("", "")
dir := filepath.Dir(filePath)
f, err := ioutil.TempFile(dir, "")
if err != nil {
return err
}