Merge pull request #685 from harshavardhana/pr_out_add_append_method_to_trove_cache_for_appending_data_to_an_existing_key

Add Append() method to trove cache for appending data to an existing key
This commit is contained in:
Harshavardhana 2015-06-28 03:28:06 +00:00
commit a4d20d1e75
3 changed files with 208 additions and 179 deletions

View File

@ -68,30 +68,24 @@ type donutDriver struct {
expiration time.Duration expiration time.Duration
} }
// This is a dummy nodeDiskMap which is going to be deprecated soon
// once the Management API is standardized, this map is useful for now
// to show multi disk API correctness and parity calculation
//
// Ideally this should be obtained from per node configuration file
func createNodeDiskMap(p string) map[string][]string {
nodes := make(map[string][]string)
nodes["localhost"] = make([]string, 16)
for i := 0; i < len(nodes["localhost"]); i++ {
diskPath := filepath.Join(p, strconv.Itoa(i))
if _, err := os.Stat(diskPath); err != nil {
if os.IsNotExist(err) {
os.MkdirAll(diskPath, 0700)
}
}
nodes["localhost"][i] = diskPath
}
return nodes
}
// This is a dummy nodeDiskMap which is going to be deprecated soon // This is a dummy nodeDiskMap which is going to be deprecated soon
// once the Management API is standardized, and we have way of adding // once the Management API is standardized, and we have way of adding
// and removing disks. This is useful for now to take inputs from CLI // and removing disks. This is useful for now to take inputs from CLI
func createNodeDiskMapFromSlice(paths []string) map[string][]string { func createNodeDiskMap(paths []string) map[string][]string {
if len(paths) == 1 {
nodes := make(map[string][]string)
nodes["localhost"] = make([]string, 16)
for i := 0; i < len(nodes["localhost"]); i++ {
diskPath := filepath.Join(paths[0], strconv.Itoa(i))
if _, err := os.Stat(diskPath); err != nil {
if os.IsNotExist(err) {
os.MkdirAll(diskPath, 0700)
}
}
nodes["localhost"][i] = diskPath
}
return nodes
}
diskPaths := make([]string, len(paths)) diskPaths := make([]string, len(paths))
nodes := make(map[string][]string) nodes := make(map[string][]string)
for i, p := range paths { for i, p := range paths {
@ -107,6 +101,43 @@ func createNodeDiskMapFromSlice(paths []string) map[string][]string {
return nodes return nodes
} }
func initialize(d *donutDriver) {
// Soon to be user configurable, when Management API is available
// we should remove "default" to something which is passed down
// from configuration paramters
var err error
d.donut, err = donut.NewDonut("default", createNodeDiskMap(d.paths))
if err != nil {
panic(iodine.New(err, nil))
}
buckets, err := d.donut.ListBuckets()
if err != nil {
panic(iodine.New(err, nil))
}
for bucketName, metadata := range buckets {
d.lock.RLock()
storedBucket := d.storedBuckets[bucketName]
d.lock.RUnlock()
if len(storedBucket.multiPartSession) == 0 {
storedBucket.multiPartSession = make(map[string]multiPartSession)
}
if len(storedBucket.objectMetadata) == 0 {
storedBucket.objectMetadata = make(map[string]drivers.ObjectMetadata)
}
if len(storedBucket.partMetadata) == 0 {
storedBucket.partMetadata = make(map[string]drivers.PartMetadata)
}
storedBucket.bucketMetadata = drivers.BucketMetadata{
Name: metadata.Name,
Created: metadata.Created,
ACL: drivers.BucketACL(metadata.ACL),
}
d.lock.Lock()
d.storedBuckets[bucketName] = storedBucket
d.lock.Unlock()
}
}
// Start a single disk subsystem // Start a single disk subsystem
func Start(paths []string, maxSize uint64, expiration time.Duration) (chan<- string, <-chan error, drivers.Driver) { func Start(paths []string, maxSize uint64, expiration time.Duration) (chan<- string, <-chan error, drivers.Driver) {
ctrlChannel := make(chan string) ctrlChannel := make(chan string)
@ -126,34 +157,17 @@ func Start(paths []string, maxSize uint64, expiration time.Duration) (chan<- str
// set up memory expiration // set up memory expiration
driver.objects.ExpireObjects(time.Second * 5) driver.objects.ExpireObjects(time.Second * 5)
// Soon to be user configurable, when Management API is available
// we should remove "default" to something which is passed down
// from configuration paramters
switch {
case len(paths) == 1:
d, err := donut.NewDonut("default", createNodeDiskMap(paths[0]))
if err != nil {
err = iodine.New(err, nil)
log.Error.Println(err)
}
driver.donut = d
default:
d, err := donut.NewDonut("default", createNodeDiskMapFromSlice(paths))
if err != nil {
err = iodine.New(err, nil)
log.Error.Println(err)
}
driver.donut = d
}
driver.paths = paths driver.paths = paths
driver.lock = new(sync.RWMutex) driver.lock = new(sync.RWMutex)
initialize(driver)
go start(ctrlChannel, errorChannel, driver) go start(ctrlChannel, errorChannel, driver)
return ctrlChannel, errorChannel, driver return ctrlChannel, errorChannel, driver
} }
func start(ctrlChannel <-chan string, errorChannel chan<- error, driver *donutDriver) { func start(ctrlChannel <-chan string, errorChannel chan<- error, driver *donutDriver) {
close(errorChannel) defer close(errorChannel)
} }
func (d donutDriver) expiredObject(a ...interface{}) { func (d donutDriver) expiredObject(a ...interface{}) {
@ -186,31 +200,8 @@ func (d donutDriver) ListBuckets() (results []drivers.BucketMetadata, err error)
if d.donut == nil { if d.donut == nil {
return nil, iodine.New(drivers.InternalError{}, nil) return nil, iodine.New(drivers.InternalError{}, nil)
} }
buckets, err := d.donut.ListBuckets() for _, storedBucket := range d.storedBuckets {
if err != nil { results = append(results, storedBucket.bucketMetadata)
return nil, iodine.New(err, nil)
}
for bucketName, metadata := range buckets {
result := drivers.BucketMetadata{
Name: metadata.Name,
Created: metadata.Created,
ACL: drivers.BucketACL(metadata.ACL),
}
d.lock.Lock()
storedBucket := d.storedBuckets[bucketName]
if len(storedBucket.multiPartSession) == 0 {
storedBucket.multiPartSession = make(map[string]multiPartSession)
}
if len(storedBucket.objectMetadata) == 0 {
storedBucket.objectMetadata = make(map[string]drivers.ObjectMetadata)
}
if len(storedBucket.partMetadata) == 0 {
storedBucket.partMetadata = make(map[string]drivers.PartMetadata)
}
storedBucket.bucketMetadata = result
d.storedBuckets[bucketName] = storedBucket
d.lock.Unlock()
results = append(results, result)
} }
sort.Sort(byBucketName(results)) sort.Sort(byBucketName(results))
return results, nil return results, nil
@ -229,7 +220,7 @@ func (d donutDriver) CreateBucket(bucketName, acl string) error {
if !drivers.IsValidBucketACL(acl) { if !drivers.IsValidBucketACL(acl) {
return iodine.New(drivers.InvalidACL{ACL: acl}, nil) return iodine.New(drivers.InvalidACL{ACL: acl}, nil)
} }
if drivers.IsValidBucket(bucketName) && !strings.Contains(bucketName, ".") { if drivers.IsValidBucket(bucketName) {
if strings.TrimSpace(acl) == "" { if strings.TrimSpace(acl) == "" {
acl = "private" acl = "private"
} }
@ -244,6 +235,15 @@ func (d donutDriver) CreateBucket(bucketName, acl string) error {
newBucket.objectMetadata = make(map[string]drivers.ObjectMetadata) newBucket.objectMetadata = make(map[string]drivers.ObjectMetadata)
newBucket.multiPartSession = make(map[string]multiPartSession) newBucket.multiPartSession = make(map[string]multiPartSession)
newBucket.partMetadata = make(map[string]drivers.PartMetadata) newBucket.partMetadata = make(map[string]drivers.PartMetadata)
metadata, err := d.donut.GetBucketMetadata(bucketName)
if err != nil {
return iodine.New(err, nil)
}
newBucket.bucketMetadata = drivers.BucketMetadata{
Name: metadata.Name,
Created: metadata.Created,
ACL: drivers.BucketACL(metadata.ACL),
}
d.storedBuckets[bucketName] = newBucket d.storedBuckets[bucketName] = newBucket
return nil return nil
} }
@ -268,7 +268,7 @@ func (d donutDriver) GetBucketMetadata(bucketName string) (drivers.BucketMetadat
return drivers.BucketMetadata{}, iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil) return drivers.BucketMetadata{}, iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil)
} }
bucketMetadata := drivers.BucketMetadata{ bucketMetadata := drivers.BucketMetadata{
Name: bucketName, Name: metadata.Name,
Created: metadata.Created, Created: metadata.Created,
ACL: drivers.BucketACL(metadata.ACL), ACL: drivers.BucketACL(metadata.ACL),
} }
@ -302,8 +302,6 @@ func (d donutDriver) SetBucketMetadata(bucketName, acl string) error {
// GetObject retrieves an object and writes it to a writer // GetObject retrieves an object and writes it to a writer
func (d donutDriver) GetObject(w io.Writer, bucketName, objectName string) (int64, error) { func (d donutDriver) GetObject(w io.Writer, bucketName, objectName string) (int64, error) {
d.lock.RLock()
defer d.lock.RUnlock()
if d.donut == nil { if d.donut == nil {
return 0, iodine.New(drivers.InternalError{}, nil) return 0, iodine.New(drivers.InternalError{}, nil)
} }
@ -316,15 +314,24 @@ func (d donutDriver) GetObject(w io.Writer, bucketName, objectName string) (int6
if _, ok := d.storedBuckets[bucketName]; ok == false { if _, ok := d.storedBuckets[bucketName]; ok == false {
return 0, iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil) return 0, iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil)
} }
d.lock.RLock()
defer d.lock.RUnlock()
objectKey := bucketName + "/" + objectName objectKey := bucketName + "/" + objectName
data, ok := d.objects.Get(objectKey) data, ok := d.objects.Get(objectKey)
if !ok { if !ok {
reader, size, err := d.donut.GetObject(bucketName, objectName) reader, size, err := d.donut.GetObject(bucketName, objectName)
if err != nil { if err != nil {
return 0, iodine.New(drivers.ObjectNotFound{ switch iodine.ToError(err).(type) {
Bucket: bucketName, case donut.BucketNotFound:
Object: objectName, return 0, iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil)
}, nil) case donut.ObjectNotFound:
return 0, iodine.New(drivers.ObjectNotFound{
Bucket: bucketName,
Object: objectName,
}, nil)
default:
return 0, iodine.New(drivers.InternalError{}, nil)
}
} }
n, err := io.CopyN(w, reader, size) n, err := io.CopyN(w, reader, size)
if err != nil { if err != nil {
@ -372,10 +379,17 @@ func (d donutDriver) GetPartialObject(w io.Writer, bucketName, objectName string
if !ok { if !ok {
reader, size, err := d.donut.GetObject(bucketName, objectName) reader, size, err := d.donut.GetObject(bucketName, objectName)
if err != nil { if err != nil {
return 0, iodine.New(drivers.ObjectNotFound{ switch iodine.ToError(err).(type) {
Bucket: bucketName, case donut.BucketNotFound:
Object: objectName, return 0, iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil)
}, nil) case donut.ObjectNotFound:
return 0, iodine.New(drivers.ObjectNotFound{
Bucket: bucketName,
Object: objectName,
}, nil)
default:
return 0, iodine.New(drivers.InternalError{}, nil)
}
} }
defer reader.Close() defer reader.Close()
if start > size || (start+length-1) > size { if start > size || (start+length-1) > size {
@ -486,25 +500,32 @@ func (d donutDriver) ListObjects(bucketName string, resources drivers.BucketReso
} }
type proxyReader struct { type proxyReader struct {
io.Reader reader io.Reader
readBytes []byte driver donutDriver
object string
} }
func (r *proxyReader) Read(p []byte) (n int, err error) { func (r *proxyReader) Read(p []byte) (n int, err error) {
n, err = r.Reader.Read(p) n, err = r.reader.Read(p)
if err == io.EOF || err == io.ErrUnexpectedEOF { if err == io.EOF || err == io.ErrUnexpectedEOF {
r.readBytes = append(r.readBytes, p[0:n]...) ok := r.driver.objects.Append(r.object, p[0:n])
if !ok {
return n, io.ErrShortBuffer
}
return return
} }
if err != nil { if err != nil {
return return
} }
r.readBytes = append(r.readBytes, p[0:n]...) ok := r.driver.objects.Append(r.object, p[0:n])
if !ok {
return n, io.ErrShortBuffer
}
return return
} }
func newProxyReader(r io.Reader) *proxyReader { func newProxyReader(r io.Reader, d donutDriver, k string) *proxyReader {
return &proxyReader{r, nil} return &proxyReader{reader: r, driver: d, object: k}
} }
// CreateObject creates a new object // CreateObject creates a new object
@ -552,7 +573,7 @@ func (d donutDriver) CreateObject(bucketName, objectName, contentType, expectedM
} }
expectedMD5Sum = hex.EncodeToString(expectedMD5SumBytes) expectedMD5Sum = hex.EncodeToString(expectedMD5SumBytes)
} }
newReader := newProxyReader(reader) newReader := newProxyReader(reader, d, objectKey)
calculatedMD5Sum, err := d.donut.PutObject(bucketName, objectName, expectedMD5Sum, newReader, metadata) calculatedMD5Sum, err := d.donut.PutObject(bucketName, objectName, expectedMD5Sum, newReader, metadata)
if err != nil { if err != nil {
switch iodine.ToError(err).(type) { switch iodine.ToError(err).(type) {
@ -561,14 +582,9 @@ func (d donutDriver) CreateObject(bucketName, objectName, contentType, expectedM
} }
return "", iodine.New(err, errParams) return "", iodine.New(err, errParams)
} }
// get object key // free up
ok := d.objects.Set(objectKey, newReader.readBytes)
// setting up for de-allocation
newReader.readBytes = nil
go debug.FreeOSMemory() go debug.FreeOSMemory()
if !ok {
return "", iodine.New(drivers.InternalError{}, nil)
}
objectMetadata, err := d.donut.GetObjectMetadata(bucketName, objectName) objectMetadata, err := d.donut.GetObjectMetadata(bucketName, objectName)
if err != nil { if err != nil {
return "", iodine.New(err, nil) return "", iodine.New(err, nil)

View File

@ -64,33 +64,6 @@ func (d donutDriver) NewMultipartUpload(bucketName, objectName, contentType stri
d.lock.RUnlock() d.lock.RUnlock()
return "", iodine.New(drivers.ObjectNameInvalid{Object: objectName}, nil) return "", iodine.New(drivers.ObjectNameInvalid{Object: objectName}, nil)
} }
d.lock.RUnlock()
buckets, err := d.donut.ListBuckets()
if err != nil {
return "", iodine.New(err, nil)
}
for bucketName, metadata := range buckets {
result := drivers.BucketMetadata{
Name: metadata.Name,
Created: metadata.Created,
ACL: drivers.BucketACL(metadata.ACL),
}
d.lock.Lock()
storedBucket := d.storedBuckets[bucketName]
storedBucket.bucketMetadata = result
if len(storedBucket.multiPartSession) == 0 {
storedBucket.multiPartSession = make(map[string]multiPartSession)
}
if len(storedBucket.objectMetadata) == 0 {
storedBucket.objectMetadata = make(map[string]drivers.ObjectMetadata)
}
if len(storedBucket.partMetadata) == 0 {
storedBucket.partMetadata = make(map[string]drivers.PartMetadata)
}
d.storedBuckets[bucketName] = storedBucket
d.lock.Unlock()
}
d.lock.RLock()
if _, ok := d.storedBuckets[bucketName]; ok == false { if _, ok := d.storedBuckets[bucketName]; ok == false {
d.lock.RUnlock() d.lock.RUnlock()
return "", iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil) return "", iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil)
@ -117,17 +90,17 @@ func (d donutDriver) NewMultipartUpload(bucketName, objectName, contentType stri
return uploadID, nil return uploadID, nil
} }
func (d donutDriver) AbortMultipartUpload(bucket, key, uploadID string) error { func (d donutDriver) AbortMultipartUpload(bucketName, objectName, uploadID string) error {
d.lock.RLock() d.lock.RLock()
storedBucket := d.storedBuckets[bucket] storedBucket := d.storedBuckets[bucketName]
if storedBucket.multiPartSession[key].uploadID != uploadID { if storedBucket.multiPartSession[objectName].uploadID != uploadID {
d.lock.RUnlock() d.lock.RUnlock()
return iodine.New(drivers.InvalidUploadID{UploadID: uploadID}, nil) return iodine.New(drivers.InvalidUploadID{UploadID: uploadID}, nil)
} }
d.lock.RUnlock() d.lock.RUnlock()
d.cleanupMultiparts(bucket, key, uploadID) d.cleanupMultiparts(bucketName, objectName, uploadID)
d.cleanupMultipartSession(bucket, key, uploadID) d.cleanupMultipartSession(bucketName, objectName, uploadID)
return nil return nil
} }
@ -135,17 +108,17 @@ func getMultipartKey(key string, uploadID string, partNumber int) string {
return key + "?uploadId=" + uploadID + "&partNumber=" + strconv.Itoa(partNumber) return key + "?uploadId=" + uploadID + "&partNumber=" + strconv.Itoa(partNumber)
} }
func (d donutDriver) CreateObjectPart(bucket, key, uploadID string, partID int, contentType, expectedMD5Sum string, size int64, data io.Reader) (string, error) { func (d donutDriver) CreateObjectPart(bucketName, objectName, uploadID string, partID int, contentType, expectedMD5Sum string, size int64, data io.Reader) (string, error) {
// Verify upload id // Verify upload id
d.lock.RLock() d.lock.RLock()
storedBucket := d.storedBuckets[bucket] storedBucket := d.storedBuckets[bucketName]
if storedBucket.multiPartSession[key].uploadID != uploadID { if storedBucket.multiPartSession[objectName].uploadID != uploadID {
d.lock.RUnlock() d.lock.RUnlock()
return "", iodine.New(drivers.InvalidUploadID{UploadID: uploadID}, nil) return "", iodine.New(drivers.InvalidUploadID{UploadID: uploadID}, nil)
} }
d.lock.RUnlock() d.lock.RUnlock()
etag, err := d.createObjectPart(bucket, key, uploadID, partID, "", expectedMD5Sum, size, data) etag, err := d.createObjectPart(bucketName, objectName, uploadID, partID, "", expectedMD5Sum, size, data)
if err != nil { if err != nil {
return "", iodine.New(err, nil) return "", iodine.New(err, nil)
} }
@ -155,23 +128,23 @@ func (d donutDriver) CreateObjectPart(bucket, key, uploadID string, partID int,
} }
// createObject - PUT object to memory buffer // createObject - PUT object to memory buffer
func (d donutDriver) createObjectPart(bucket, key, uploadID string, partID int, contentType, expectedMD5Sum string, size int64, data io.Reader) (string, error) { func (d donutDriver) createObjectPart(bucketName, objectName, uploadID string, partID int, contentType, expectedMD5Sum string, size int64, data io.Reader) (string, error) {
d.lock.RLock() d.lock.RLock()
if !drivers.IsValidBucket(bucket) { if !drivers.IsValidBucket(bucketName) {
d.lock.RUnlock() d.lock.RUnlock()
return "", iodine.New(drivers.BucketNameInvalid{Bucket: bucket}, nil) return "", iodine.New(drivers.BucketNameInvalid{Bucket: bucketName}, nil)
} }
if !drivers.IsValidObjectName(key) { if !drivers.IsValidObjectName(objectName) {
d.lock.RUnlock() d.lock.RUnlock()
return "", iodine.New(drivers.ObjectNameInvalid{Object: key}, nil) return "", iodine.New(drivers.ObjectNameInvalid{Object: objectName}, nil)
} }
if _, ok := d.storedBuckets[bucket]; ok == false { if _, ok := d.storedBuckets[bucketName]; ok == false {
d.lock.RUnlock() d.lock.RUnlock()
return "", iodine.New(drivers.BucketNotFound{Bucket: bucket}, nil) return "", iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil)
} }
storedBucket := d.storedBuckets[bucket] storedBucket := d.storedBuckets[bucketName]
// get object key // get object key
partKey := bucket + "/" + getMultipartKey(key, uploadID, partID) partKey := bucketName + "/" + getMultipartKey(objectName, uploadID, partID)
if _, ok := storedBucket.partMetadata[partKey]; ok == true { if _, ok := storedBucket.partMetadata[partKey]; ok == true {
d.lock.RUnlock() d.lock.RUnlock()
return storedBucket.partMetadata[partKey].ETag, nil return storedBucket.partMetadata[partKey].ETag, nil
@ -225,7 +198,11 @@ func (d donutDriver) createObjectPart(bucket, key, uploadID string, partID int,
// Verify if the written object is equal to what is expected, only if it is requested as such // Verify if the written object is equal to what is expected, only if it is requested as such
if strings.TrimSpace(expectedMD5Sum) != "" { if strings.TrimSpace(expectedMD5Sum) != "" {
if err := isMD5SumEqual(strings.TrimSpace(expectedMD5Sum), md5Sum); err != nil { if err := isMD5SumEqual(strings.TrimSpace(expectedMD5Sum), md5Sum); err != nil {
return "", iodine.New(drivers.BadDigest{Md5: expectedMD5Sum, Bucket: bucket, Key: key}, nil) return "", iodine.New(drivers.BadDigest{
Md5: expectedMD5Sum,
Bucket: bucketName,
Key: objectName,
}, nil)
} }
} }
newPart := drivers.PartMetadata{ newPart := drivers.PartMetadata{
@ -237,43 +214,43 @@ func (d donutDriver) createObjectPart(bucket, key, uploadID string, partID int,
d.lock.Lock() d.lock.Lock()
storedBucket.partMetadata[partKey] = newPart storedBucket.partMetadata[partKey] = newPart
multiPartSession := storedBucket.multiPartSession[key] multiPartSession := storedBucket.multiPartSession[objectName]
multiPartSession.totalParts++ multiPartSession.totalParts++
storedBucket.multiPartSession[key] = multiPartSession storedBucket.multiPartSession[objectName] = multiPartSession
d.storedBuckets[bucket] = storedBucket d.storedBuckets[bucketName] = storedBucket
d.lock.Unlock() d.lock.Unlock()
return md5Sum, nil return md5Sum, nil
} }
func (d donutDriver) cleanupMultipartSession(bucket, key, uploadID string) { func (d donutDriver) cleanupMultipartSession(bucketName, objectName, uploadID string) {
d.lock.Lock() d.lock.Lock()
defer d.lock.Unlock() defer d.lock.Unlock()
delete(d.storedBuckets[bucket].multiPartSession, key) delete(d.storedBuckets[bucketName].multiPartSession, objectName)
} }
func (d donutDriver) cleanupMultiparts(bucket, key, uploadID string) { func (d donutDriver) cleanupMultiparts(bucketName, objectName, uploadID string) {
for i := 1; i <= d.storedBuckets[bucket].multiPartSession[key].totalParts; i++ { for i := 1; i <= d.storedBuckets[bucketName].multiPartSession[objectName].totalParts; i++ {
objectKey := bucket + "/" + getMultipartKey(key, uploadID, i) objectKey := bucketName + "/" + getMultipartKey(objectName, uploadID, i)
d.multiPartObjects.Delete(objectKey) d.multiPartObjects.Delete(objectKey)
} }
} }
func (d donutDriver) CompleteMultipartUpload(bucket, key, uploadID string, parts map[int]string) (string, error) { func (d donutDriver) CompleteMultipartUpload(bucketName, objectName, uploadID string, parts map[int]string) (string, error) {
if !drivers.IsValidBucket(bucket) { if !drivers.IsValidBucket(bucketName) {
return "", iodine.New(drivers.BucketNameInvalid{Bucket: bucket}, nil) return "", iodine.New(drivers.BucketNameInvalid{Bucket: bucketName}, nil)
} }
if !drivers.IsValidObjectName(key) { if !drivers.IsValidObjectName(objectName) {
return "", iodine.New(drivers.ObjectNameInvalid{Object: key}, nil) return "", iodine.New(drivers.ObjectNameInvalid{Object: objectName}, nil)
} }
// Verify upload id // Verify upload id
d.lock.RLock() d.lock.RLock()
if _, ok := d.storedBuckets[bucket]; ok == false { if _, ok := d.storedBuckets[bucketName]; ok == false {
d.lock.RUnlock() d.lock.RUnlock()
return "", iodine.New(drivers.BucketNotFound{Bucket: bucket}, nil) return "", iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil)
} }
storedBucket := d.storedBuckets[bucket] storedBucket := d.storedBuckets[bucketName]
if storedBucket.multiPartSession[key].uploadID != uploadID { if storedBucket.multiPartSession[objectName].uploadID != uploadID {
d.lock.RUnlock() d.lock.RUnlock()
return "", iodine.New(drivers.InvalidUploadID{UploadID: uploadID}, nil) return "", iodine.New(drivers.InvalidUploadID{UploadID: uploadID}, nil)
} }
@ -284,7 +261,7 @@ func (d donutDriver) CompleteMultipartUpload(bucket, key, uploadID string, parts
var fullObject bytes.Buffer var fullObject bytes.Buffer
for i := 1; i <= len(parts); i++ { for i := 1; i <= len(parts); i++ {
recvMD5 := parts[i] recvMD5 := parts[i]
object, ok := d.multiPartObjects.Get(bucket + "/" + getMultipartKey(key, uploadID, i)) object, ok := d.multiPartObjects.Get(bucketName + "/" + getMultipartKey(objectName, uploadID, i))
if ok == false { if ok == false {
d.lock.Unlock() d.lock.Unlock()
return "", iodine.New(errors.New("missing part: "+strconv.Itoa(i)), nil) return "", iodine.New(errors.New("missing part: "+strconv.Itoa(i)), nil)
@ -297,7 +274,11 @@ func (d donutDriver) CompleteMultipartUpload(bucket, key, uploadID string, parts
return "", iodine.New(drivers.InvalidDigest{Md5: recvMD5}, nil) return "", iodine.New(drivers.InvalidDigest{Md5: recvMD5}, nil)
} }
if !bytes.Equal(recvMD5Bytes, calcMD5Bytes[:]) { if !bytes.Equal(recvMD5Bytes, calcMD5Bytes[:]) {
return "", iodine.New(drivers.BadDigest{Md5: recvMD5, Bucket: bucket, Key: getMultipartKey(key, uploadID, i)}, nil) return "", iodine.New(drivers.BadDigest{
Md5: recvMD5,
Bucket: bucketName,
Key: getMultipartKey(objectName, uploadID, i),
}, nil)
} }
_, err = io.Copy(&fullObject, bytes.NewBuffer(object)) _, err = io.Copy(&fullObject, bytes.NewBuffer(object))
if err != nil { if err != nil {
@ -311,15 +292,15 @@ func (d donutDriver) CompleteMultipartUpload(bucket, key, uploadID string, parts
md5sumSlice := md5.Sum(fullObject.Bytes()) md5sumSlice := md5.Sum(fullObject.Bytes())
// this is needed for final verification inside CreateObject, do not convert this to hex // this is needed for final verification inside CreateObject, do not convert this to hex
md5sum := base64.StdEncoding.EncodeToString(md5sumSlice[:]) md5sum := base64.StdEncoding.EncodeToString(md5sumSlice[:])
etag, err := d.CreateObject(bucket, key, "", md5sum, size, &fullObject) etag, err := d.CreateObject(bucketName, objectName, "", md5sum, size, &fullObject)
if err != nil { if err != nil {
// No need to call internal cleanup functions here, caller will call AbortMultipartUpload() // No need to call internal cleanup functions here, caller will call AbortMultipartUpload()
// which would in-turn cleanup properly in accordance with S3 Spec // which would in-turn cleanup properly in accordance with S3 Spec
return "", iodine.New(err, nil) return "", iodine.New(err, nil)
} }
fullObject.Reset() fullObject.Reset()
d.cleanupMultiparts(bucket, key, uploadID) d.cleanupMultiparts(bucketName, objectName, uploadID)
d.cleanupMultipartSession(bucket, key, uploadID) d.cleanupMultipartSession(bucketName, objectName, uploadID)
return etag, nil return etag, nil
} }
@ -330,14 +311,13 @@ func (a byKey) Len() int { return len(a) }
func (a byKey) Swap(i, j int) { a[i], a[j] = a[j], a[i] } func (a byKey) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
func (a byKey) Less(i, j int) bool { return a[i].Key < a[j].Key } func (a byKey) Less(i, j int) bool { return a[i].Key < a[j].Key }
func (d donutDriver) ListMultipartUploads(bucket string, resources drivers.BucketMultipartResourcesMetadata) (drivers.BucketMultipartResourcesMetadata, error) { func (d donutDriver) ListMultipartUploads(bucketName string, resources drivers.BucketMultipartResourcesMetadata) (drivers.BucketMultipartResourcesMetadata, error) {
// TODO handle delimiter
d.lock.RLock() d.lock.RLock()
defer d.lock.RUnlock() defer d.lock.RUnlock()
if _, ok := d.storedBuckets[bucket]; ok == false { if _, ok := d.storedBuckets[bucketName]; ok == false {
return drivers.BucketMultipartResourcesMetadata{}, iodine.New(drivers.BucketNotFound{Bucket: bucket}, nil) return drivers.BucketMultipartResourcesMetadata{}, iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil)
} }
storedBucket := d.storedBuckets[bucket] storedBucket := d.storedBuckets[bucketName]
var uploads []*drivers.UploadMetadata var uploads []*drivers.UploadMetadata
for key, session := range storedBucket.multiPartSession { for key, session := range storedBucket.multiPartSession {
@ -391,23 +371,23 @@ func (a partNumber) Len() int { return len(a) }
func (a partNumber) Swap(i, j int) { a[i], a[j] = a[j], a[i] } func (a partNumber) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
func (a partNumber) Less(i, j int) bool { return a[i].PartNumber < a[j].PartNumber } func (a partNumber) Less(i, j int) bool { return a[i].PartNumber < a[j].PartNumber }
func (d donutDriver) ListObjectParts(bucket, key string, resources drivers.ObjectResourcesMetadata) (drivers.ObjectResourcesMetadata, error) { func (d donutDriver) ListObjectParts(bucketName, objectName string, resources drivers.ObjectResourcesMetadata) (drivers.ObjectResourcesMetadata, error) {
// Verify upload id // Verify upload id
d.lock.RLock() d.lock.RLock()
defer d.lock.RUnlock() defer d.lock.RUnlock()
if _, ok := d.storedBuckets[bucket]; ok == false { if _, ok := d.storedBuckets[bucketName]; ok == false {
return drivers.ObjectResourcesMetadata{}, iodine.New(drivers.BucketNotFound{Bucket: bucket}, nil) return drivers.ObjectResourcesMetadata{}, iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil)
} }
storedBucket := d.storedBuckets[bucket] storedBucket := d.storedBuckets[bucketName]
if _, ok := storedBucket.multiPartSession[key]; ok == false { if _, ok := storedBucket.multiPartSession[objectName]; ok == false {
return drivers.ObjectResourcesMetadata{}, iodine.New(drivers.ObjectNotFound{Bucket: bucket, Object: key}, nil) return drivers.ObjectResourcesMetadata{}, iodine.New(drivers.ObjectNotFound{Bucket: bucketName, Object: objectName}, nil)
} }
if storedBucket.multiPartSession[key].uploadID != resources.UploadID { if storedBucket.multiPartSession[objectName].uploadID != resources.UploadID {
return drivers.ObjectResourcesMetadata{}, iodine.New(drivers.InvalidUploadID{UploadID: resources.UploadID}, nil) return drivers.ObjectResourcesMetadata{}, iodine.New(drivers.InvalidUploadID{UploadID: resources.UploadID}, nil)
} }
objectResourcesMetadata := resources objectResourcesMetadata := resources
objectResourcesMetadata.Bucket = bucket objectResourcesMetadata.Bucket = bucketName
objectResourcesMetadata.Key = key objectResourcesMetadata.Key = objectName
var parts []*drivers.PartMetadata var parts []*drivers.PartMetadata
var startPartNumber int var startPartNumber int
switch { switch {
@ -416,7 +396,7 @@ func (d donutDriver) ListObjectParts(bucket, key string, resources drivers.Objec
default: default:
startPartNumber = objectResourcesMetadata.PartNumberMarker startPartNumber = objectResourcesMetadata.PartNumberMarker
} }
for i := startPartNumber; i <= storedBucket.multiPartSession[key].totalParts; i++ { for i := startPartNumber; i <= storedBucket.multiPartSession[objectName].totalParts; i++ {
if len(parts) > objectResourcesMetadata.MaxParts { if len(parts) > objectResourcesMetadata.MaxParts {
sort.Sort(partNumber(parts)) sort.Sort(partNumber(parts))
objectResourcesMetadata.IsTruncated = true objectResourcesMetadata.IsTruncated = true
@ -424,7 +404,7 @@ func (d donutDriver) ListObjectParts(bucket, key string, resources drivers.Objec
objectResourcesMetadata.NextPartNumberMarker = i objectResourcesMetadata.NextPartNumberMarker = i
return objectResourcesMetadata, nil return objectResourcesMetadata, nil
} }
part, ok := storedBucket.partMetadata[bucket+"/"+getMultipartKey(key, resources.UploadID, i)] part, ok := storedBucket.partMetadata[bucketName+"/"+getMultipartKey(objectName, resources.UploadID, i)]
if !ok { if !ok {
return drivers.ObjectResourcesMetadata{}, iodine.New(errors.New("missing part: "+strconv.Itoa(i)), nil) return drivers.ObjectResourcesMetadata{}, iodine.New(errors.New("missing part: "+strconv.Itoa(i)), nil)
} }

View File

@ -115,6 +115,39 @@ func (r *Cache) Get(key string) ([]byte, bool) {
return value, true return value, true
} }
// Append will append new data to an existing key,
// if key doesn't exist it behaves like Set()
func (r *Cache) Append(key string, value []byte) bool {
r.Lock()
defer r.Unlock()
valueLen := uint64(len(value))
if r.maxSize > 0 {
// check if the size of the object is not bigger than the
// capacity of the cache
if valueLen > r.maxSize {
return false
}
// remove random key if only we reach the maxSize threshold
for (r.currentSize + valueLen) > r.maxSize {
for randomKey := range r.items {
r.doDelete(randomKey)
break
}
}
}
_, ok := r.items[key]
if !ok {
r.items[key] = value
r.currentSize += valueLen
r.updatedAt[key] = time.Now()
return true
}
r.items[key] = append(r.items[key], value...)
r.currentSize += valueLen
r.updatedAt[key] = time.Now()
return true
}
// Set will persist a value to the cache // Set will persist a value to the cache
func (r *Cache) Set(key string, value []byte) bool { func (r *Cache) Set(key string, value []byte) bool {
r.Lock() r.Lock()
@ -128,8 +161,8 @@ func (r *Cache) Set(key string, value []byte) bool {
} }
// remove random key if only we reach the maxSize threshold // remove random key if only we reach the maxSize threshold
for (r.currentSize + valueLen) > r.maxSize { for (r.currentSize + valueLen) > r.maxSize {
for key := range r.items { for randomKey := range r.items {
r.doDelete(key) r.doDelete(randomKey)
break break
} }
} }