diff --git a/h2quic/server.go b/h2quic/server.go index cbb1eb86c..445b8422b 100644 --- a/h2quic/server.go +++ b/h2quic/server.go @@ -20,6 +20,7 @@ import ( ) type streamCreator interface { + AcceptStream() (utils.Stream, error) GetOrOpenStream(protocol.StreamID) (utils.Stream, error) Close(error) error RemoteAddr() net.Addr @@ -91,7 +92,7 @@ func (s *Server) serveImpl(tlsConfig *tls.Config, conn *net.UDPConn) error { } func (s *Server) handleStreamCb(session quic.Session, stream utils.Stream) { - s.handleStream(session, stream) + s.handleStream(session.(streamCreator), stream) } func (s *Server) handleStream(session streamCreator, stream utils.Stream) { diff --git a/h2quic/server_test.go b/h2quic/server_test.go index cf0bb5d0d..efdc57b2e 100644 --- a/h2quic/server_test.go +++ b/h2quic/server_test.go @@ -31,6 +31,9 @@ type mockSession struct { func (s *mockSession) GetOrOpenStream(id protocol.StreamID) (utils.Stream, error) { return s.dataStream, nil } +func (s *mockSession) AcceptStream() (utils.Stream, error) { + panic("not implemented") +} func (s *mockSession) Close(e error) error { s.closed = true s.closedWithError = e diff --git a/interface.go b/interface.go index 918934c9e..18aa14acb 100644 --- a/interface.go +++ b/interface.go @@ -3,7 +3,6 @@ package quic import ( "net" - "github.com/lucas-clemente/quic-go/protocol" "github.com/lucas-clemente/quic-go/utils" ) @@ -20,6 +19,4 @@ type Session interface { // OpenStreamSync() (utils.Stream, error) RemoteAddr() net.Addr Close(error) error - // TODO: remove this - GetOrOpenStream(protocol.StreamID) (utils.Stream, error) }