SDK - add timeout from CustomiseDiff func to context (#24485)

`SDK` - add timeout from `CustomiseDiff` func to context
hashicorp · Jan 15, 2024 · f32f73c · f32f73c
2 parents d3e1ad7 + f6175c4
commit f32f73c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/internal/sdk/wrapper_resource.go b/internal/sdk/wrapper_resource.go
@@ -133,6 +133,8 @@ func (rw *ResourceWrapper) Resource() (*schema.Resource, error) {
 	if v, ok := rw.resource.(ResourceWithCustomizeDiff); ok {
 		resource.CustomizeDiff = func(ctx context.Context, d *schema.ResourceDiff, meta interface{}) error {
 			client := meta.(*clients.Client)
+			ctx, cancel := context.WithTimeout(ctx, v.CustomizeDiff().Timeout)
+			defer cancel()
 			metaData := ResourceMetaData{
 				Client:                   client,
 				Logger:                   rw.logger,