apps/engine
1
0
Fork 0
mirror of https://github.com/Monibuca/engine.git synced 2025-10-11 19:40:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: speed up sse output

Browse Source
This commit is contained in:
langhuihui
2023-08-21 14:50:56 +08:00
parent 3fac2cd570
commit 3d14717f69
1 changed files with 5 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
util/socket.go
View File

@@ -96,17 +96,11 @@ func ReturnFetchValue[T any](fetch func() T, rw http.ResponseWriter, r *http.Req
sse := NewSSE(rw, r.Context()) sse := NewSSE(rw, r.Context())
tick := time.NewTicker(tickDur) tick := time.NewTicker(tickDur)
defer tick.Stop() defer tick.Stop()
if isYaml { writer := Conditoinal(isYaml, sse.WriteYAML, sse.WriteJSON)
for range tick.C { writer(fetch())
if sse.WriteYAML(fetch()) != nil { for range tick.C {
return if writer(fetch()) != nil {
} return
}
} else {
for range tick.C {
if sse.WriteJSON(fetch()) != nil {
return
}
} }
} }
} else { } else {