gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

change StreamCallback to include the session

Browse Source
This commit is contained in:
Lucas Clemente
2016-04-18 14:01:35 +02:00
parent 57b144bdef
commit 07e7708506
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
example/main.go
View File

@@ -32,7 +32,7 @@ func main() {
}
}
func handleStream(stream *quic.Stream) []frames.Frame {
func handleStream(session *quic.Session, stream *quic.Stream) []frames.Frame {
var reply bytes.Buffer
h2framer := http2.NewFramer(&reply, stream)
h2framer.ReadMetaHeaders = hpack.NewDecoder(1024, nil)

4
session.go
View File

@@ -13,7 +13,7 @@ import (
)
// StreamCallback gets a stream frame and returns a reply frame
type StreamCallback func(*Stream) []frames.Frame
type StreamCallback func(*Session, *Stream) []frames.Frame
// A Session is a QUIC session
type Session struct {
@@ -156,7 +156,7 @@ func (s *Session) handleStreamFrame(r *bytes.Reader) error {
}
if !newStream {
replyFrames := s.streamCallback(stream)
replyFrames := s.streamCallback(s, stream)
if replyFrames != nil {
s.SendFrames(replyFrames)
}