Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

cdn: Add test sweeper and retry with backoff. #947

Merged
merged 5 commits into from
Feb 14, 2023
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
cdn: Add retry with backoff for GET requests.
  • Loading branch information
andrewsomething committed Feb 13, 2023
commit 2856d7e2fa5cccc6748193608dba27ac7bd12696
46 changes: 41 additions & 5 deletions digitalocean/cdn/resource_cdn.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,12 +3,16 @@ package cdn
import (
"context"
"log"
"net/http"
"strings"
"time"

"github.com/digitalocean/godo"
"github.com/digitalocean/terraform-provider-digitalocean/digitalocean/certificate"
"github.com/digitalocean/terraform-provider-digitalocean/digitalocean/config"
"github.com/digitalocean/terraform-provider-digitalocean/digitalocean/util"
"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/validation"
)
Expand Down Expand Up @@ -185,8 +189,7 @@ func resourceDigitalOceanCDNCreate(ctx context.Context, d *schema.ResourceData,
func resourceDigitalOceanCDNRead(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
client := meta.(*config.CombinedConfig).GodoClient()

cdn, resp, err := client.CDNs.Get(context.Background(), d.Id())

cdn, resp, err := getCDNWithRetryBackoff(ctx, client, d.Id())
if err != nil {
if resp != nil && resp.StatusCode == 404 {
log.Printf("[DEBUG] CDN (%s) was not found - removing from state", d.Id())
Expand Down Expand Up @@ -217,6 +220,39 @@ func resourceDigitalOceanCDNRead(ctx context.Context, d *schema.ResourceData, me
return nil
}

func getCDNWithRetryBackoff(ctx context.Context, client *godo.Client, id string) (*godo.CDN, *godo.Response, error) {
var (
cdn *godo.CDN
resp *godo.Response
timeout = 30 * time.Second
err error
)
err = resource.RetryContext(ctx, timeout, func() *resource.RetryError {
cdn, resp, err = client.CDNs.Get(ctx, id)
if err != nil {
if util.IsDigitalOceanError(err, http.StatusNotFound, "") {
log.Printf("[DEBUG] Received %s, retrying CDN", err.Error())
return resource.RetryableError(err)
}

if util.IsDigitalOceanError(err, http.StatusTooManyRequests, "") {
log.Printf("[DEBUG] Received %s, backing off", err.Error())
time.Sleep(10 * time.Second)
return resource.RetryableError(err)
}

return resource.NonRetryableError(err)
}

return nil
})
if err != nil {
return nil, resp, err
}

return cdn, resp, nil
}

func resourceDigitalOceanCDNUpdate(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
client := meta.(*config.CombinedConfig).GodoClient()

Expand Down Expand Up @@ -263,15 +299,15 @@ func resourceDigitalOceanCDNUpdate(ctx context.Context, d *schema.ResourceData,

func resourceDigitalOceanCDNDelete(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
client := meta.(*config.CombinedConfig).GodoClient()
resourceId := d.Id()
resourceID := d.Id()

_, err := client.CDNs.Delete(context.Background(), resourceId)
_, err := client.CDNs.Delete(context.Background(), resourceID)
if err != nil {
return diag.Errorf("Error deleting CDN: %s", err)
}

d.SetId("")
log.Printf("[INFO] CDN deleted, ID: %s", resourceId)
log.Printf("[INFO] CDN deleted, ID: %s", resourceID)

return nil
}
6 changes: 2 additions & 4 deletions digitalocean/cdn/resource_cdn_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -181,7 +181,6 @@ func testAccCheckDigitalOceanCDNExists(resource string) resource.TestCheckFunc {
}

foundCDN, _, err := client.CDNs.Get(context.Background(), rs.Primary.ID)

if err != nil {
return err
}
Expand Down Expand Up @@ -224,12 +223,11 @@ resource "digitalocean_cdn" "foobar" {
func testAccCheckDigitalOceanCDNConfig_CustomDomain(domain string, spaceName string, certName string) string {
return fmt.Sprintf(`
resource "tls_private_key" "example" {
algorithm = "ECDSA"
ecdsa_curve = "P384"
algorithm = "RSA"
}

resource "tls_self_signed_cert" "example" {
key_algorithm = "ECDSA"
key_algorithm = "RSA"
private_key_pem = tls_private_key.example.private_key_pem
dns_names = ["foo.%s"]
subject {
Expand Down