@@ -29,10 +29,7 @@ func RegisterSweepers() {
29
29
},
30
30
})
31
31
32
- resource .AddTestSweepers ("aws_athena_database" , & resource.Sweeper {
33
- Name : "aws_athena_database" ,
34
- F : sweepDatabases ,
35
- })
32
+ awsv2 .Register ("aws_athena_database" , sweepDatabases )
36
33
37
34
resource .AddTestSweepers ("aws_athena_workgroup" , & resource.Sweeper {
38
35
Name : "aws_athena_workgroup" ,
@@ -124,27 +121,16 @@ func sweepDataCatalogs(region string) error {
124
121
return nil
125
122
}
126
123
127
- func sweepDatabases (region string ) error {
128
- ctx := sweep .Context (region )
129
- client , err := sweep .SharedRegionalSweepClient (ctx , region )
130
- if err != nil {
131
- return fmt .Errorf ("getting client: %w" , err )
132
- }
124
+ func sweepDatabases (ctx context.Context , client * conns.AWSClient ) ([]sweep.Sweepable , error ) {
133
125
conn := client .AthenaClient (ctx )
134
- input := & athena.ListDataCatalogsInput {}
135
- sweepResources := make ([]sweep.Sweepable , 0 )
126
+ var sweepResources []sweep.Sweepable
136
127
137
- pages := athena .NewListDataCatalogsPaginator (conn , input )
128
+ input := athena.ListDataCatalogsInput {}
129
+ pages := athena .NewListDataCatalogsPaginator (conn , & input )
138
130
for pages .HasMorePages () {
139
131
page , err := pages .NextPage (ctx )
140
-
141
- if awsv2 .SkipSweepError (err ) {
142
- log .Printf ("[WARN] Skipping Athena Database sweep for %s: %s" , region , err )
143
- return nil
144
- }
145
-
146
132
if err != nil {
147
- return fmt . Errorf ( "error listing Athena Data Catalogs (%s): %w" , region , err )
133
+ return nil , err
148
134
}
149
135
150
136
for _ , v := range page .DataCatalogsSummary {
@@ -158,7 +144,10 @@ func sweepDatabases(region string) error {
158
144
page , err := pages .NextPage (ctx )
159
145
160
146
if err != nil {
161
- log .Printf ("[WARN] Skipping Athena Database sweep for Data Catalog %q in %s: %s" , catalogName , region , err )
147
+ tflog .Warn (ctx , "Skipping resource" , map [string ]any {
148
+ "error" : err .Error (),
149
+ "catalog_name" : catalogName ,
150
+ })
162
151
break
163
152
}
164
153
@@ -181,13 +170,7 @@ func sweepDatabases(region string) error {
181
170
}
182
171
}
183
172
184
- err = sweep .SweepOrchestrator (ctx , sweepResources )
185
-
186
- if err != nil {
187
- return fmt .Errorf ("error sweeping Athena Databases (%s): %w" , region , err )
188
- }
189
-
190
- return nil
173
+ return sweepResources , nil
191
174
}
192
175
193
176
func sweepWorkGroups (region string ) error {
0 commit comments