110 lines
2.5 KiB
Go
110 lines
2.5 KiB
Go
package upload
|
|
|
|
import (
|
|
"context"
|
|
"mime/multipart"
|
|
"path"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
"github.com/aws/aws-sdk-go-v2/config"
|
|
"github.com/aws/aws-sdk-go-v2/credentials"
|
|
"github.com/aws/aws-sdk-go-v2/service/s3"
|
|
)
|
|
|
|
// AwsS3 AWS S3存储
|
|
type AwsS3 struct {
|
|
Bucket string
|
|
Region string
|
|
Endpoint string
|
|
SecretID string
|
|
SecretKey string
|
|
BaseURL string
|
|
PathPrefix string
|
|
S3ForcePathStyle bool
|
|
DisableSSL bool
|
|
}
|
|
|
|
// NewAwsS3 创建AWS S3
|
|
func NewAwsS3(bucket, region, endpoint, secretID, secretKey, baseURL, pathPrefix string, s3ForcePathStyle, disableSSL bool) *AwsS3 {
|
|
return &AwsS3{
|
|
Bucket: bucket,
|
|
Region: region,
|
|
Endpoint: endpoint,
|
|
SecretID: secretID,
|
|
SecretKey: secretKey,
|
|
BaseURL: baseURL,
|
|
PathPrefix: pathPrefix,
|
|
S3ForcePathStyle: s3ForcePathStyle,
|
|
DisableSSL: disableSSL,
|
|
}
|
|
}
|
|
|
|
func (a *AwsS3) getClient(ctx context.Context) (*s3.Client, error) {
|
|
cfg, err := config.LoadDefaultConfig(ctx,
|
|
config.WithRegion(a.Region),
|
|
config.WithCredentialsProvider(credentials.NewStaticCredentialsProvider(a.SecretID, a.SecretKey, "")),
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
client := s3.NewFromConfig(cfg, func(o *s3.Options) {
|
|
if a.Endpoint != "" {
|
|
o.BaseEndpoint = aws.String(a.Endpoint)
|
|
}
|
|
o.UsePathStyle = a.S3ForcePathStyle
|
|
})
|
|
return client, nil
|
|
}
|
|
|
|
// UploadFile 上传文件
|
|
func (a *AwsS3) UploadFile(file *multipart.FileHeader) (string, string, error) {
|
|
ctx := context.Background()
|
|
client, err := a.getClient(ctx)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
|
|
// 生成文件名和路径
|
|
fileName := GenerateFileName(file.Filename)
|
|
filePath := GeneratePath(a.PathPrefix)
|
|
key := path.Join(filePath, fileName)
|
|
|
|
// 打开文件
|
|
src, err := file.Open()
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
defer src.Close()
|
|
|
|
// 上传
|
|
contentType := file.Header.Get("Content-Type")
|
|
_, err = client.PutObject(ctx, &s3.PutObjectInput{
|
|
Bucket: aws.String(a.Bucket),
|
|
Key: aws.String(key),
|
|
Body: src,
|
|
ContentType: aws.String(contentType),
|
|
})
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
|
|
// 返回访问URL和key
|
|
accessURL := a.BaseURL + "/" + key
|
|
return accessURL, key, nil
|
|
}
|
|
|
|
// DeleteFile 删除文件
|
|
func (a *AwsS3) DeleteFile(key string) error {
|
|
ctx := context.Background()
|
|
client, err := a.getClient(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = client.DeleteObject(ctx, &s3.DeleteObjectInput{
|
|
Bucket: aws.String(a.Bucket),
|
|
Key: aws.String(key),
|
|
})
|
|
return err
|
|
}
|