From 5b59c2c2b0ccf80c096592c9ead6f5697586a6c3 Mon Sep 17 00:00:00 2001
From: Lucas Clemente <lucas@clemente.io>
Date: Mon, 18 Apr 2016 14:54:02 +0200
Subject: [PATCH] serve multiple http requests per connection

---
 example/main.go | 56 +++++++++++++++++++++++++++----------------------
 1 file changed, 31 insertions(+), 25 deletions(-)

diff --git a/example/main.go b/example/main.go
index ac9d5f56..a3c6970c 100644
--- a/example/main.go
+++ b/example/main.go
@@ -32,33 +32,39 @@ func main() {
 }
 
 func handleStream(session *quic.Session, stream *quic.Stream) {
+	hpackDecoder := hpack.NewDecoder(1024, nil)
 	h2framer := http2.NewFramer(stream, stream)
-	h2framer.ReadMetaHeaders = hpack.NewDecoder(1024, nil)
-	h2frame, err := h2framer.ReadFrame()
-	if err != nil {
-		fmt.Printf("invalid http2 frame: %s", err.Error())
-		return
-	}
-	h2headersFrame := h2frame.(*http2.MetaHeadersFrame)
-	fmt.Printf("Request: %s %s://%s%s\n", h2headersFrame.PseudoValue("method"), h2headersFrame.PseudoValue("scheme"), h2headersFrame.PseudoValue("authority"), h2headersFrame.PseudoValue("path"))
+	h2framer.ReadMetaHeaders = hpackDecoder
 
-	var replyHeaders bytes.Buffer
-	enc := hpack.NewEncoder(&replyHeaders)
-	enc.WriteField(hpack.HeaderField{Name: ":status", Value: "200"})
-	enc.WriteField(hpack.HeaderField{Name: "content-type", Value: "text/plain"})
-	enc.WriteField(hpack.HeaderField{Name: "content-length", Value: "12"})
-	h2framer.WriteHeaders(http2.HeadersFrameParam{
-		StreamID:      h2frame.Header().StreamID,
-		EndHeaders:    true,
-		BlockFragment: replyHeaders.Bytes(),
-	})
+	go func() {
+		for {
+			h2frame, err := h2framer.ReadFrame()
+			if err != nil {
+				fmt.Printf("invalid http2 frame: %s", err.Error())
+				return
+			}
+			h2headersFrame := h2frame.(*http2.MetaHeadersFrame)
+			fmt.Printf("Request: %s %s://%s%s\n", h2headersFrame.PseudoValue("method"), h2headersFrame.PseudoValue("scheme"), h2headersFrame.PseudoValue("authority"), h2headersFrame.PseudoValue("path"))
 
-	dataStream, err := session.NewStream(protocol.StreamID(h2frame.Header().StreamID))
-	if err != nil {
-		fmt.Printf("error creating stream: %s", err.Error())
-		return
-	}
+			var replyHeaders bytes.Buffer
+			enc := hpack.NewEncoder(&replyHeaders)
+			enc.WriteField(hpack.HeaderField{Name: ":status", Value: "200"})
+			enc.WriteField(hpack.HeaderField{Name: "content-type", Value: "text/plain"})
+			enc.WriteField(hpack.HeaderField{Name: "content-length", Value: "12"})
+			h2framer.WriteHeaders(http2.HeadersFrameParam{
+				StreamID:      h2frame.Header().StreamID,
+				EndHeaders:    true,
+				BlockFragment: replyHeaders.Bytes(),
+			})
 
-	dataStream.Write([]byte("Hello World!"))
-	dataStream.Close()
+			dataStream, err := session.NewStream(protocol.StreamID(h2frame.Header().StreamID))
+			if err != nil {
+				fmt.Printf("error creating stream: %s", err.Error())
+				return
+			}
+
+			dataStream.Write([]byte("Hello World!"))
+			dataStream.Close()
+		}
+	}()
 }