achuprov/xk6-frostfs
1
0
Fork 0
forked from TrueCloudLab/xk6-frostfs
Code Pull requests Activity

[#4] s3: Include error message in responses

Browse source 
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
Alex Vanin 2022-05-20 12:34:34 +03:00 committed by Alex Vanin
parent 822c674c92
commit 0cafcd2375
2 changed files with 17 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
examples/s3.js
View file

@ -13,7 +13,7 @@ export const options = {
export default function () {
const key = uuidv4();
if (s3_cli.put(bucket, key, payload)) {
if (s3_cli.put(bucket, key, payload).success) {
s3_cli.get(bucket, key )
}
}

22
internal/s3/client.go
View file

@ -17,9 +17,19 @@ type (
vu modules.VU
cli *s3.Client
}
PutResponse struct {
Success bool
Error string
}
GetResponse struct {
Success bool
Error string
}
)
func (c *Client) Put(bucket, key string, payload []byte) bool {
func (c *Client) Put(bucket, key string, payload []byte) PutResponse {
rdr := bytes.NewReader(payload)
sz := rdr.Size()
@ -33,15 +43,15 @@ func (c *Client) Put(bucket, key string, payload []byte) bool {
})
if err != nil {
stats.Report(c.vu, objPutFails, 1)
return false
return PutResponse{Success: false, Error: err.Error()}
}
stats.ReportDataSent(c.vu, float64(sz))
stats.Report(c.vu, objPutDuration, metrics.D(time.Since(start)))
return true
return PutResponse{Success: true}
}
func (c *Client) Get(bucket, key string) bool {
func (c *Client) Get(bucket, key string) GetResponse {
var (
buf = make([]byte, 4*1024)
sz int
@ -54,7 +64,7 @@ func (c *Client) Get(bucket, key string) bool {
})
if err != nil {
stats.Report(c.vu, objGetFails, 1)
return false
return GetResponse{Success: false, Error: err.Error()}
}
stats.Report(c.vu, objGetDuration, metrics.D(time.Since(start)))
for {
@ -67,5 +77,5 @@ func (c *Client) Get(bucket, key string) bool {
}
}
stats.ReportDataReceived(c.vu, float64(sz))
return true
return GetResponse{Success: true}
}