WqyJh
3 years ago
7 changed files with 125 additions and 1 deletions
-
9server/config.yaml
-
1server/config/config.go
-
10server/config/oss_aws.go
-
1server/go.mod
-
9server/go.sum
-
94server/utils/upload/aws_s3.go
-
2server/utils/upload/upload.go
@ -0,0 +1,10 @@ |
|||
package config |
|||
|
|||
type AwsS3 struct { |
|||
Bucket string `mapstructure:"bucket" json:"bucket" yaml:"bucket"` |
|||
Region string `mapstructure:"region" json:"region" yaml:"region"` |
|||
SecretID string `mapstructure:"secret-id" json:"secretID" yaml:"secret-id"` |
|||
SecretKey string `mapstructure:"secret-key" json:"secretKey" yaml:"secret-key"` |
|||
BaseURL string `mapstructure:"base-url" json:"baseURL" yaml:"base-url"` |
|||
PathPrefix string `mapstructure:"path-prefix" json:"pathPrefix" yaml:"path-prefix"` |
|||
} |
|||
@ -0,0 +1,94 @@ |
|||
package upload |
|||
|
|||
import ( |
|||
"errors" |
|||
"fmt" |
|||
"mime/multipart" |
|||
"time" |
|||
|
|||
"github.com/flipped-aurora/gin-vue-admin/server/global" |
|||
|
|||
"github.com/aws/aws-sdk-go/aws" |
|||
"github.com/aws/aws-sdk-go/aws/credentials" |
|||
"github.com/aws/aws-sdk-go/aws/session" |
|||
"github.com/aws/aws-sdk-go/service/s3" |
|||
"github.com/aws/aws-sdk-go/service/s3/s3manager" |
|||
"go.uber.org/zap" |
|||
) |
|||
|
|||
type AwsS3 struct{} |
|||
|
|||
//@author: [WqyJh](https://github.com/WqyJh)
|
|||
//@object: *AwsS3
|
|||
//@function: UploadFile
|
|||
//@description: Upload file to Aws S3 using aws-sdk-go. See https://docs.aws.amazon.com/sdk-for-go/v1/developer-guide/s3-example-basic-bucket-operations.html#s3-examples-bucket-ops-upload-file-to-bucket
|
|||
//@param: file *multipart.FileHeader
|
|||
//@return: string, string, error
|
|||
|
|||
func (*AwsS3) UploadFile(file *multipart.FileHeader) (string, string, error) { |
|||
session := newSession() |
|||
uploader := s3manager.NewUploader(session) |
|||
|
|||
fileKey := fmt.Sprintf("%d%s", time.Now().Unix(), file.Filename) |
|||
filename := global.GVA_CONFIG.AwsS3.PathPrefix + "/" + fileKey |
|||
f, openError := file.Open() |
|||
if openError != nil { |
|||
global.GVA_LOG.Error("function file.Open() Filed", zap.Any("err", openError.Error())) |
|||
return "", "", errors.New("function file.Open() Filed, err:" + openError.Error()) |
|||
} |
|||
defer f.Close() // 创建文件 defer 关闭
|
|||
|
|||
_, err := uploader.Upload(&s3manager.UploadInput{ |
|||
Bucket: aws.String(global.GVA_CONFIG.AwsS3.Bucket), |
|||
Key: aws.String(filename), |
|||
Body: f, |
|||
}) |
|||
if err != nil { |
|||
global.GVA_LOG.Error("function uploader.Upload() Filed", zap.Any("err", err.Error())) |
|||
return "", "", err |
|||
} |
|||
|
|||
return global.GVA_CONFIG.AwsS3.BaseURL + "/" + filename, fileKey, nil |
|||
} |
|||
|
|||
//@author: [WqyJh](https://github.com/WqyJh)
|
|||
//@object: *AwsS3
|
|||
//@function: DeleteFile
|
|||
//@description: Delete file from Aws S3 using aws-sdk-go. See https://docs.aws.amazon.com/sdk-for-go/v1/developer-guide/s3-example-basic-bucket-operations.html#s3-examples-bucket-ops-delete-bucket-item
|
|||
//@param: file *multipart.FileHeader
|
|||
//@return: string, string, error
|
|||
|
|||
func (*AwsS3) DeleteFile(key string) error { |
|||
session := newSession() |
|||
svc := s3.New(session) |
|||
filename := global.GVA_CONFIG.AwsS3.PathPrefix + "/" + key |
|||
bucket := global.GVA_CONFIG.AwsS3.Bucket |
|||
|
|||
_, err := svc.DeleteObject(&s3.DeleteObjectInput{ |
|||
Bucket: aws.String(bucket), |
|||
Key: aws.String(filename), |
|||
}) |
|||
if err != nil { |
|||
global.GVA_LOG.Error("function svc.DeleteObject() Filed", zap.Any("err", err.Error())) |
|||
return errors.New("function svc.DeleteObject() Filed, err:" + err.Error()) |
|||
} |
|||
|
|||
_ = svc.WaitUntilObjectNotExists(&s3.HeadObjectInput{ |
|||
Bucket: aws.String(bucket), |
|||
Key: aws.String(filename), |
|||
}) |
|||
return nil |
|||
} |
|||
|
|||
// newSession Create S3 session
|
|||
func newSession() *session.Session { |
|||
sess, _ := session.NewSession(&aws.Config{ |
|||
Region: aws.String(global.GVA_CONFIG.AwsS3.Region), |
|||
Credentials: credentials.NewStaticCredentials( |
|||
global.GVA_CONFIG.AwsS3.SecretID, |
|||
global.GVA_CONFIG.AwsS3.SecretKey, |
|||
"", |
|||
), |
|||
}) |
|||
return sess |
|||
} |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue