storj/uplink/storage/buckets/store.go
2019-08-06 16:56:23 +02:00

77 lines
2.3 KiB
Go

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
package buckets
import (
"context"
"gopkg.in/spacemonkeygo/monkit.v2"
"storj.io/storj/pkg/storj"
"storj.io/storj/uplink/metainfo"
)
var mon = monkit.Package()
// Store for buckets
type Store interface {
Create(ctx context.Context, bucket storj.Bucket) (_ storj.Bucket, err error)
Get(ctx context.Context, bucketName string) (_ storj.Bucket, err error)
Delete(ctx context.Context, bucketName string) (err error)
List(ctx context.Context, listOpts storj.BucketListOptions) (_ storj.BucketList, err error)
}
// BucketStore is an object to interact with buckets
// via the metainfo client
type BucketStore struct {
metainfoClient metainfo.Client
}
// NewStore creates a new bucket store
func NewStore(metainfoClient metainfo.Client) *BucketStore {
return &BucketStore{
metainfoClient: metainfoClient,
}
}
// Create creates a bucket
func (store *BucketStore) Create(ctx context.Context, bucket storj.Bucket) (_ storj.Bucket, err error) {
defer mon.Task()(&ctx)(&err)
// uuid MarshalJSON implementation always returns err == nil
partnerID, _ := bucket.PartnerID.MarshalJSON()
return store.metainfoClient.CreateBucket(ctx, metainfo.CreateBucketParams{
Name: []byte(bucket.Name),
PathCipher: bucket.PathCipher,
PartnerID: partnerID,
DefaultSegmentsSize: bucket.DefaultSegmentsSize,
DefaultRedundancyScheme: bucket.DefaultRedundancyScheme,
DefaultEncryptionParameters: bucket.DefaultEncryptionParameters,
})
}
// Get returns a bucket
func (store *BucketStore) Get(ctx context.Context, bucketName string) (_ storj.Bucket, err error) {
defer mon.Task()(&ctx)(&err)
return store.metainfoClient.GetBucket(ctx, metainfo.GetBucketParams{
Name: []byte(bucketName),
})
}
// Delete deletes a bucket
func (store *BucketStore) Delete(ctx context.Context, bucketName string) (err error) {
defer mon.Task()(&ctx)(&err)
return store.metainfoClient.DeleteBucket(ctx, metainfo.DeleteBucketParams{
Name: []byte(bucketName),
})
}
// List returns a list of buckets
func (store *BucketStore) List(ctx context.Context, listOpts storj.BucketListOptions) (_ storj.BucketList, err error) {
defer mon.Task()(&ctx)(&err)
return store.metainfoClient.ListBuckets(ctx, metainfo.ListBucketsParams{
ListOpts: listOpts,
})
}