Skip to content

Commit b674649

Browse files
committed
r/aws_transfer_connector: Add sweeper.
1 parent 189220f commit b674649

File tree

1 file changed

+25
-0
lines changed

1 file changed

+25
-0
lines changed

internal/service/transfer/sweep.go

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,36 @@ import (
1616
)
1717

1818
func RegisterSweepers() {
19+
awsv2.Register("aws_transfer_connector", sweepConnectors)
1920
awsv2.Register("aws_transfer_server", sweepServers)
2021
awsv2.Register("aws_transfer_web_app", sweepWebApps)
2122
awsv2.Register("aws_transfer_workflow", sweepWorkflows, "aws_transfer_server")
2223
}
2324

25+
func sweepConnectors(ctx context.Context, client *conns.AWSClient) ([]sweep.Sweepable, error) {
26+
conn := client.TransferClient(ctx)
27+
var input transfer.ListConnectorsInput
28+
sweepResources := make([]sweep.Sweepable, 0)
29+
30+
pages := transfer.NewListConnectorsPaginator(conn, &input)
31+
for pages.HasMorePages() {
32+
page, err := pages.NextPage(ctx)
33+
if err != nil {
34+
return nil, err
35+
}
36+
37+
for _, v := range page.Connectors {
38+
r := resourceConnector()
39+
d := r.Data(nil)
40+
d.SetId(aws.ToString(v.ConnectorId))
41+
42+
sweepResources = append(sweepResources, sweep.NewSweepResource(r, d, client))
43+
}
44+
}
45+
46+
return sweepResources, nil
47+
}
48+
2449
func sweepServers(ctx context.Context, client *conns.AWSClient) ([]sweep.Sweepable, error) {
2550
conn := client.TransferClient(ctx)
2651
var input transfer.ListServersInput

0 commit comments

Comments
 (0)