Skip to content

Commit

Permalink
add sweepers for datazone (#41738)
Browse files Browse the repository at this point in the history
* add sweepers for datazone

* add copyright headers
  • Loading branch information
johnsonaj authored Mar 7, 2025
1 parent 098ce2e commit 384941c
Show file tree
Hide file tree
Showing 2 changed files with 46 additions and 0 deletions.
44 changes: 44 additions & 0 deletions internal/service/datazone/sweep.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0

package datazone

import (
"context"

"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/service/datazone"
"github.com/hashicorp/terraform-provider-aws/internal/conns"
"github.com/hashicorp/terraform-provider-aws/internal/sweep"
"github.com/hashicorp/terraform-provider-aws/internal/sweep/awsv2"
"github.com/hashicorp/terraform-provider-aws/internal/sweep/framework"
"github.com/hashicorp/terraform-provider-aws/names"
)

func RegisterSweepers() {
awsv2.Register("aws_datazone_domain", sweepDomains)
}

func sweepDomains(ctx context.Context, client *conns.AWSClient) ([]sweep.Sweepable, error) {
conn := client.DataZoneClient(ctx)
input := &datazone.ListDomainsInput{}
sweepResources := make([]sweep.Sweepable, 0)

pages := datazone.NewListDomainsPaginator(conn, input)
for pages.HasMorePages() {
page, err := pages.NextPage(ctx)
if err != nil {
return nil, err
}

for _, da := range page.Items {
id := aws.ToString(da.Id)

sweepResources = append(sweepResources, framework.NewSweepResource(newResourceDomain, client,
framework.NewAttribute(names.AttrID, id),
))
}
}

return sweepResources, nil
}
2 changes: 2 additions & 0 deletions internal/sweep/register_gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit 384941c

Please sign in to comment.