TrueCloudLab/xk6-frostfs
42
1
Fork 20
Code Issues 32 Pull requests 3 Releases Activity Actions

Fix

Browse source 
Signed-off-by: Мария Малыгина <m.malygina@yadro.com>
This commit is contained in:
Мария Малыгина 2023-07-17 16:33:51 +03:00
parent 88ae8f47a3
commit 6b7a27df4f
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
internal/s3/client.go
View file

@ -5,6 +5,7 @@ import (
"context" "context"
"crypto/sha256" "crypto/sha256"
"encoding/hex" "encoding/hex"
"fmt"
"strconv" "strconv"
"time" "time"
@ -71,6 +72,8 @@ func (c *Client) Put(bucket, key string, payload goja.ArrayBuffer) PutResponse {
return PutResponse{Success: true} return PutResponse{Success: true}
} }
const multipartUploadMinPartSize = 5 * 1024 * 1024 // 5MB
func (c *Client) Multipart(bucket, key string, objPartSize int, payload goja.ArrayBuffer, concurrency int) PutResponse { func (c *Client) Multipart(bucket, key string, objPartSize int, payload goja.ArrayBuffer, concurrency int) PutResponse {
if objPartSize < multipartUploadMinPartSize { if objPartSize < multipartUploadMinPartSize {
stats.Report(c.vu, objPutFails, 1) stats.Report(c.vu, objPutFails, 1)
@ -78,10 +81,10 @@ func (c *Client) Multipart(bucket, key string, objPartSize int, payload goja.Arr
} }
start := time.Now() start := time.Now()
uploader := manager.NewUploader(c.cli, func(u *manager.Uploader) { uploader := manager.NewUploader(c.cli, func(u *manager.Uploader) {
u.PartSize = objPartSize u.PartSize = int64(objPartSize)
uploader.Concurrency = concurrency u.Concurrency = concurrency
}) })
result, err := uploader.Upload(context.TODO(), &s3.PutObjectInput{ _, err := uploader.Upload(context.TODO(), &s3.PutObjectInput{
Bucket: aws.String(bucket), Bucket: aws.String(bucket),
Key: aws.String(key), Key: aws.String(key),
Body: bytes.NewReader(payload.Bytes()), Body: bytes.NewReader(payload.Bytes()),