nano-run/worker/request_io.go
2020-09-17 18:48:42 +08:00

42 lines
765 B
Go

package worker
import (
"io"
"net/http"
"nano-run/services/meta"
)
func openResponse(writer io.Writer) *responseStream {
return &responseStream{stream: writer, meta: meta.AttemptHeader{Headers: make(http.Header)}}
}
type responseStream struct {
meta meta.AttemptHeader
statusSent bool
stream io.Writer
}
func (mo *responseStream) Header() http.Header {
return mo.meta.Headers
}
func (mo *responseStream) Write(bytes []byte) (int, error) {
if !mo.statusSent {
mo.WriteHeader(http.StatusOK)
}
return mo.stream.Write(bytes)
}
func (mo *responseStream) WriteHeader(statusCode int) {
if mo.statusSent {
return
}
mo.statusSent = true
mo.meta.Code = statusCode
}
func (mo *responseStream) Status(status int) {
mo.meta.Code = status
}