mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-05 21:27:35 +03:00
Code Issues Projects Releases Packages Wiki Activity

only write response header once

Browse source
This commit is contained in:
Lucas Clemente 2016-04-19 13:55:31 +02:00
parent 4e1942a76e
commit 999b4e7b0a
1 changed files with 14 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

25
example/main.go
View file

@ -38,11 +38,13 @@ func main() {
}
type responseWriter struct {
header http.Header
headerStream *quic.Stream
session *quic.Session
status int
dataStreamID protocol.StreamID
headerStream *quic.Stream
dataStream *quic.Stream
header http.Header
headerWritten bool
}
func (w *responseWriter) Header() http.Header {
@ -50,17 +52,11 @@ func (w *responseWriter) Header() http.Header {
}
func (w *responseWriter) WriteHeader(status int) {
w.status = status
}
func (w *responseWriter) Write(p []byte) (int, error) {
if w.status == 0 {
w.WriteHeader(200)
}
w.headerWritten = true
var headers bytes.Buffer
enc := hpack.NewEncoder(&headers)
enc.WriteField(hpack.HeaderField{Name: ":status", Value: strconv.Itoa(w.status)})
enc.WriteField(hpack.HeaderField{Name: ":status", Value: strconv.Itoa(status)})
// enc.WriteField(hpack.HeaderField{Name: "content-length", Value: strconv.Itoa(len(p))})
// enc.WriteField(hpack.HeaderField{Name: "content-type", Value: http.DetectContentType(p)})
@ -68,12 +64,19 @@ func (w *responseWriter) Write(p []byte) (int, error) {
enc.WriteField(hpack.HeaderField{Name: k, Value: v[0]})
}
fmt.Printf("responding with %d %#v\n", status, w.header)
h2framer := http2.NewFramer(w.headerStream, nil)
h2framer.WriteHeaders(http2.HeadersFrameParam{
StreamID: uint32(w.dataStreamID),
EndHeaders: true,
BlockFragment: headers.Bytes(),
})
}
func (w *responseWriter) Write(p []byte) (int, error) {
if !w.headerWritten {
w.WriteHeader(200)
}
if len(p) != 0 {
dataStream, err := w.session.NewStream(w.dataStreamID)