fix: 流式过滤重置

This commit is contained in:
tbphp
2025-06-11 15:17:29 +08:00
parent 3450a05615
commit 245fd96790

View File

@@ -218,6 +218,13 @@ func RateLimiter(config types.PerformanceConfig) gin.HandlerFunc {
// Timeout creates a timeout middleware
func Timeout(timeout time.Duration) gin.HandlerFunc {
return func(c *gin.Context) {
acceptHeader := c.Request.Header.Get("Accept")
if strings.Contains(acceptHeader, "text/event-stream") ||
strings.Contains(acceptHeader, "application/x-ndjson") ||
c.Request.Header.Get("X-Accel-Buffering") == "no" {
c.Next()
return
}
ctx, cancel := context.WithTimeout(c.Request.Context(), timeout)
defer cancel()