potyarkin/lego
1
0
Fork 0
forked from TrueCloudLab/lego
Code Pull requests Activity

constellix: follow rate limiting headers (#2085)

Browse source
This commit is contained in:
Ludovic Fernandez 2024-01-23 15:18:54 +01:00 committed by GitHub
parent 7fd1704282
commit bb830677d1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 23 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
providers/dns/constellix/constellix.go
View file

@ -6,11 +6,13 @@ import (
"errors"
"fmt"
"net/http"
"strconv"
"time"
"github.com/go-acme/lego/v4/challenge/dns01"
"github.com/go-acme/lego/v4/platform/config/env"
"github.com/go-acme/lego/v4/providers/dns/constellix/internal"
"github.com/hashicorp/go-retryablehttp"
)
// Environment variables names.
@ -85,7 +87,12 @@ func NewDNSProviderConfig(config *Config) (*DNSProvider, error) {
return nil, fmt.Errorf("constellix: %w", err)
}
client := internal.NewClient(tr.Wrap(config.HTTPClient))
retryClient := retryablehttp.NewClient()
retryClient.RetryMax = 5
retryClient.HTTPClient = tr.Wrap(config.HTTPClient)
retryClient.Backoff = backoff
client := internal.NewClient(retryClient.StandardClient())
return &DNSProvider{config: config, client: client}, nil
}
@ -273,3 +280,18 @@ func containsValue(record *internal.Record, value string) bool {
return false
}
func backoff(min, max time.Duration, attemptNum int, resp *http.Response) time.Duration {
if resp != nil {
// https://api.dns.constellix.com/v4/docs#section/Using-the-API/Rate-Limiting
if resp.StatusCode == http.StatusTooManyRequests {
if s, ok := resp.Header["X-Ratelimit-Reset"]; ok {
if sleep, err := strconv.ParseInt(s[0], 10, 64); err == nil {
return time.Second * time.Duration(sleep)
}
}
}
}
return retryablehttp.DefaultBackoff(min, max, attemptNum, resp)
}