beyhan/opantoantro
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: capture client IP before stream writer to prevent race condition with fiber context

Browse Source
This commit is contained in:
Beyhan Oğur
2026-05-13 15:17:08 +03:00
parent 28eb7e849b
commit 0896410906
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
handlers/anthropic.go
+6 -3
View File
@@ -299,6 +299,9 @@ func handleStreaming(
} }
// Stream the SSE response // Stream the SSE response
// Capture values before SetBodyStreamWriter — the callback runs in a
// separate goroutine where fiber.Ctx is no longer valid.
clientIP := c.IP()
transformer := transform.NewStreamTransformer("", "") transformer := transform.NewStreamTransformer("", "")
c.Set("Content-Type", "text/event-stream") c.Set("Content-Type", "text/event-stream")
c.Set("Cache-Control", "no-cache") c.Set("Cache-Control", "no-cache")
@@ -320,17 +323,17 @@ func handleStreaming(
event := transformer.TransformChunk(chunk) event := transformer.TransformChunk(chunk)
if event != "" { if event != "" {
if _, err := w.WriteString(event); err != nil { if _, err := w.WriteString(event); err != nil {
logger.Warn("[ANTHROPIC] Stream write error [IP: %s]: %v", c.IP(), err) logger.Warn("[ANTHROPIC] Stream write error [IP: %s]: %v", clientIP, err)
return return
} }
if err := w.Flush(); err != nil { if err := w.Flush(); err != nil {
logger.Warn("[ANTHROPIC] Stream flush error [IP: %s]: %v", c.IP(), err) logger.Warn("[ANTHROPIC] Stream flush error [IP: %s]: %v", clientIP, err)
return return
} }
} }
} }
if err := scanner.Err(); err != nil { if err := scanner.Err(); err != nil {
logger.Warn("[ANTHROPIC] Stream scan error [IP: %s]: %v", c.IP(), err) logger.Warn("[ANTHROPIC] Stream scan error [IP: %s]: %v", clientIP, err)
return return
} }
// Ensure final events are sent // Ensure final events are sent