func cleanupBucket(bucket string, S3 *s3.S3) error {
req := &s3.ListObjectVersionsInput{
Bucket: aws.String(bucket),
}
res, err := S3.ListObjectVersions(req)
if err != nil {
return err
}
for _, d := range res.DeleteMarkers {
go cleanupBucketObject(bucket, *d.Key, *d.VersionId, S3)
}
for _, v := range res.Versions {
go cleanupBucketObject(bucket, *v.Key, *v.VersionId, S3)
}
return nil
}
func deleteBucket(bucket string, wg *sync.WaitGroup, S3 *s3.S3) error {
keyMarkers := []Obj{}
versionIdMarkers := []Obj{}
nextKeyMarker := aws.String("")
nextVersionIdMarker := aws.String("")
req := &s3.ListObjectVersionsInput{
Bucket: aws.String(bucket),
}
res, err := S3.ListObjectVersions(req)
if err != nil {
wg.Done()
return err
}
for _, d := range res.DeleteMarkers {
keyMarkers = append(keyMarkers, Obj{key: *d.Key, id: *d.VersionId})
}
for _, v := range res.Versions {
versionIdMarkers = append(versionIdMarkers, Obj{key: *v.Key, id: *v.VersionId})
}
nextKeyMarker = res.NextKeyMarker
nextVersionIdMarker = res.NextVersionIdMarker
for nextKeyMarker != nil && nextVersionIdMarker != nil {
req.KeyMarker = nextKeyMarker
req.VersionIdMarker = nextVersionIdMarker
res, err := S3.ListObjectVersions(req)
if err != nil {
wg.Done()
return err
}
for _, d := range res.DeleteMarkers {
keyMarkers = append(keyMarkers, Obj{key: *d.Key, id: *d.VersionId})
}
for _, v := range res.Versions {
versionIdMarkers = append(versionIdMarkers, Obj{key: *v.Key, id: *v.VersionId})
}
nextKeyMarker = res.NextKeyMarker
nextVersionIdMarker = res.NextVersionIdMarker
}
fmt.Printf("Emptying S3 Bucket %s...\n", bucket)
owg := new(sync.WaitGroup)
owg.Add(len(keyMarkers))
owg.Add(len(versionIdMarkers))
go deleteObjects(bucket, keyMarkers, owg, S3)
go deleteObjects(bucket, versionIdMarkers, owg, S3)
owg.Wait()
fmt.Printf("Deleting S3 Bucket %s...\n", bucket)
_, err = S3.DeleteBucket(&s3.DeleteBucketInput{
Bucket: aws.String(bucket),
})
if err != nil {
fmt.Printf("Failed: %s\n", err)
}
wg.Done()
return nil
}