|
|
|
|
@ -33,6 +33,7 @@ import ( |
|
|
|
|
"time" |
|
|
|
|
|
|
|
|
|
"github.com/minio/minio/pkg/iodine" |
|
|
|
|
"github.com/minio/minio/pkg/quick" |
|
|
|
|
"github.com/minio/minio/pkg/storage/donut/trove" |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
@ -41,15 +42,24 @@ const ( |
|
|
|
|
totalBuckets = 100 |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
// Cache - local variables
|
|
|
|
|
type Cache struct { |
|
|
|
|
storedBuckets map[string]storedBucket |
|
|
|
|
// Config donut config
|
|
|
|
|
type Config struct { |
|
|
|
|
Version string `json:"version"` |
|
|
|
|
MaxSize uint64 `json:"max-size"` |
|
|
|
|
Expiration time.Duration `json:"expiration"` |
|
|
|
|
DonutName string `json:"donut-name"` |
|
|
|
|
NodeDiskMap map[string][]string `json:"node-disk-map"` |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// API - local variables
|
|
|
|
|
type API struct { |
|
|
|
|
config *Config |
|
|
|
|
lock *sync.RWMutex |
|
|
|
|
objects *trove.Cache |
|
|
|
|
multiPartObjects *trove.Cache |
|
|
|
|
maxSize uint64 |
|
|
|
|
expiration time.Duration |
|
|
|
|
donut Donut |
|
|
|
|
storedBuckets map[string]storedBucket |
|
|
|
|
nodes map[string]node |
|
|
|
|
buckets map[string]bucket |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// storedBucket saved bucket
|
|
|
|
|
@ -67,79 +77,85 @@ type multiPartSession struct { |
|
|
|
|
initiated time.Time |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
type proxyWriter struct { |
|
|
|
|
writer io.Writer |
|
|
|
|
writtenBytes []byte |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (r *proxyWriter) Write(p []byte) (n int, err error) { |
|
|
|
|
n, err = r.writer.Write(p) |
|
|
|
|
if err != nil { |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
r.writtenBytes = append(r.writtenBytes, p[0:n]...) |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func newProxyWriter(w io.Writer) *proxyWriter { |
|
|
|
|
return &proxyWriter{writer: w, writtenBytes: nil} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// NewCache new cache
|
|
|
|
|
func NewCache(maxSize uint64, expiration time.Duration, donutName string, nodeDiskMap map[string][]string) Cache { |
|
|
|
|
c := Cache{} |
|
|
|
|
c.storedBuckets = make(map[string]storedBucket) |
|
|
|
|
c.objects = trove.NewCache(maxSize, expiration) |
|
|
|
|
c.multiPartObjects = trove.NewCache(0, time.Duration(0)) |
|
|
|
|
c.objects.OnExpired = c.expiredObject |
|
|
|
|
c.multiPartObjects.OnExpired = c.expiredPart |
|
|
|
|
c.lock = new(sync.RWMutex) |
|
|
|
|
c.maxSize = maxSize |
|
|
|
|
c.expiration = expiration |
|
|
|
|
// New instantiate a new donut
|
|
|
|
|
func New(c *Config) (Interface, error) { |
|
|
|
|
if err := quick.CheckData(c); err != nil { |
|
|
|
|
return nil, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
a := API{config: c} |
|
|
|
|
a.storedBuckets = make(map[string]storedBucket) |
|
|
|
|
a.nodes = make(map[string]node) |
|
|
|
|
a.buckets = make(map[string]bucket) |
|
|
|
|
a.objects = trove.NewCache(a.config.MaxSize, a.config.Expiration) |
|
|
|
|
a.multiPartObjects = trove.NewCache(0, time.Duration(0)) |
|
|
|
|
a.objects.OnExpired = a.expiredObject |
|
|
|
|
a.multiPartObjects.OnExpired = a.expiredPart |
|
|
|
|
a.lock = new(sync.RWMutex) |
|
|
|
|
|
|
|
|
|
// set up cache expiration
|
|
|
|
|
c.objects.ExpireObjects(time.Second * 5) |
|
|
|
|
c.donut, _ = NewDonut(donutName, nodeDiskMap) |
|
|
|
|
return c |
|
|
|
|
a.objects.ExpireObjects(time.Second * 5) |
|
|
|
|
|
|
|
|
|
if len(a.config.NodeDiskMap) > 0 { |
|
|
|
|
for k, v := range a.config.NodeDiskMap { |
|
|
|
|
if len(v) == 0 { |
|
|
|
|
return nil, iodine.New(InvalidDisksArgument{}, nil) |
|
|
|
|
} |
|
|
|
|
err := a.AttachNode(k, v) |
|
|
|
|
if err != nil { |
|
|
|
|
return nil, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
/// Initialization, populate all buckets into memory
|
|
|
|
|
buckets, err := a.listBuckets() |
|
|
|
|
if err != nil { |
|
|
|
|
return nil, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
for k, v := range buckets { |
|
|
|
|
storedBucket := a.storedBuckets[k] |
|
|
|
|
storedBucket.bucketMetadata = v |
|
|
|
|
a.storedBuckets[k] = storedBucket |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
return a, nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// GetObject - GET object from cache buffer
|
|
|
|
|
func (cache Cache) GetObject(w io.Writer, bucket string, object string) (int64, error) { |
|
|
|
|
cache.lock.RLock() |
|
|
|
|
func (donut API) GetObject(w io.Writer, bucket string, object string) (int64, error) { |
|
|
|
|
donut.lock.RLock() |
|
|
|
|
if !IsValidBucket(bucket) { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return 0, iodine.New(BucketNameInvalid{Bucket: bucket}, nil) |
|
|
|
|
} |
|
|
|
|
if !IsValidObjectName(object) { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return 0, iodine.New(ObjectNameInvalid{Object: object}, nil) |
|
|
|
|
} |
|
|
|
|
if _, ok := cache.storedBuckets[bucket]; ok == false { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
if _, ok := donut.storedBuckets[bucket]; ok == false { |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return 0, iodine.New(BucketNotFound{Bucket: bucket}, nil) |
|
|
|
|
} |
|
|
|
|
objectKey := bucket + "/" + object |
|
|
|
|
data, ok := cache.objects.Get(objectKey) |
|
|
|
|
data, ok := donut.objects.Get(objectKey) |
|
|
|
|
if !ok { |
|
|
|
|
if cache.donut != nil { |
|
|
|
|
reader, size, err := cache.donut.GetObject(bucket, object) |
|
|
|
|
if len(donut.config.NodeDiskMap) > 0 { |
|
|
|
|
reader, size, err := donut.getObject(bucket, object) |
|
|
|
|
if err != nil { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return 0, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
// new proxy writer to capture data read from disk
|
|
|
|
|
pw := newProxyWriter(w) |
|
|
|
|
pw := NewProxyWriter(w) |
|
|
|
|
written, err := io.CopyN(pw, reader, size) |
|
|
|
|
if err != nil { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return 0, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
/// cache object read from disk
|
|
|
|
|
{ |
|
|
|
|
cache.lock.Lock() |
|
|
|
|
ok := cache.objects.Set(objectKey, pw.writtenBytes) |
|
|
|
|
cache.lock.Unlock() |
|
|
|
|
donut.lock.Lock() |
|
|
|
|
ok := donut.objects.Set(objectKey, pw.writtenBytes) |
|
|
|
|
donut.lock.Unlock() |
|
|
|
|
pw.writtenBytes = nil |
|
|
|
|
go debug.FreeOSMemory() |
|
|
|
|
if !ok { |
|
|
|
|
@ -148,65 +164,65 @@ func (cache Cache) GetObject(w io.Writer, bucket string, object string) (int64, |
|
|
|
|
} |
|
|
|
|
return written, nil |
|
|
|
|
} |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return 0, iodine.New(ObjectNotFound{Object: object}, nil) |
|
|
|
|
} |
|
|
|
|
written, err := io.CopyN(w, bytes.NewBuffer(data), int64(cache.objects.Len(objectKey))) |
|
|
|
|
written, err := io.CopyN(w, bytes.NewBuffer(data), int64(donut.objects.Len(objectKey))) |
|
|
|
|
if err != nil { |
|
|
|
|
return 0, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return written, nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// GetPartialObject - GET object from cache buffer range
|
|
|
|
|
func (cache Cache) GetPartialObject(w io.Writer, bucket, object string, start, length int64) (int64, error) { |
|
|
|
|
func (donut API) GetPartialObject(w io.Writer, bucket, object string, start, length int64) (int64, error) { |
|
|
|
|
errParams := map[string]string{ |
|
|
|
|
"bucket": bucket, |
|
|
|
|
"object": object, |
|
|
|
|
"start": strconv.FormatInt(start, 10), |
|
|
|
|
"length": strconv.FormatInt(length, 10), |
|
|
|
|
} |
|
|
|
|
cache.lock.RLock() |
|
|
|
|
donut.lock.RLock() |
|
|
|
|
if !IsValidBucket(bucket) { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return 0, iodine.New(BucketNameInvalid{Bucket: bucket}, errParams) |
|
|
|
|
} |
|
|
|
|
if !IsValidObjectName(object) { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return 0, iodine.New(ObjectNameInvalid{Object: object}, errParams) |
|
|
|
|
} |
|
|
|
|
if start < 0 { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return 0, iodine.New(InvalidRange{ |
|
|
|
|
Start: start, |
|
|
|
|
Length: length, |
|
|
|
|
}, errParams) |
|
|
|
|
} |
|
|
|
|
objectKey := bucket + "/" + object |
|
|
|
|
data, ok := cache.objects.Get(objectKey) |
|
|
|
|
data, ok := donut.objects.Get(objectKey) |
|
|
|
|
if !ok { |
|
|
|
|
if cache.donut != nil { |
|
|
|
|
reader, _, err := cache.donut.GetObject(bucket, object) |
|
|
|
|
if len(donut.config.NodeDiskMap) > 0 { |
|
|
|
|
reader, _, err := donut.getObject(bucket, object) |
|
|
|
|
if err != nil { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return 0, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
if _, err := io.CopyN(ioutil.Discard, reader, start); err != nil { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return 0, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
pw := newProxyWriter(w) |
|
|
|
|
pw := NewProxyWriter(w) |
|
|
|
|
written, err := io.CopyN(w, reader, length) |
|
|
|
|
if err != nil { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return 0, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
{ |
|
|
|
|
cache.lock.Lock() |
|
|
|
|
ok := cache.objects.Set(objectKey, pw.writtenBytes) |
|
|
|
|
cache.lock.Unlock() |
|
|
|
|
donut.lock.Lock() |
|
|
|
|
ok := donut.objects.Set(objectKey, pw.writtenBytes) |
|
|
|
|
donut.lock.Unlock() |
|
|
|
|
pw.writtenBytes = nil |
|
|
|
|
go debug.FreeOSMemory() |
|
|
|
|
if !ok { |
|
|
|
|
@ -215,72 +231,70 @@ func (cache Cache) GetPartialObject(w io.Writer, bucket, object string, start, l |
|
|
|
|
} |
|
|
|
|
return written, nil |
|
|
|
|
} |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return 0, iodine.New(ObjectNotFound{Object: object}, nil) |
|
|
|
|
} |
|
|
|
|
written, err := io.CopyN(w, bytes.NewBuffer(data[start:]), length) |
|
|
|
|
if err != nil { |
|
|
|
|
return 0, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return written, nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// GetBucketMetadata -
|
|
|
|
|
func (cache Cache) GetBucketMetadata(bucket string) (BucketMetadata, error) { |
|
|
|
|
cache.lock.RLock() |
|
|
|
|
func (donut API) GetBucketMetadata(bucket string) (BucketMetadata, error) { |
|
|
|
|
donut.lock.RLock() |
|
|
|
|
if !IsValidBucket(bucket) { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return BucketMetadata{}, iodine.New(BucketNameInvalid{Bucket: bucket}, nil) |
|
|
|
|
} |
|
|
|
|
if _, ok := cache.storedBuckets[bucket]; ok == false { |
|
|
|
|
if cache.donut == nil { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
return BucketMetadata{}, iodine.New(BucketNotFound{Bucket: bucket}, nil) |
|
|
|
|
} |
|
|
|
|
bucketMetadata, err := cache.donut.GetBucketMetadata(bucket) |
|
|
|
|
if err != nil { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
return BucketMetadata{}, iodine.New(err, nil) |
|
|
|
|
if _, ok := donut.storedBuckets[bucket]; ok == false { |
|
|
|
|
if len(donut.config.NodeDiskMap) > 0 { |
|
|
|
|
bucketMetadata, err := donut.getBucketMetadata(bucket) |
|
|
|
|
if err != nil { |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return BucketMetadata{}, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
storedBucket := donut.storedBuckets[bucket] |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
{ |
|
|
|
|
donut.lock.Lock() |
|
|
|
|
storedBucket.bucketMetadata = bucketMetadata |
|
|
|
|
donut.storedBuckets[bucket] = storedBucket |
|
|
|
|
donut.lock.Unlock() |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
storedBucket := cache.storedBuckets[bucket] |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
cache.lock.Lock() |
|
|
|
|
storedBucket.bucketMetadata = bucketMetadata |
|
|
|
|
cache.storedBuckets[bucket] = storedBucket |
|
|
|
|
cache.lock.Unlock() |
|
|
|
|
} |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
return cache.storedBuckets[bucket].bucketMetadata, nil |
|
|
|
|
return BucketMetadata{}, iodine.New(BucketNotFound{Bucket: bucket}, nil) |
|
|
|
|
} |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return donut.storedBuckets[bucket].bucketMetadata, nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// SetBucketMetadata -
|
|
|
|
|
func (cache Cache) SetBucketMetadata(bucket, acl string) error { |
|
|
|
|
cache.lock.RLock() |
|
|
|
|
func (donut API) SetBucketMetadata(bucket string, metadata map[string]string) error { |
|
|
|
|
donut.lock.RLock() |
|
|
|
|
if !IsValidBucket(bucket) { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return iodine.New(BucketNameInvalid{Bucket: bucket}, nil) |
|
|
|
|
} |
|
|
|
|
if _, ok := cache.storedBuckets[bucket]; ok == false { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
if _, ok := donut.storedBuckets[bucket]; ok == false { |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return iodine.New(BucketNotFound{Bucket: bucket}, nil) |
|
|
|
|
} |
|
|
|
|
if strings.TrimSpace(acl) == "" { |
|
|
|
|
acl = "private" |
|
|
|
|
} |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
cache.lock.Lock() |
|
|
|
|
m := make(map[string]string) |
|
|
|
|
m["acl"] = acl |
|
|
|
|
if cache.donut != nil { |
|
|
|
|
if err := cache.donut.SetBucketMetadata(bucket, m); err != nil { |
|
|
|
|
return iodine.New(err, nil) |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
donut.lock.Lock() |
|
|
|
|
{ |
|
|
|
|
if len(donut.config.NodeDiskMap) > 0 { |
|
|
|
|
if err := donut.setBucketMetadata(bucket, metadata); err != nil { |
|
|
|
|
return iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
storedBucket := donut.storedBuckets[bucket] |
|
|
|
|
storedBucket.bucketMetadata.ACL = BucketACL(metadata["acl"]) |
|
|
|
|
donut.storedBuckets[bucket] = storedBucket |
|
|
|
|
} |
|
|
|
|
storedBucket := cache.storedBuckets[bucket] |
|
|
|
|
storedBucket.bucketMetadata.ACL = BucketACL(acl) |
|
|
|
|
cache.storedBuckets[bucket] = storedBucket |
|
|
|
|
cache.lock.Unlock() |
|
|
|
|
donut.lock.Unlock() |
|
|
|
|
return nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
@ -304,44 +318,45 @@ func isMD5SumEqual(expectedMD5Sum, actualMD5Sum string) error { |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// CreateObject -
|
|
|
|
|
func (cache Cache) CreateObject(bucket, key, contentType, expectedMD5Sum string, size int64, data io.Reader) (ObjectMetadata, error) { |
|
|
|
|
if size > int64(cache.maxSize) { |
|
|
|
|
func (donut API) CreateObject(bucket, key, expectedMD5Sum string, size int64, data io.Reader, metadata map[string]string) (ObjectMetadata, error) { |
|
|
|
|
if size > int64(donut.config.MaxSize) { |
|
|
|
|
generic := GenericObjectError{Bucket: bucket, Object: key} |
|
|
|
|
return ObjectMetadata{}, iodine.New(EntityTooLarge{ |
|
|
|
|
GenericObjectError: generic, |
|
|
|
|
Size: strconv.FormatInt(size, 10), |
|
|
|
|
MaxSize: strconv.FormatUint(cache.maxSize, 10), |
|
|
|
|
MaxSize: strconv.FormatUint(donut.config.MaxSize, 10), |
|
|
|
|
}, nil) |
|
|
|
|
} |
|
|
|
|
objectMetadata, err := cache.createObject(bucket, key, contentType, expectedMD5Sum, size, data) |
|
|
|
|
contentType := metadata["contentType"] |
|
|
|
|
objectMetadata, err := donut.createObject(bucket, key, contentType, expectedMD5Sum, size, data) |
|
|
|
|
// free
|
|
|
|
|
debug.FreeOSMemory() |
|
|
|
|
return objectMetadata, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// createObject - PUT object to cache buffer
|
|
|
|
|
func (cache Cache) createObject(bucket, key, contentType, expectedMD5Sum string, size int64, data io.Reader) (ObjectMetadata, error) { |
|
|
|
|
cache.lock.RLock() |
|
|
|
|
func (donut API) createObject(bucket, key, contentType, expectedMD5Sum string, size int64, data io.Reader) (ObjectMetadata, error) { |
|
|
|
|
donut.lock.RLock() |
|
|
|
|
if !IsValidBucket(bucket) { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return ObjectMetadata{}, iodine.New(BucketNameInvalid{Bucket: bucket}, nil) |
|
|
|
|
} |
|
|
|
|
if !IsValidObjectName(key) { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return ObjectMetadata{}, iodine.New(ObjectNameInvalid{Object: key}, nil) |
|
|
|
|
} |
|
|
|
|
if _, ok := cache.storedBuckets[bucket]; ok == false { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
if _, ok := donut.storedBuckets[bucket]; ok == false { |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return ObjectMetadata{}, iodine.New(BucketNotFound{Bucket: bucket}, nil) |
|
|
|
|
} |
|
|
|
|
storedBucket := cache.storedBuckets[bucket] |
|
|
|
|
storedBucket := donut.storedBuckets[bucket] |
|
|
|
|
// get object key
|
|
|
|
|
objectKey := bucket + "/" + key |
|
|
|
|
if _, ok := storedBucket.objectMetadata[objectKey]; ok == true { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return ObjectMetadata{}, iodine.New(ObjectExists{Object: key}, nil) |
|
|
|
|
} |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
|
|
|
|
|
if contentType == "" { |
|
|
|
|
contentType = "application/octet-stream" |
|
|
|
|
@ -356,15 +371,15 @@ func (cache Cache) createObject(bucket, key, contentType, expectedMD5Sum string, |
|
|
|
|
expectedMD5Sum = hex.EncodeToString(expectedMD5SumBytes) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if cache.donut != nil { |
|
|
|
|
objMetadata, err := cache.donut.PutObject(bucket, key, expectedMD5Sum, data, map[string]string{"contentType": contentType}) |
|
|
|
|
if len(donut.config.NodeDiskMap) > 0 { |
|
|
|
|
objMetadata, err := donut.putObject(bucket, key, expectedMD5Sum, data, map[string]string{"contentType": contentType}) |
|
|
|
|
if err != nil { |
|
|
|
|
return ObjectMetadata{}, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
cache.lock.Lock() |
|
|
|
|
donut.lock.Lock() |
|
|
|
|
storedBucket.objectMetadata[objectKey] = objMetadata |
|
|
|
|
cache.storedBuckets[bucket] = storedBucket |
|
|
|
|
cache.lock.Unlock() |
|
|
|
|
donut.storedBuckets[bucket] = storedBucket |
|
|
|
|
donut.lock.Unlock() |
|
|
|
|
return objMetadata, nil |
|
|
|
|
} |
|
|
|
|
// calculate md5
|
|
|
|
|
@ -382,9 +397,9 @@ func (cache Cache) createObject(bucket, key, contentType, expectedMD5Sum string, |
|
|
|
|
break |
|
|
|
|
} |
|
|
|
|
hash.Write(byteBuffer[0:length]) |
|
|
|
|
cache.lock.Lock() |
|
|
|
|
ok := cache.objects.Append(objectKey, byteBuffer[0:length]) |
|
|
|
|
cache.lock.Unlock() |
|
|
|
|
donut.lock.Lock() |
|
|
|
|
ok := donut.objects.Append(objectKey, byteBuffer[0:length]) |
|
|
|
|
donut.lock.Unlock() |
|
|
|
|
if !ok { |
|
|
|
|
return ObjectMetadata{}, iodine.New(InternalError{}, nil) |
|
|
|
|
} |
|
|
|
|
@ -416,40 +431,40 @@ func (cache Cache) createObject(bucket, key, contentType, expectedMD5Sum string, |
|
|
|
|
Size: int64(totalLength), |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
cache.lock.Lock() |
|
|
|
|
donut.lock.Lock() |
|
|
|
|
storedBucket.objectMetadata[objectKey] = newObject |
|
|
|
|
cache.storedBuckets[bucket] = storedBucket |
|
|
|
|
cache.lock.Unlock() |
|
|
|
|
donut.storedBuckets[bucket] = storedBucket |
|
|
|
|
donut.lock.Unlock() |
|
|
|
|
return newObject, nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// MakeBucket - create bucket in cache
|
|
|
|
|
func (cache Cache) MakeBucket(bucketName, acl string) error { |
|
|
|
|
cache.lock.RLock() |
|
|
|
|
if len(cache.storedBuckets) == totalBuckets { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
func (donut API) MakeBucket(bucketName, acl string) error { |
|
|
|
|
donut.lock.RLock() |
|
|
|
|
if len(donut.storedBuckets) == totalBuckets { |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return iodine.New(TooManyBuckets{Bucket: bucketName}, nil) |
|
|
|
|
} |
|
|
|
|
if !IsValidBucket(bucketName) { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return iodine.New(BucketNameInvalid{Bucket: bucketName}, nil) |
|
|
|
|
} |
|
|
|
|
if !IsValidBucketACL(acl) { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return iodine.New(InvalidACL{ACL: acl}, nil) |
|
|
|
|
} |
|
|
|
|
if _, ok := cache.storedBuckets[bucketName]; ok == true { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
if _, ok := donut.storedBuckets[bucketName]; ok == true { |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return iodine.New(BucketExists{Bucket: bucketName}, nil) |
|
|
|
|
} |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
|
|
|
|
|
if strings.TrimSpace(acl) == "" { |
|
|
|
|
// default is private
|
|
|
|
|
acl = "private" |
|
|
|
|
} |
|
|
|
|
if cache.donut != nil { |
|
|
|
|
if err := cache.donut.MakeBucket(bucketName, BucketACL(acl)); err != nil { |
|
|
|
|
if len(donut.config.NodeDiskMap) > 0 { |
|
|
|
|
if err := donut.makeBucket(bucketName, BucketACL(acl)); err != nil { |
|
|
|
|
return iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
@ -461,29 +476,29 @@ func (cache Cache) MakeBucket(bucketName, acl string) error { |
|
|
|
|
newBucket.bucketMetadata.Name = bucketName |
|
|
|
|
newBucket.bucketMetadata.Created = time.Now().UTC() |
|
|
|
|
newBucket.bucketMetadata.ACL = BucketACL(acl) |
|
|
|
|
cache.lock.Lock() |
|
|
|
|
cache.storedBuckets[bucketName] = newBucket |
|
|
|
|
cache.lock.Unlock() |
|
|
|
|
donut.lock.Lock() |
|
|
|
|
donut.storedBuckets[bucketName] = newBucket |
|
|
|
|
donut.lock.Unlock() |
|
|
|
|
return nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// ListObjects - list objects from cache
|
|
|
|
|
func (cache Cache) ListObjects(bucket string, resources BucketResourcesMetadata) ([]ObjectMetadata, BucketResourcesMetadata, error) { |
|
|
|
|
cache.lock.RLock() |
|
|
|
|
defer cache.lock.RUnlock() |
|
|
|
|
func (donut API) ListObjects(bucket string, resources BucketResourcesMetadata) ([]ObjectMetadata, BucketResourcesMetadata, error) { |
|
|
|
|
donut.lock.RLock() |
|
|
|
|
defer donut.lock.RUnlock() |
|
|
|
|
if !IsValidBucket(bucket) { |
|
|
|
|
return nil, BucketResourcesMetadata{IsTruncated: false}, iodine.New(BucketNameInvalid{Bucket: bucket}, nil) |
|
|
|
|
} |
|
|
|
|
if !IsValidPrefix(resources.Prefix) { |
|
|
|
|
return nil, BucketResourcesMetadata{IsTruncated: false}, iodine.New(ObjectNameInvalid{Object: resources.Prefix}, nil) |
|
|
|
|
} |
|
|
|
|
if _, ok := cache.storedBuckets[bucket]; ok == false { |
|
|
|
|
if _, ok := donut.storedBuckets[bucket]; ok == false { |
|
|
|
|
return nil, BucketResourcesMetadata{IsTruncated: false}, iodine.New(BucketNotFound{Bucket: bucket}, nil) |
|
|
|
|
} |
|
|
|
|
var results []ObjectMetadata |
|
|
|
|
var keys []string |
|
|
|
|
if cache.donut != nil { |
|
|
|
|
listObjects, err := cache.donut.ListObjects( |
|
|
|
|
if len(donut.config.NodeDiskMap) > 0 { |
|
|
|
|
listObjects, err := donut.listObjects( |
|
|
|
|
bucket, |
|
|
|
|
resources.Prefix, |
|
|
|
|
resources.Marker, |
|
|
|
|
@ -507,7 +522,7 @@ func (cache Cache) ListObjects(bucket string, resources BucketResourcesMetadata) |
|
|
|
|
} |
|
|
|
|
return results, resources, nil |
|
|
|
|
} |
|
|
|
|
storedBucket := cache.storedBuckets[bucket] |
|
|
|
|
storedBucket := donut.storedBuckets[bucket] |
|
|
|
|
for key := range storedBucket.objectMetadata { |
|
|
|
|
if strings.HasPrefix(key, bucket+"/") { |
|
|
|
|
key = key[len(bucket)+1:] |
|
|
|
|
@ -561,11 +576,11 @@ func (b byBucketName) Swap(i, j int) { b[i], b[j] = b[j], b[i] } |
|
|
|
|
func (b byBucketName) Less(i, j int) bool { return b[i].Name < b[j].Name } |
|
|
|
|
|
|
|
|
|
// ListBuckets - List buckets from cache
|
|
|
|
|
func (cache Cache) ListBuckets() ([]BucketMetadata, error) { |
|
|
|
|
cache.lock.RLock() |
|
|
|
|
defer cache.lock.RUnlock() |
|
|
|
|
func (donut API) ListBuckets() ([]BucketMetadata, error) { |
|
|
|
|
donut.lock.RLock() |
|
|
|
|
defer donut.lock.RUnlock() |
|
|
|
|
var results []BucketMetadata |
|
|
|
|
for _, bucket := range cache.storedBuckets { |
|
|
|
|
for _, bucket := range donut.storedBuckets { |
|
|
|
|
results = append(results, bucket.bucketMetadata) |
|
|
|
|
} |
|
|
|
|
sort.Sort(byBucketName(results)) |
|
|
|
|
@ -573,50 +588,50 @@ func (cache Cache) ListBuckets() ([]BucketMetadata, error) { |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// GetObjectMetadata - get object metadata from cache
|
|
|
|
|
func (cache Cache) GetObjectMetadata(bucket, key string) (ObjectMetadata, error) { |
|
|
|
|
cache.lock.RLock() |
|
|
|
|
func (donut API) GetObjectMetadata(bucket, key string) (ObjectMetadata, error) { |
|
|
|
|
donut.lock.RLock() |
|
|
|
|
// check if bucket exists
|
|
|
|
|
if !IsValidBucket(bucket) { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return ObjectMetadata{}, iodine.New(BucketNameInvalid{Bucket: bucket}, nil) |
|
|
|
|
} |
|
|
|
|
if !IsValidObjectName(key) { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return ObjectMetadata{}, iodine.New(ObjectNameInvalid{Object: key}, nil) |
|
|
|
|
} |
|
|
|
|
if _, ok := cache.storedBuckets[bucket]; ok == false { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
if _, ok := donut.storedBuckets[bucket]; ok == false { |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return ObjectMetadata{}, iodine.New(BucketNotFound{Bucket: bucket}, nil) |
|
|
|
|
} |
|
|
|
|
storedBucket := cache.storedBuckets[bucket] |
|
|
|
|
storedBucket := donut.storedBuckets[bucket] |
|
|
|
|
objectKey := bucket + "/" + key |
|
|
|
|
if objMetadata, ok := storedBucket.objectMetadata[objectKey]; ok == true { |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return objMetadata, nil |
|
|
|
|
} |
|
|
|
|
if cache.donut != nil { |
|
|
|
|
objMetadata, err := cache.donut.GetObjectMetadata(bucket, key) |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
if len(donut.config.NodeDiskMap) > 0 { |
|
|
|
|
objMetadata, err := donut.getObjectMetadata(bucket, key) |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
if err != nil { |
|
|
|
|
return ObjectMetadata{}, iodine.New(err, nil) |
|
|
|
|
} |
|
|
|
|
// update
|
|
|
|
|
cache.lock.Lock() |
|
|
|
|
donut.lock.Lock() |
|
|
|
|
storedBucket.objectMetadata[objectKey] = objMetadata |
|
|
|
|
cache.lock.Unlock() |
|
|
|
|
donut.lock.Unlock() |
|
|
|
|
return objMetadata, nil |
|
|
|
|
} |
|
|
|
|
cache.lock.RUnlock() |
|
|
|
|
donut.lock.RUnlock() |
|
|
|
|
return ObjectMetadata{}, iodine.New(ObjectNotFound{Object: key}, nil) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (cache Cache) expiredObject(a ...interface{}) { |
|
|
|
|
cacheStats := cache.objects.Stats() |
|
|
|
|
func (donut API) expiredObject(a ...interface{}) { |
|
|
|
|
cacheStats := donut.objects.Stats() |
|
|
|
|
log.Printf("CurrentSize: %d, CurrentItems: %d, TotalExpirations: %d", |
|
|
|
|
cacheStats.Bytes, cacheStats.Items, cacheStats.Expired) |
|
|
|
|
key := a[0].(string) |
|
|
|
|
// loop through all buckets
|
|
|
|
|
for _, storedBucket := range cache.storedBuckets { |
|
|
|
|
for _, storedBucket := range donut.storedBuckets { |
|
|
|
|
delete(storedBucket.objectMetadata, key) |
|
|
|
|
} |
|
|
|
|
debug.FreeOSMemory() |
Harshavardhana
9 years ago