chore: switch to AWS SDK v2
This commit is contained in:
+14
-8
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user