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

simplify the stream by directly popping STREAM frames from it

Browse source
This commit is contained in:
Marten Seemann 2017-12-11 18:26:27 +07:00
parent b75e1df4f0
commit 73ab97aa95
9 changed files with 224 additions and 303 deletions
Download patch file Download diff file Expand all files Collapse all files

14
crypto_stream.go
View file

@ -12,10 +12,9 @@ type cryptoStreamI interface {
io.Reader
io.Writer
AddStreamFrame(*wire.StreamFrame) error
HasDataForWriting() bool
GetDataForWriting(maxBytes protocol.ByteCount) (data []byte, shouldSendFin bool)
GetWriteOffset() protocol.ByteCount
PopStreamFrame(protocol.ByteCount) *wire.StreamFrame
Cancel(error)
HasDataForWriting() bool
SetReadOffset(protocol.ByteCount)
// methods needed for flow control
GetWindowUpdate() protocol.ByteCount
@ -27,6 +26,8 @@ type cryptoStream struct {
*stream
}
var _ cryptoStreamI = &cryptoStream{}
func newCryptoStream(onData func(), flowController flowcontrol.StreamFlowController, version protocol.VersionNumber) cryptoStreamI {
str := newStream(version.CryptoStreamID(), onData, nil, flowController, version)
return &cryptoStream{str}
@ -39,3 +40,10 @@ func (s *cryptoStream) SetReadOffset(offset protocol.ByteCount) {
s.readOffset = offset
s.frameQueue.readPosition = offset
}
func (s *cryptoStream) HasDataForWriting() bool {
s.mutex.Lock()
hasData := s.dataForWriting != nil
s.mutex.Unlock()
return hasData
}