修复AI分析SSE流断开:中间件透传Flusher、DB Key生效、错误提示不覆盖

This commit is contained in:
fish
2026-05-10 17:11:51 +08:00
parent 1094e82e88
commit 4cdc542291
4 changed files with 35 additions and 6 deletions

View File

@@ -39,6 +39,13 @@ func (r *statusRecorder) WriteHeader(code int) {
r.ResponseWriter.WriteHeader(code)
}
// Flush 透传 http.Flusher,避免 SSE 流式响应被中间件阻断。
func (r *statusRecorder) Flush() {
if f, ok := r.ResponseWriter.(http.Flusher); ok {
f.Flush()
}
}
func writeJSON(w http.ResponseWriter, status int, body any) {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(status)