66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/keegancsmith/rpc"
|
|
|
|
"forge.cadoles.com/arcad/edge/pkg/storage"
|
|
"forge.cadoles.com/arcad/edge/pkg/storage/rpc/server/blob"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type BlobStore struct {
|
|
client *rpc.Client
|
|
}
|
|
|
|
// DeleteBucket implements storage.BlobStore.
|
|
func (s *BlobStore) DeleteBucket(ctx context.Context, name string) error {
|
|
args := &blob.DeleteBucketArgs{
|
|
BucketName: name,
|
|
}
|
|
|
|
if err := s.client.Call(ctx, "Service.DeleteBucket", args, nil); err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// ListBuckets implements storage.BlobStore.
|
|
func (s *BlobStore) ListBuckets(ctx context.Context) ([]string, error) {
|
|
args := &blob.ListBucketsArgs{}
|
|
|
|
reply := blob.ListBucketsReply{}
|
|
|
|
if err := s.client.Call(ctx, "Service.ListBuckets", args, &reply); err != nil {
|
|
return nil, errors.WithStack(err)
|
|
}
|
|
|
|
return reply.Buckets, nil
|
|
}
|
|
|
|
// OpenBucket implements storage.BlobStore.
|
|
func (s *BlobStore) OpenBucket(ctx context.Context, name string) (storage.BlobBucket, error) {
|
|
args := &blob.OpenBucketArgs{
|
|
BucketName: name,
|
|
}
|
|
reply := &blob.OpenBucketReply{}
|
|
|
|
if err := s.client.Call(ctx, "Service.OpenBucket", args, reply); err != nil {
|
|
return nil, errors.WithStack(err)
|
|
}
|
|
|
|
return &BlobBucket{
|
|
name: name,
|
|
id: reply.BucketID,
|
|
client: s.client,
|
|
}, nil
|
|
}
|
|
|
|
func NewBlobStore(client *rpc.Client) *BlobStore {
|
|
return &BlobStore{client}
|
|
}
|
|
|
|
var _ storage.BlobStore = &BlobStore{}
|