storj/uplink/storage/buckets/store.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)
	return store.metainfoClient.CreateBucket(ctx, bucket)
}

// 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, 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, 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, listOpts)
}