Skip to content

Commit 245fd96

Browse files
committed
fix: 流式过滤重置
1 parent 3450a05 commit 245fd96

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

internal/middleware/middleware.go

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -218,6 +218,13 @@ func RateLimiter(config types.PerformanceConfig) gin.HandlerFunc {
218218
// Timeout creates a timeout middleware
219219
func Timeout(timeout time.Duration) gin.HandlerFunc {
220220
return func(c *gin.Context) {
221+
acceptHeader := c.Request.Header.Get("Accept")
222+
if strings.Contains(acceptHeader, "text/event-stream") ||
223+
strings.Contains(acceptHeader, "application/x-ndjson") ||
224+
c.Request.Header.Get("X-Accel-Buffering") == "no" {
225+
c.Next()
226+
return
227+
}
221228
ctx, cancel := context.WithTimeout(c.Request.Context(), timeout)
222229
defer cancel()
223230

0 commit comments

Comments
 (0)