lzh/picoclaw
1
0
Fork 0
mirror of https://github.com/sipeed/picoclaw.git synced 2026-06-12 18:08:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

utils: make retry-after numeric clamp overflow-safe

Browse Source
This commit is contained in:
Alix-007
2026-03-30 22:02:16 +08:00
parent 9440bebca6
commit 345d4fddc9
2 changed files with 60 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/utils/http_retry.go
+14 -2
View File
@@ -60,8 +60,8 @@ func retryDelayForAttempt(resp *http.Response, attempt int) time.Duration {
return clampRetryDelay(fallback)
}
if seconds, err := strconv.Atoi(retryAfter); err == nil && seconds >= 0 {
return clampRetryDelay(time.Duration(seconds) * time.Second)
if delay, ok := numericRetryAfterDelay(retryAfter); ok {
return delay
}
if when, err := http.ParseTime(retryAfter); err == nil {
@@ -78,6 +78,18 @@ func retryDelayForAttempt(resp *http.Response, attempt int) time.Duration {
return clampRetryDelay(fallback)
}
func numericRetryAfterDelay(retryAfter string) (time.Duration, bool) {
seconds, err := strconv.ParseInt(retryAfter, 10, 64)
if err != nil || seconds < 0 {
return 0, false
}
maxSeconds := int64(maxRetrySleepDuration / time.Second)
if seconds > maxSeconds {
return maxRetrySleepDuration, true
}
return clampRetryDelay(time.Duration(seconds) * time.Second), true
}
func clampRetryDelay(delay time.Duration) time.Duration {
if delay <= 0 {
return 0