Skip to content

Commit

Permalink
chore: add timeout consts
Browse files Browse the repository at this point in the history
  • Loading branch information
DanielRieske committed Aug 13, 2024
1 parent ddf7702 commit 4f257d1
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 5 deletions.
12 changes: 8 additions & 4 deletions internal/service/redshiftserverless/namespace.go
Original file line number Diff line number Diff line change
Expand Up @@ -269,7 +269,7 @@ func resourceNamespaceDelete(ctx context.Context, d *schema.ResourceData, meta i
conn := meta.(*conns.AWSClient).RedshiftServerlessClient(ctx)

log.Printf("[DEBUG] Deleting Redshift Serverless Namespace: %s", d.Id())
_, err := tfresource.RetryWhenIsAErrorMessageContains[*awstypes.ConflictException](ctx, 10*time.Minute,
_, err := tfresource.RetryWhenIsAErrorMessageContains[*awstypes.ConflictException](ctx, namespaceDeletedTimeout,
func() (interface{}, error) {
return conn.DeleteNamespace(ctx, &redshiftserverless.DeleteNamespaceInput{
NamespaceName: aws.String(d.Id()),
Expand All @@ -293,6 +293,11 @@ func resourceNamespaceDelete(ctx context.Context, d *schema.ResourceData, meta i
return diags
}

const (
namespaceDeletedTimeout = 10 * time.Minute
namespaceUpdatedTimeout = 10 * time.Minute
)

func findNamespaceByName(ctx context.Context, conn *redshiftserverless.Client, name string) (*awstypes.Namespace, error) {
input := &redshiftserverless.GetNamespaceInput{
NamespaceName: aws.String(name),
Expand Down Expand Up @@ -339,7 +344,7 @@ func waitNamespaceDeleted(ctx context.Context, conn *redshiftserverless.Client,
Pending: enum.Slice(awstypes.NamespaceStatusDeleting),
Target: []string{},
Refresh: statusNamespace(ctx, conn, name),
Timeout: 10 * time.Minute,
Timeout: namespaceDeletedTimeout,
}

outputRaw, err := stateConf.WaitForStateContext(ctx)
Expand All @@ -356,7 +361,7 @@ func waitNamespaceUpdated(ctx context.Context, conn *redshiftserverless.Client,
Pending: enum.Slice(awstypes.NamespaceStatusModifying),
Target: enum.Slice(awstypes.NamespaceStatusAvailable),
Refresh: statusNamespace(ctx, conn, name),
Timeout: 10 * time.Minute,
Timeout: namespaceUpdatedTimeout,
}

outputRaw, err := stateConf.WaitForStateContext(ctx)
Expand All @@ -376,7 +381,6 @@ func flattenNamespaceIAMRoles(iamRoles []string) []string {
var tfList []string

for _, iamRole := range iamRoles {

if arn.IsARN(iamRole) {
tfList = append(tfList, iamRole)
continue
Expand Down
6 changes: 5 additions & 1 deletion internal/service/redshiftserverless/workgroup.go
Original file line number Diff line number Diff line change
Expand Up @@ -450,7 +450,7 @@ func resourceWorkgroupDelete(ctx context.Context, d *schema.ResourceData, meta i
conn := meta.(*conns.AWSClient).RedshiftServerlessClient(ctx)

log.Printf("[DEBUG] Deleting Redshift Serverless Workgroup: %s", d.Id())
_, err := tfresource.RetryWhenIsAErrorMessageContains[*awstypes.ConflictException](ctx, 10*time.Minute,
_, err := tfresource.RetryWhenIsAErrorMessageContains[*awstypes.ConflictException](ctx, workgroupDeletedTimeout,
func() (interface{}, error) {
return conn.DeleteWorkgroup(ctx, &redshiftserverless.DeleteWorkgroupInput{
WorkgroupName: aws.String(d.Id()),
Expand Down Expand Up @@ -489,6 +489,10 @@ func updateWorkgroup(ctx context.Context, conn *redshiftserverless.Client, input
return nil
}

const (
workgroupDeletedTimeout = 10 * time.Minute
)

func findWorkgroupByName(ctx context.Context, conn *redshiftserverless.Client, name string) (*awstypes.Workgroup, error) {
input := &redshiftserverless.GetWorkgroupInput{
WorkgroupName: aws.String(name),
Expand Down

0 comments on commit 4f257d1

Please sign in to comment.