34 lines
901 B
Go
34 lines
901 B
Go
|
package handler
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
const statusClientClosedRequest = 499
|
||
|
|
|
||
|
|
func concurrencyErrorResponse(err error, slotType string) (int, string, string) {
|
||
|
|
var waitQueueFullErr *WaitQueueFullError
|
||
|
|
if errors.As(err, &waitQueueFullErr) {
|
||
|
|
return http.StatusTooManyRequests, "rate_limit_error",
|
||
|
|
"Too many pending requests, please retry later"
|
||
|
|
}
|
||
|
|
|
||
|
|
var concurrencyErr *ConcurrencyError
|
||
|
|
if errors.As(err, &concurrencyErr) {
|
||
|
|
if concurrencyErr.SlotType != "" {
|
||
|
|
slotType = concurrencyErr.SlotType
|
||
|
|
}
|
||
|
|
return http.StatusTooManyRequests, "rate_limit_error",
|
||
|
|
fmt.Sprintf("Concurrency limit exceeded for %s, please retry later", slotType)
|
||
|
|
}
|
||
|
|
|
||
|
|
if errors.Is(err, context.Canceled) {
|
||
|
|
return statusClientClosedRequest, "api_error", "context canceled"
|
||
|
|
}
|
||
|
|
|
||
|
|
return http.StatusServiceUnavailable, "api_error", "Service temporarily unavailable, please retry later"
|
||
|
|
}
|