40 lines
860 B
Go
40 lines
860 B
Go
package storage
|
|
|
|
import (
|
|
"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"
|
|
)
|
|
|
|
type Storage interface {
|
|
Store(path string, content io.Reader) error
|
|
}
|
|
|
|
type S3 struct {
|
|
bucket string
|
|
svc s3manageriface.UploaderAPI
|
|
}
|
|
|
|
func (s *S3) Store(path string, content io.Reader) error {
|
|
_, err := s.svc.Upload(&s3manager.UploadInput{
|
|
Bucket: aws.String(s.bucket),
|
|
Key: aws.String(path),
|
|
Body: content,
|
|
})
|
|
return err
|
|
}
|
|
|
|
func New(bucket string) (Storage, error) {
|
|
sess, err := session.NewSession(aws.NewConfig())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
uploader := s3manager.NewUploader(sess, func(u *s3manager.Uploader) {
|
|
u.PartSize = 5 * 1024 * 1024
|
|
})
|
|
return &S3{bucket: bucket, svc: uploader}, nil
|
|
}
|