mirror of
https://github.com/zeromicro/go-zero.git
synced 2026-05-13 18:00:00 +08:00
feat: support http stream response (#4055)
This commit is contained in:
@@ -5,6 +5,7 @@ import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"sync"
|
||||
|
||||
@@ -88,6 +89,26 @@ func SetOkHandler(handler func(context.Context, any) any) {
|
||||
okHandler = handler
|
||||
}
|
||||
|
||||
// Stream writes data into w with streaming mode.
|
||||
// The ctx is used to control the streaming loop, typically use r.Context().
|
||||
// The fn is called repeatedly until it returns false.
|
||||
func Stream(ctx context.Context, w http.ResponseWriter, fn func(w io.Writer) bool) {
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
default:
|
||||
hasMore := fn(w)
|
||||
if flusher, ok := w.(http.Flusher); ok {
|
||||
flusher.Flush()
|
||||
}
|
||||
if !hasMore {
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// WriteJson writes v as json string into w with code.
|
||||
func WriteJson(w http.ResponseWriter, code int, v any) {
|
||||
if err := doWriteJson(w, code, v); err != nil {
|
||||
|
||||
Reference in New Issue
Block a user