mirror of
https://github.com/gin-gonic/gin.git
synced 2025-10-16 21:32:11 +08:00
32 lines
745 B
Go
32 lines
745 B
Go
// Copyright 2018 Gin Core Team. All rights reserved.
|
|
// Use of this source code is governed by a MIT style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package render
|
|
|
|
import (
|
|
"net/http"
|
|
)
|
|
|
|
// StreamReader contains the IO reader and its length, and custom ContentType and other headers.
|
|
type StreamReader struct {
|
|
Reader
|
|
}
|
|
|
|
type writerFlusher struct {
|
|
http.ResponseWriter
|
|
}
|
|
|
|
func (w *writerFlusher) Write(buf []byte) (n int, err error) {
|
|
n, err = w.ResponseWriter.Write(buf)
|
|
if err == nil {
|
|
w.ResponseWriter.(http.Flusher).Flush()
|
|
}
|
|
return
|
|
}
|
|
|
|
// Render (StreamReader) writes data with custom ContentType and headers.
|
|
func (r StreamReader) Render(w http.ResponseWriter) (err error) {
|
|
return r.Reader.Render(&writerFlusher{w})
|
|
}
|