mirror of https://github.com/minio/minio.git
Adding object donut type
This commit is contained in:
parent
3af2894a9e
commit
7971b66e60
|
@ -0,0 +1,65 @@
|
||||||
|
package objectv1
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/binary"
|
||||||
|
"encoding/gob"
|
||||||
|
"errors"
|
||||||
|
"io"
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Package Version
|
||||||
|
const Version = uint32(1)
|
||||||
|
|
||||||
|
// ObjectType is the type of object stored. It is either an Object or Multipart Object.
|
||||||
|
type ObjectType uint8
|
||||||
|
|
||||||
|
const (
|
||||||
|
// Object is a full object
|
||||||
|
Object ObjectType = iota
|
||||||
|
// MultipartObject is a collection of Objects uploaded separately that represent a large object.
|
||||||
|
MultipartObject
|
||||||
|
)
|
||||||
|
|
||||||
|
// Object Metadata
|
||||||
|
type ObjectMetadata struct {
|
||||||
|
Bucket string
|
||||||
|
Key string
|
||||||
|
ErasurePart uint16
|
||||||
|
EncodedPart uint8
|
||||||
|
|
||||||
|
ContentType string
|
||||||
|
Created time.Time
|
||||||
|
Length uint64
|
||||||
|
Md5 []byte
|
||||||
|
ObjectType ObjectType
|
||||||
|
}
|
||||||
|
|
||||||
|
func Write(target io.Writer, metadata ObjectMetadata, reader io.Reader) error {
|
||||||
|
buffer := new(bytes.Buffer)
|
||||||
|
binary.Write(buffer, binary.LittleEndian, uint32(Version))
|
||||||
|
encoder := gob.NewEncoder(buffer)
|
||||||
|
if err := encoder.Encode(metadata); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
reader = io.MultiReader(buffer, reader)
|
||||||
|
_, err := io.Copy(target, reader)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func ReadMetadata(reader io.Reader) (metadata ObjectMetadata, err error) {
|
||||||
|
versionBytes := make([]byte, 4)
|
||||||
|
if err := binary.Read(reader, binary.LittleEndian, versionBytes); err != nil {
|
||||||
|
return metadata, err
|
||||||
|
}
|
||||||
|
var version uint32
|
||||||
|
version = binary.LittleEndian.Uint32(versionBytes)
|
||||||
|
if version != 1 {
|
||||||
|
return metadata, errors.New("Unknown Version: " + strconv.FormatUint(uint64(version), 10))
|
||||||
|
}
|
||||||
|
decoder := gob.NewDecoder(reader)
|
||||||
|
err = decoder.Decode(&metadata)
|
||||||
|
return metadata, err
|
||||||
|
}
|
|
@ -0,0 +1,61 @@
|
||||||
|
package objectv1
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"bytes"
|
||||||
|
"crypto/md5"
|
||||||
|
"encoding/binary"
|
||||||
|
"encoding/gob"
|
||||||
|
. "gopkg.in/check.v1"
|
||||||
|
"io"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Test(t *testing.T) { TestingT(t) }
|
||||||
|
|
||||||
|
type MySuite struct{}
|
||||||
|
|
||||||
|
var _ = Suite(&MySuite{})
|
||||||
|
|
||||||
|
func (s *MySuite) TestObjectV1ReadWrite(c *C) {
|
||||||
|
var buffer bytes.Buffer
|
||||||
|
|
||||||
|
data := "Hello, World"
|
||||||
|
|
||||||
|
hash := md5.New()
|
||||||
|
hash.Sum([]byte(data))
|
||||||
|
sum := hash.Sum(nil)
|
||||||
|
|
||||||
|
objectMetadata := ObjectMetadata{
|
||||||
|
Bucket: "bucket",
|
||||||
|
Key: "key",
|
||||||
|
ErasurePart: 1,
|
||||||
|
EncodedPart: 2,
|
||||||
|
|
||||||
|
ObjectType: Object,
|
||||||
|
Created: time.Now(),
|
||||||
|
ContentType: "application/text",
|
||||||
|
Md5: sum,
|
||||||
|
Length: uint64(len(sum)),
|
||||||
|
}
|
||||||
|
|
||||||
|
err := Write(&buffer, objectMetadata, bytes.NewBufferString(data))
|
||||||
|
c.Assert(err, IsNil)
|
||||||
|
|
||||||
|
versionBuffer := make([]byte, 4)
|
||||||
|
buffer.Read(versionBuffer)
|
||||||
|
c.Assert(binary.LittleEndian.Uint32(versionBuffer), Equals, uint32(1))
|
||||||
|
|
||||||
|
actualMetadata := ObjectMetadata{}
|
||||||
|
decoder := gob.NewDecoder(&buffer)
|
||||||
|
decoder.Decode(&actualMetadata)
|
||||||
|
|
||||||
|
c.Assert(actualMetadata, DeepEquals, objectMetadata)
|
||||||
|
|
||||||
|
var actualData bytes.Buffer
|
||||||
|
|
||||||
|
_, err = io.Copy(&actualData, &buffer)
|
||||||
|
c.Assert(err, IsNil)
|
||||||
|
c.Assert(actualData.Bytes(), DeepEquals, []byte(data))
|
||||||
|
}
|
Loading…
Reference in New Issue