chore: switch to AWS SDK v2

This commit is contained in:
2023-06-28 06:55:56 +02:00
parent d8bc211683
commit 96abde4148
5 changed files with 90 additions and 56 deletions
+14 -8
View File
@@ -1,25 +1,30 @@
package storage
import (
"context"
"io"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/s3/s3manager"
"github.com/aws/aws-sdk-go/service/s3/s3manager/s3manageriface"
"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/config"
"github.com/aws/aws-sdk-go-v2/feature/s3/manager"
"github.com/aws/aws-sdk-go-v2/service/s3"
)
type Storage interface {
Store(path string, content io.Reader) error
}
type Uploader interface {
Upload(ctx context.Context, input *s3.PutObjectInput, opts ...func(*manager.Uploader)) (*manager.UploadOutput, error)
}
type S3 struct {
bucket string
svc s3manageriface.UploaderAPI
svc Uploader
}
func (s *S3) Store(path string, content io.Reader) error {
_, err := s.svc.Upload(&s3manager.UploadInput{
_, err := s.svc.Upload(context.Background(), &s3.PutObjectInput{
Bucket: aws.String(s.bucket),
Key: aws.String(path),
Body: content,
@@ -28,11 +33,12 @@ func (s *S3) Store(path string, content io.Reader) error {
}
func New(bucket string) (Storage, error) {
sess, err := session.NewSession(aws.NewConfig())
cfg, err := config.LoadDefaultConfig(context.Background())
if err != nil {
return nil, err
}
uploader := s3manager.NewUploader(sess, func(u *s3manager.Uploader) {
client := s3.NewFromConfig(cfg)
uploader := manager.NewUploader(client, func(u *manager.Uploader) {
u.PartSize = 5 * 1024 * 1024
})
return &S3{bucket: bucket, svc: uploader}, nil