1
0
Fork 0
mirror of https://github.com/diamondburned/arikawa.git synced 2025-12-08 04:57:19 +00:00
This commit is contained in:
Merlin 2025-09-04 03:52:52 -03:00 committed by GitHub
commit 9bd51e62ac
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -17,6 +17,10 @@ import (
// StatusTooManyRequests is the HTTP status code discord sends on rate-limiting.
const StatusTooManyRequests = 429
// StatusNirnTimeout is the HTTP status code Nirn Proxy sends when the request
// times out.
const StatusNirnTimeout = 408
// Retries is the default attempts to retry if the API returns an error before
// giving up. If the value is smaller than 1, then requests will retry forever.
var Retries uint = 5
@ -230,10 +234,34 @@ func (c *Client) request(
continue
}
if status = r.GetStatus(); status == StatusTooManyRequests || status >= 500 {
status = r.GetStatus()
if status == StatusNirnTimeout || status >= 500 {
continue
}
if status == StatusTooManyRequests {
var body = r.GetBody()
defer body.Close()
buf := bytes.Buffer{}
buf.ReadFrom(body)
var errBody struct {
RetryAfter float32 `json:"retry_after"`
}
err := json.Unmarshal(buf.Bytes(), &errBody)
if err == nil {
time.Sleep(time.Duration(errBody.RetryAfter) * time.Second)
continue
} else {
doErr = fmt.Errorf("failed to parse rate limit body: %w", err)
continue
}
}
break
}