57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
|
package cache
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"forge.cadoles.com/arcad/edge/pkg/storage"
|
||
|
|
"github.com/hashicorp/golang-lru/v2/expirable"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
type BlobStore struct {
|
||
|
|
store storage.BlobStore
|
||
|
|
cache *expirable.LRU[string, []byte]
|
||
|
|
}
|
||
|
|
|
||
|
|
// DeleteBucket implements storage.BlobStore.
|
||
|
|
func (s *BlobStore) DeleteBucket(ctx context.Context, name string) error {
|
||
|
|
if err := s.store.DeleteBucket(ctx, name); err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// ListBuckets implements storage.BlobStore.
|
||
|
|
func (s *BlobStore) ListBuckets(ctx context.Context) ([]string, error) {
|
||
|
|
buckets, err := s.store.ListBuckets(ctx)
|
||
|
|
if err != nil {
|
||
|
|
return nil, errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return buckets, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// OpenBucket implements storage.BlobStore.
|
||
|
|
func (s *BlobStore) OpenBucket(ctx context.Context, name string) (storage.BlobBucket, error) {
|
||
|
|
bucket, err := s.store.OpenBucket(ctx, name)
|
||
|
|
if err != nil {
|
||
|
|
return nil, errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return &BlobBucket{
|
||
|
|
bucket: bucket,
|
||
|
|
cache: s.cache,
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewBlobStore(store storage.BlobStore, cacheSize int, cacheTTL time.Duration) *BlobStore {
|
||
|
|
return &BlobStore{
|
||
|
|
store: store,
|
||
|
|
cache: expirable.NewLRU[string, []byte](cacheSize, nil, cacheTTL),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ storage.BlobStore = &BlobStore{}
|