MyFavMirrors/minio
1
0
Fork 0
mirror of https://github.com/minio/minio.git synced 2024-12-26 07:05:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

Seek to beginning for tempbuffer

Browse Source
This commit is contained in:
Frederick F. Kautz IV 2015-02-27 12:18:08 -08:00
parent da10fd5a83
commit be768d01a2
2 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
pkg/storage/donut/v1/donut.go
View File

@ -22,6 +22,8 @@ import (
"encoding/gob"
"errors"
"io"
"io/ioutil"
"os"
"sync"
"github.com/minio-io/minio/pkg/storage/erasure"
@ -206,26 +208,30 @@ func (donut *Donut) Write(gobHeader GobHeader, object io.Reader) error {
BlockEnd: MagicINIM,
}
var tempBuffer bytes.Buffer
tempBuffer, err := ioutil.TempFile(os.TempDir(), "minio-staging")
if err != nil {
return err
}
defer os.Remove(tempBuffer.Name())
// write header
if err := donut.WriteBegin(&tempBuffer, donutFormat); err != nil {
if err := donut.WriteBegin(tempBuffer, donutFormat); err != nil {
return err
}
// write data
if err := donut.WriteData(&tempBuffer, donutFormat); err != nil {
if err := donut.WriteData(tempBuffer, donutFormat); err != nil {
return err
}
// write footer crc
if err := donut.WriteEnd(&tempBuffer, donutFormat); err != nil {
if err := donut.WriteEnd(tempBuffer, donutFormat); err != nil {
return err
}
// write footer
donut.file.Seek(0, 2)
io.Copy(donut.file, &tempBuffer)
tempBuffer.Seek(0, 0)
io.Copy(donut.file, tempBuffer)
return nil
}

4
pkg/storage/donut/v1/donut_test.go
View File

@ -19,7 +19,6 @@ package v1
import (
"bytes"
. "gopkg.in/check.v1"
"io"
"io/ioutil"
"os"
"testing"
@ -32,10 +31,11 @@ type MySuite struct{}
var _ = Suite(&MySuite{})
func (s *MySuite) TestAPISuite(c *C) {
var b io.ReadWriteSeeker
//var b io.ReadWriteSeeker
var o bytes.Buffer
b, err := ioutil.TempFile(os.TempDir(), "minio-donut-test")
defer os.RemoveAll(b.Name())
c.Assert(err, IsNil)
donut := New(b)