add VPC endpoints restriction's FSM logic to cover the resource deletion outside terraform

Signed-off-by: Dmitry Kisler <admin@dkisler.com>
#209

Author: kislerdm

provider/resource_vpc_endpoint_restriction.go

@@ -3,8 +3,10 @@ package provider
 import (
 	"context"
 	"fmt"
+	"net/http"
 	"strings"
 
+	"github.com/hashicorp/terraform-plugin-log/tflog"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
 	neon "github.com/kislerdm/neon-sdk-go"
@@ -115,11 +117,29 @@ func resourceVPCEndpointRestrictionCreateRetry(ctx context.Context, d *schema.Re
 }
 
 func resourceVPCEndpointRestrictionReadRetry(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
-	return projectReadiness.Retry(resourceVPCEndpointRestrictionRead, ctx, d, meta)
+	return projectReadiness.RetryWithFallback(resourceVPCEndpointRestrictionRead, ctx, d, meta, map[int]FallbackFn{
+		http.StatusNotFound: func(ctx context.Context, d *schema.ResourceData, meta interface{}) error {
+			tflog.Debug(ctx, "VPC endpoint's restriction not found, removing from state",
+				map[string]interface{}{
+					"project_id":      d.Get("project_id"),
+					"vpc_endpoint_id": d.Get("vpc_endpoint_id"),
+				})
+			d.SetId("")
+			return nil
+		}})
 }
 
 func resourceVPCEndpointRestrictionDeleteRetry(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
-	return projectReadiness.Retry(resourceVPCEndpointRestrictionDelete, ctx, d, meta)
+	return projectReadiness.RetryWithFallback(resourceVPCEndpointRestrictionDelete, ctx, d, meta, map[int]FallbackFn{
+		http.StatusNotFound: func(ctx context.Context, d *schema.ResourceData, meta interface{}) error {
+			d.SetId("")
+			return nil
+		},
+		http.StatusUnprocessableEntity: func(ctx context.Context, d *schema.ResourceData, meta interface{}) error {
+			d.SetId("")
+			return nil
+		},
+	})
 }
 
 func resourceVPCEndpointRestrictionImport(ctx context.Context, d *schema.ResourceData, meta interface{}) ([]*schema.ResourceData, error) {