diff --git a/internal/s3/client.go b/internal/s3/client.go index b4824b0..a503f45 100644 --- a/internal/s3/client.go +++ b/internal/s3/client.go @@ -142,6 +142,68 @@ func (c *Client) Get(bucket, key string) GetResponse { return GetResponse{Success: true} } +func (c *Client) DeleteObjectVersion(bucket, key, version string) DeleteResponse { + var toDelete []types.ObjectIdentifier + + if version != "" { + toDelete = append(toDelete, types.ObjectIdentifier{ + Key: aws.String(key), + VersionId: aws.String(version), + }) + } else { + versions, err := c.cli.ListObjectVersions(c.vu.Context(), &s3.ListObjectVersionsInput{ + Bucket: aws.String(bucket), + Prefix: aws.String(key), + }) + if err != nil { + stats.Report(c.vu, objDeleteFails, 1) + return DeleteResponse{Success: false, Error: err.Error()} + } + toDelete = filterObjectVersions(versions, key) + } + if len(toDelete) == 0 { + return c.Delete(bucket, key) + } else { + _, err := c.cli.DeleteObjects(c.vu.Context(), &s3.DeleteObjectsInput{ + Bucket: aws.String(bucket), + Delete: &types.Delete{ + Objects: toDelete, + Quiet: true, + }, + }) + if err != nil { + stats.Report(c.vu, objDeleteFails, 1) + return DeleteResponse{Success: false, Error: err.Error()} + } + } + + return DeleteResponse{Success: true} +} + +func filterObjectVersions(versions *s3.ListObjectVersionsOutput, key string) []types.ObjectIdentifier { + var result []types.ObjectIdentifier + + for _, v := range versions.Versions { + if *v.Key == key { + result = append(result, types.ObjectIdentifier{ + Key: v.Key, + VersionId: v.VersionId, + }) + } + } + + for _, marker := range versions.DeleteMarkers { + if *marker.Key == key { + result = append(result, types.ObjectIdentifier{ + Key: marker.Key, + VersionId: marker.VersionId, + }) + } + } + + return result +} + func get( c *s3.Client, bucket string,