KurlesHS/restic
1
0
Fork 0
forked from TrueCloudLab/restic
Code Pull requests Activity

local: split out tempfile write function

Browse source
This commit is contained in:
Alexander Neumann 2016-01-26 22:12:53 +01:00
parent 9ec435d863
commit 0bbad683c5
1 changed files with 38 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

70
backend/local/local.go
View file

@ -126,67 +126,73 @@ func (b *Local) Load(h backend.Handle, p []byte, off int64) (n int, err error) {
return io.ReadFull(f, p) return io.ReadFull(f, p)
} }
// writeToTempfile saves p into a tempfile in tempdir.
func writeToTempfile(tempdir string, p []byte) (filename string, err error) {
tmpfile, err := ioutil.TempFile(tempdir, "temp-")
if err != nil {
return "", err
}
n, err := tmpfile.Write(p)
if err != nil {
return "", err
}
if n != len(p) {
return "", errors.New("not all bytes writen")
}
if err = tmpfile.Sync(); err != nil {
return "", err
}
err = tmpfile.Close()
if err != nil {
return "", err
}
return tmpfile.Name(), nil
}
// Save stores data in the backend at the handle. // Save stores data in the backend at the handle.
func (b *Local) Save(h backend.Handle, p []byte) (err error) { func (b *Local) Save(h backend.Handle, p []byte) (err error) {
if err := h.Valid(); err != nil { if err := h.Valid(); err != nil {
return err return err
} }
tmpfile, err := ioutil.TempFile(filepath.Join(b.p, backend.Paths.Temp), "temp-") tmpfile, err := writeToTempfile(filepath.Join(b.p, backend.Paths.Temp), p)
if err != nil { debug.Log("local.Save", "saved %v (%d bytes) to %v", h, len(p), tmpfile)
return err
}
debug.Log("local.Save", "save %v (%d bytes) to %v", h, len(p), tmpfile.Name()) filename := filename(b.p, h.Type, h.Name)
n, err := tmpfile.Write(p)
if err != nil {
return err
}
if n != len(p) {
return errors.New("not all bytes writen")
}
if err = tmpfile.Sync(); err != nil {
return err
}
err = tmpfile.Close()
if err != nil {
return err
}
f := filename(b.p, h.Type, h.Name)
// test if new path already exists // test if new path already exists
if _, err := os.Stat(f); err == nil { if _, err := os.Stat(filename); err == nil {
return fmt.Errorf("Rename(): file %v already exists", f) return fmt.Errorf("Rename(): file %v already exists", filename)
} }
// create directories if necessary, ignore errors // create directories if necessary, ignore errors
if h.Type == backend.Data { if h.Type == backend.Data {
err = os.MkdirAll(filepath.Dir(f), backend.Modes.Dir) err = os.MkdirAll(filepath.Dir(filename), backend.Modes.Dir)
if err != nil { if err != nil {
return err return err
} }
} }
err = os.Rename(tmpfile.Name(), f) err = os.Rename(tmpfile, filename)
debug.Log("local.Save", "save %v: rename %v -> %v: %v", debug.Log("local.Save", "save %v: rename %v -> %v: %v",
h, filepath.Base(tmpfile.Name()), filepath.Base(f), err) h, filepath.Base(tmpfile), filepath.Base(filename), err)
if err != nil { if err != nil {
return err return err
} }
// set mode to read-only // set mode to read-only
fi, err := os.Stat(f) fi, err := os.Stat(filename)
if err != nil { if err != nil {
return err return err
} }
return setNewFileMode(f, fi) return setNewFileMode(filename, fi)
} }
// Stat returns information about a blob. // Stat returns information about a blob.