You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
410 lines
13 KiB
410 lines
13 KiB
/*
|
|
* Minimalist Object Storage, (C) 2015 Minio, Inc.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package donut
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"encoding/hex"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
|
|
"io/ioutil"
|
|
|
|
"github.com/minio/minio/pkg/iodine"
|
|
"github.com/minio/minio/pkg/storage/donut"
|
|
"github.com/minio/minio/pkg/storage/drivers"
|
|
)
|
|
|
|
// donutDriver - creates a new single disk drivers driver using donut
|
|
type donutDriver struct {
|
|
donut donut.Donut
|
|
paths []string
|
|
lock *sync.RWMutex
|
|
}
|
|
|
|
// This is a dummy nodeDiskMap which is going to be deprecated soon
|
|
// 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
|
|
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))
|
|
nodes := make(map[string][]string)
|
|
for i, p := range paths {
|
|
diskPath := filepath.Join(p, strconv.Itoa(i))
|
|
if _, err := os.Stat(diskPath); err != nil {
|
|
if os.IsNotExist(err) {
|
|
os.MkdirAll(diskPath, 0700)
|
|
}
|
|
}
|
|
diskPaths[i] = diskPath
|
|
}
|
|
nodes["localhost"] = diskPaths
|
|
return nodes
|
|
}
|
|
|
|
// NewDriver instantiate a donut driver
|
|
func NewDriver(paths []string) (drivers.Driver, error) {
|
|
driver := new(donutDriver)
|
|
driver.paths = paths
|
|
driver.lock = new(sync.RWMutex)
|
|
|
|
// 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
|
|
driver.donut, err = donut.NewDonut("default", createNodeDiskMap(driver.paths))
|
|
return driver, iodine.New(err, nil)
|
|
}
|
|
|
|
// byBucketName is a type for sorting bucket metadata by bucket name
|
|
type byBucketName []drivers.BucketMetadata
|
|
|
|
func (b byBucketName) Len() int { return len(b) }
|
|
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 returns a list of buckets
|
|
func (d donutDriver) ListBuckets() (results []drivers.BucketMetadata, err error) {
|
|
if d.donut == nil {
|
|
return nil, iodine.New(drivers.InternalError{}, nil)
|
|
}
|
|
buckets, err := d.donut.ListBuckets()
|
|
if err != nil {
|
|
return nil, iodine.New(err, nil)
|
|
}
|
|
for _, metadata := range buckets {
|
|
result := drivers.BucketMetadata{
|
|
Name: metadata.Name,
|
|
Created: metadata.Created,
|
|
ACL: drivers.BucketACL(metadata.ACL),
|
|
}
|
|
results = append(results, result)
|
|
}
|
|
sort.Sort(byBucketName(results))
|
|
return results, nil
|
|
}
|
|
|
|
// CreateBucket creates a new bucket
|
|
func (d donutDriver) CreateBucket(bucketName, acl string) error {
|
|
d.lock.Lock()
|
|
defer d.lock.Unlock()
|
|
if d.donut == nil {
|
|
return iodine.New(drivers.InternalError{}, nil)
|
|
}
|
|
if !drivers.IsValidBucketACL(acl) {
|
|
return iodine.New(drivers.InvalidACL{ACL: acl}, nil)
|
|
}
|
|
if drivers.IsValidBucket(bucketName) {
|
|
if strings.TrimSpace(acl) == "" {
|
|
acl = "private"
|
|
}
|
|
if err := d.donut.MakeBucket(bucketName, acl); err != nil {
|
|
switch iodine.ToError(err).(type) {
|
|
case donut.BucketExists:
|
|
return iodine.New(drivers.BucketExists{Bucket: bucketName}, nil)
|
|
}
|
|
return iodine.New(err, nil)
|
|
}
|
|
return nil
|
|
}
|
|
return iodine.New(drivers.BucketNameInvalid{Bucket: bucketName}, nil)
|
|
}
|
|
|
|
// GetBucketMetadata retrieves an bucket's metadata
|
|
func (d donutDriver) GetBucketMetadata(bucketName string) (drivers.BucketMetadata, error) {
|
|
d.lock.RLock()
|
|
defer d.lock.RUnlock()
|
|
if d.donut == nil {
|
|
return drivers.BucketMetadata{}, iodine.New(drivers.InternalError{}, nil)
|
|
}
|
|
if !drivers.IsValidBucket(bucketName) {
|
|
return drivers.BucketMetadata{}, drivers.BucketNameInvalid{Bucket: bucketName}
|
|
}
|
|
metadata, err := d.donut.GetBucketMetadata(bucketName)
|
|
if err != nil {
|
|
return drivers.BucketMetadata{}, iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil)
|
|
}
|
|
bucketMetadata := drivers.BucketMetadata{
|
|
Name: metadata.Name,
|
|
Created: metadata.Created,
|
|
ACL: drivers.BucketACL(metadata.ACL),
|
|
}
|
|
return bucketMetadata, nil
|
|
}
|
|
|
|
// SetBucketMetadata sets bucket's metadata
|
|
func (d donutDriver) SetBucketMetadata(bucketName, acl string) error {
|
|
d.lock.Lock()
|
|
defer d.lock.Unlock()
|
|
if d.donut == nil {
|
|
return iodine.New(drivers.InternalError{}, nil)
|
|
}
|
|
if !drivers.IsValidBucket(bucketName) {
|
|
return iodine.New(drivers.BucketNameInvalid{Bucket: bucketName}, nil)
|
|
}
|
|
if strings.TrimSpace(acl) == "" {
|
|
acl = "private"
|
|
}
|
|
bucketMetadata := make(map[string]string)
|
|
bucketMetadata["acl"] = acl
|
|
err := d.donut.SetBucketMetadata(bucketName, bucketMetadata)
|
|
if err != nil {
|
|
return iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetObject retrieves an object and writes it to a writer
|
|
func (d donutDriver) GetObject(w io.Writer, bucketName, objectName string) (int64, error) {
|
|
if d.donut == nil {
|
|
return 0, iodine.New(drivers.InternalError{}, nil)
|
|
}
|
|
if !drivers.IsValidBucket(bucketName) {
|
|
return 0, iodine.New(drivers.BucketNameInvalid{Bucket: bucketName}, nil)
|
|
}
|
|
if !drivers.IsValidObjectName(objectName) {
|
|
return 0, iodine.New(drivers.ObjectNameInvalid{Object: objectName}, nil)
|
|
}
|
|
d.lock.RLock()
|
|
defer d.lock.RUnlock()
|
|
reader, size, err := d.donut.GetObject(bucketName, objectName)
|
|
if err != nil {
|
|
switch iodine.ToError(err).(type) {
|
|
case donut.BucketNotFound:
|
|
return 0, iodine.New(drivers.BucketNotFound{Bucket: bucketName}, nil)
|
|
case donut.ObjectNotFound:
|
|
return 0, iodine.New(drivers.ObjectNotFound{
|
|
Bucket: bucketName,
|
|
Object: objectName,
|
|
}, nil)
|
|
default:
|
|
return 0, iodine.New(drivers.InternalError{}, nil)
|
|
}
|
|
}
|
|
written, err := io.CopyN(w, reader, size)
|
|
if err != nil {
|
|
return 0, iodine.New(err, nil)
|
|
}
|
|
return written, nil
|
|
}
|
|
|
|
// GetPartialObject retrieves an object range and writes it to a writer
|
|
func (d donutDriver) GetPartialObject(w io.Writer, bucketName, objectName string, start, length int64) (int64, error) {
|
|
d.lock.RLock()
|
|
defer d.lock.RUnlock()
|
|
if d.donut == nil {
|
|
return 0, iodine.New(drivers.InternalError{}, nil)
|
|
}
|
|
errParams := map[string]string{
|
|
"bucketName": bucketName,
|
|
"objectName": objectName,
|
|
"start": strconv.FormatInt(start, 10),
|
|
"length": strconv.FormatInt(length, 10),
|
|
}
|
|
if !drivers.IsValidBucket(bucketName) {
|
|
return 0, iodine.New(drivers.BucketNameInvalid{Bucket: bucketName}, errParams)
|
|
}
|
|
if !drivers.IsValidObjectName(objectName) {
|
|
return 0, iodine.New(drivers.ObjectNameInvalid{Object: objectName}, errParams)
|
|
}
|
|
if start < 0 {
|
|
return 0, iodine.New(drivers.InvalidRange{
|
|
Start: start,
|
|
Length: length,
|
|
}, errParams)
|
|
}
|
|
reader, size, err := d.donut.GetObject(bucketName, objectName)
|
|
if err != nil {
|
|
switch iodine.ToError(err).(type) {
|
|
case donut.BucketNotFound:
|
|
return 0, iodine.New(drivers.BucketNotFound{Bucket: bucketName}, 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()
|
|
if start > size || (start+length-1) > size {
|
|
return 0, iodine.New(drivers.InvalidRange{
|
|
Start: start,
|
|
Length: length,
|
|
}, errParams)
|
|
}
|
|
_, err = io.CopyN(ioutil.Discard, reader, start)
|
|
if err != nil {
|
|
return 0, iodine.New(err, errParams)
|
|
}
|
|
n, err := io.CopyN(w, reader, length)
|
|
if err != nil {
|
|
return 0, iodine.New(err, errParams)
|
|
}
|
|
return n, nil
|
|
}
|
|
|
|
// GetObjectMetadata retrieves an object's metadata
|
|
func (d donutDriver) GetObjectMetadata(bucketName, objectName string) (drivers.ObjectMetadata, error) {
|
|
d.lock.RLock()
|
|
defer d.lock.RUnlock()
|
|
|
|
errParams := map[string]string{
|
|
"bucketName": bucketName,
|
|
"objectName": objectName,
|
|
}
|
|
if d.donut == nil {
|
|
return drivers.ObjectMetadata{}, iodine.New(drivers.InternalError{}, errParams)
|
|
}
|
|
if !drivers.IsValidBucket(bucketName) {
|
|
return drivers.ObjectMetadata{}, iodine.New(drivers.BucketNameInvalid{Bucket: bucketName}, errParams)
|
|
}
|
|
if !drivers.IsValidObjectName(objectName) {
|
|
return drivers.ObjectMetadata{}, iodine.New(drivers.ObjectNameInvalid{Object: objectName}, errParams)
|
|
}
|
|
metadata, err := d.donut.GetObjectMetadata(bucketName, objectName)
|
|
if err != nil {
|
|
return drivers.ObjectMetadata{}, iodine.New(drivers.ObjectNotFound{
|
|
Bucket: bucketName,
|
|
Object: objectName,
|
|
}, errParams)
|
|
}
|
|
objectMetadata := drivers.ObjectMetadata{
|
|
Bucket: bucketName,
|
|
Key: objectName,
|
|
|
|
ContentType: metadata.Metadata["contentType"],
|
|
Created: metadata.Created,
|
|
Md5: metadata.MD5Sum,
|
|
Size: metadata.Size,
|
|
}
|
|
return objectMetadata, nil
|
|
}
|
|
|
|
type byObjectName []drivers.ObjectMetadata
|
|
|
|
func (b byObjectName) Len() int { return len(b) }
|
|
func (b byObjectName) Swap(i, j int) { b[i], b[j] = b[j], b[i] }
|
|
func (b byObjectName) Less(i, j int) bool { return b[i].Key < b[j].Key }
|
|
|
|
// ListObjects - returns list of objects
|
|
func (d donutDriver) ListObjects(bucketName string, resources drivers.BucketResourcesMetadata) ([]drivers.ObjectMetadata, drivers.BucketResourcesMetadata, error) {
|
|
d.lock.RLock()
|
|
defer d.lock.RUnlock()
|
|
errParams := map[string]string{
|
|
"bucketName": bucketName,
|
|
}
|
|
if d.donut == nil {
|
|
return nil, drivers.BucketResourcesMetadata{}, iodine.New(drivers.InternalError{}, errParams)
|
|
}
|
|
if !drivers.IsValidBucket(bucketName) {
|
|
return nil, drivers.BucketResourcesMetadata{}, iodine.New(drivers.BucketNameInvalid{Bucket: bucketName}, nil)
|
|
}
|
|
if !drivers.IsValidObjectName(resources.Prefix) {
|
|
return nil, drivers.BucketResourcesMetadata{}, iodine.New(drivers.ObjectNameInvalid{Object: resources.Prefix}, nil)
|
|
}
|
|
listObjects, err := d.donut.ListObjects(bucketName, resources.Prefix, resources.Marker, resources.Delimiter, resources.Maxkeys)
|
|
if err != nil {
|
|
return nil, drivers.BucketResourcesMetadata{}, iodine.New(err, errParams)
|
|
}
|
|
resources.CommonPrefixes = listObjects.CommonPrefixes
|
|
resources.IsTruncated = listObjects.IsTruncated
|
|
var results []drivers.ObjectMetadata
|
|
for _, objMetadata := range listObjects.Objects {
|
|
metadata := drivers.ObjectMetadata{
|
|
Key: objMetadata.Object,
|
|
Created: objMetadata.Created,
|
|
Size: objMetadata.Size,
|
|
}
|
|
results = append(results, metadata)
|
|
}
|
|
sort.Sort(byObjectName(results))
|
|
if resources.IsTruncated && resources.IsDelimiterSet() {
|
|
resources.NextMarker = results[len(results)-1].Key
|
|
}
|
|
return results, resources, nil
|
|
}
|
|
|
|
// CreateObject creates a new object
|
|
func (d donutDriver) CreateObject(bucketName, objectName, contentType, expectedMD5Sum string, size int64, reader io.Reader) (string, error) {
|
|
d.lock.Lock()
|
|
defer d.lock.Unlock()
|
|
errParams := map[string]string{
|
|
"bucketName": bucketName,
|
|
"objectName": objectName,
|
|
"contentType": contentType,
|
|
}
|
|
if d.donut == nil {
|
|
return "", iodine.New(drivers.InternalError{}, errParams)
|
|
}
|
|
if !drivers.IsValidBucket(bucketName) {
|
|
return "", iodine.New(drivers.BucketNameInvalid{Bucket: bucketName}, nil)
|
|
}
|
|
if !drivers.IsValidObjectName(objectName) {
|
|
return "", iodine.New(drivers.ObjectNameInvalid{Object: objectName}, nil)
|
|
}
|
|
if strings.TrimSpace(contentType) == "" {
|
|
contentType = "application/octet-stream"
|
|
}
|
|
metadata := make(map[string]string)
|
|
metadata["contentType"] = strings.TrimSpace(contentType)
|
|
metadata["contentLength"] = strconv.FormatInt(size, 10)
|
|
if strings.TrimSpace(expectedMD5Sum) != "" {
|
|
expectedMD5SumBytes, err := base64.StdEncoding.DecodeString(strings.TrimSpace(expectedMD5Sum))
|
|
if err != nil {
|
|
return "", iodine.New(drivers.InvalidDigest{Md5: expectedMD5Sum}, nil)
|
|
}
|
|
expectedMD5Sum = hex.EncodeToString(expectedMD5SumBytes)
|
|
}
|
|
objMetadata, err := d.donut.PutObject(bucketName, objectName, expectedMD5Sum, reader, metadata)
|
|
if err != nil {
|
|
switch iodine.ToError(err).(type) {
|
|
case donut.BadDigest:
|
|
return "", iodine.New(drivers.BadDigest{Md5: expectedMD5Sum, Bucket: bucketName, Key: objectName}, nil)
|
|
}
|
|
return "", iodine.New(err, errParams)
|
|
}
|
|
newObject := drivers.ObjectMetadata{
|
|
Bucket: bucketName,
|
|
Key: objectName,
|
|
|
|
ContentType: objMetadata.Metadata["contentType"],
|
|
Created: objMetadata.Created,
|
|
Md5: objMetadata.MD5Sum,
|
|
Size: objMetadata.Size,
|
|
}
|
|
return newObject.Md5, nil
|
|
}
|
|
|