blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-12-03 02:42:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update aws.go

Browse Source
This commit is contained in:
Xinwei Xiong 2024-02-21 17:09:01 +08:00 committed by GitHub
parent f10be97001
commit f3318c6a39
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pkg/common/db/s3/aws/aws.go
View File

@ -33,9 +33,9 @@ import (
) )
const ( const (
minPartSize = 1024 * 1024 * 1 // 1MB minPartSize int64 = 1024 * 1024 * 1 // 1MB
maxPartSize = 1024 * 1024 * 1024 * 5 // 5GB maxPartSize int64 = 1024 * 1024 * 1024 * 5 // 5GB
maxNumSize = 10000 maxNumSize int64 = 10000
) )
const ( const (
@ -87,7 +87,7 @@ func (a *Aws) Engine() string {
func (a *Aws) InitiateMultipartUpload(ctx context.Context, name string) (*s3.InitiateMultipartUploadResult, error) { func (a *Aws) InitiateMultipartUpload(ctx context.Context, name string) (*s3.InitiateMultipartUploadResult, error) {
input := &sdk.CreateMultipartUploadInput{ input := &sdk.CreateMultipartUploadInput{
Bucket: aws.String(a.bucket), // 待验证是否需要 Bucket: aws.String(a.bucket), // TODO: To be verified whether it is required
Key: aws.String(name), Key: aws.String(name),
} }
result, err := a.client.CreateMultipartUploadWithContext(ctx, input) result, err := a.client.CreateMultipartUploadWithContext(ctx, input)
@ -110,7 +110,7 @@ func (a *Aws) CompleteMultipartUpload(ctx context.Context, uploadID string, name
} }
} }
input := &sdk.CompleteMultipartUploadInput{ input := &sdk.CompleteMultipartUploadInput{
Bucket: aws.String(a.bucket), // 待验证是否需要 Bucket: aws.String(a.bucket), // TODO: To be verified whether it is required
Key: aws.String(name), Key: aws.String(name),
UploadId: aws.String(uploadID), UploadId: aws.String(uploadID),
MultipartUpload: &sdk.CompletedMultipartUpload{ MultipartUpload: &sdk.CompletedMultipartUpload{
@ -134,7 +134,7 @@ func (a *Aws) PartSize(ctx context.Context, size int64) (int64, error) {
return 0, errors.New("size must be greater than 0") return 0, errors.New("size must be greater than 0")
} }
if size > maxPartSize*maxNumSize { if size > maxPartSize*maxNumSize {
return 0, fmt.Errorf("size must be less than %db", maxPartSize*maxNumSize) return 0, fmt.Errorf("AWS size must be less than the maximum allowed limit")
} }
if size <= minPartSize*maxNumSize { if size <= minPartSize*maxNumSize {
return minPartSize, nil return minPartSize, nil