diff options
Diffstat (limited to 'internal/storage')
-rw-r--r-- | internal/storage/local.go | 5 | ||||
-rw-r--r-- | internal/storage/s3.go | 5 | ||||
-rw-r--r-- | internal/storage/storage.go | 4 |
3 files changed, 11 insertions, 3 deletions
diff --git a/internal/storage/local.go b/internal/storage/local.go index da57631f9..8ed3ca8d8 100644 --- a/internal/storage/local.go +++ b/internal/storage/local.go @@ -33,18 +33,23 @@ type Local struct { func (l *Local) Get(ctx context.Context, key string) ([]byte, error) { return l.KVStore.Get(key) } + func (l *Local) GetStream(ctx context.Context, key string) (io.ReadCloser, error) { return l.KVStore.GetStream(key) } + func (l *Local) PutStream(ctx context.Context, key string, r io.Reader) error { return l.KVStore.PutStream(key, r) } + func (l *Local) Put(ctx context.Context, key string, value []byte) error { return l.KVStore.Put(key, value) } + func (l *Local) Delete(ctx context.Context, key string) error { return l.KVStore.Delete(key) } + func (l *Local) URL(ctx context.Context, key string) *url.URL { return nil } diff --git a/internal/storage/s3.go b/internal/storage/s3.go index a15114488..ee51bea89 100644 --- a/internal/storage/s3.go +++ b/internal/storage/s3.go @@ -55,6 +55,7 @@ func (s *S3) Get(ctx context.Context, key string) ([]byte, error) { } return b, nil } + func (s *S3) GetStream(ctx context.Context, key string) (io.ReadCloser, error) { o, err := s.mc.GetObject(ctx, s.bucket, key, minio.GetObjectOptions{}) if err != nil { @@ -62,21 +63,25 @@ func (s *S3) GetStream(ctx context.Context, key string) (io.ReadCloser, error) { } return o, err } + func (s *S3) PutStream(ctx context.Context, key string, r io.Reader) error { if _, err := s.mc.PutObject(ctx, s.bucket, key, r, -1, minio.PutObjectOptions{}); err != nil { return fmt.Errorf("uploading data stream: %w", err) } return nil } + func (s *S3) Put(ctx context.Context, key string, value []byte) error { if _, err := s.mc.PutObject(ctx, s.bucket, key, bytes.NewBuffer(value), -1, minio.PutObjectOptions{}); err != nil { return fmt.Errorf("uploading data slice: %w", err) } return nil } + func (s *S3) Delete(ctx context.Context, key string) error { return s.mc.RemoveObject(ctx, s.bucket, key, minio.RemoveObjectOptions{}) } + func (s *S3) URL(ctx context.Context, key string) *url.URL { // it's safe to ignore the error here, as we just fall back to fetching the // file if the url request fails diff --git a/internal/storage/storage.go b/internal/storage/storage.go index 88577442f..7a4ebc6c3 100644 --- a/internal/storage/storage.go +++ b/internal/storage/storage.go @@ -33,9 +33,7 @@ import ( "github.com/superseriousbusiness/gotosocial/internal/config" ) -var ( - ErrNotSupported = errors.New("driver does not suppport functionality") -) +var ErrNotSupported = errors.New("driver does not suppport functionality") // Driver implements the functionality to store and retrieve blobs // (images,video,audio) |