Skip to content

Commit

Permalink
pretty output msg
Browse files Browse the repository at this point in the history
  • Loading branch information
shvc committed May 10, 2023
1 parent 6d79a7d commit 380a00c
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 12 deletions.
4 changes: 2 additions & 2 deletions main_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ package main
import (
"bytes"
"log"
mand "math/rand"
mrand "math/rand"
"net/http/httptest"
"os"
"testing"
Expand All @@ -26,7 +26,7 @@ var s3cliTest = S3Cli{
}

func TestMain(m *testing.M) {
mand.Seed(time.Now().UTC().UnixNano())
mrand.Seed(time.Now().UTC().UnixNano())
// init fake s3
s3Backend = s3mem.New()
faker := gofakes3.New(s3Backend)
Expand Down
23 changes: 13 additions & 10 deletions s3cli.go
Original file line number Diff line number Diff line change
Expand Up @@ -1304,16 +1304,16 @@ func (sc *S3Cli) copyObject(ctx context.Context, source, dstBucket, dstKey, cont
// deletePrefix delete Objects with prefix
func (sc *S3Cli) deletePrefix(ctx context.Context, bucket, prefix string) error {
var objNum int64
loi := &s3.ListObjectsInput{
listObjectsInput := &s3.ListObjectsInput{
Bucket: aws.String(bucket),
Prefix: aws.String(prefix),
}
for {
req, resp := sc.Client.ListObjectsRequest(loi)
req, resp := sc.Client.ListObjectsRequest(listObjectsInput)
req.SetContext(ctx)
err := req.Send()
if err != nil {
return fmt.Errorf("list object failed: %w", err)
return fmt.Errorf("list objects failed: %w", err)
}
objectNum := len(resp.Contents)
if objectNum == 0 {
Expand All @@ -1327,14 +1327,14 @@ func (sc *S3Cli) deletePrefix(ctx context.Context, bucket, prefix string) error
for i, obj := range resp.Contents {
objects[i] = &s3.ObjectIdentifier{Key: obj.Key}
}
doi := &s3.DeleteObjectsInput{
deleteObjectsInput := &s3.DeleteObjectsInput{
Bucket: aws.String(bucket),
Delete: &s3.Delete{
Quiet: aws.Bool(true),
Objects: objects,
},
}
deleteReq, _ := sc.Client.DeleteObjectsRequest(doi)
deleteReq, _ := sc.Client.DeleteObjectsRequest(deleteObjectsInput)
if e := deleteReq.Send(); e != nil {
fmt.Printf("delete Objects failed: %s", e)
} else {
Expand All @@ -1345,13 +1345,16 @@ func (sc *S3Cli) deletePrefix(ctx context.Context, bucket, prefix string) error
}

if resp.NextMarker != nil {
loi.Marker = resp.NextMarker
} else if resp.IsTruncated != nil && *resp.IsTruncated {
loi.Marker = resp.Contents[objectNum-1].Key
} else {
break
listObjectsInput.Marker = resp.NextMarker
continue
}
if resp.IsTruncated != nil && *resp.IsTruncated {
listObjectsInput.Marker = resp.Contents[objectNum-1].Key
continue
}
break
}

return nil
}

Expand Down

0 comments on commit 380a00c

Please sign in to comment.