mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-03 20:27:35 +03:00
d9c107b034
The queue holds all streams that have data to send. When stream.Write or stream.Close are called, a stream is added to this queue. It is removed from the queue when all available (at that moment) data was sent. This way, we don't need the round robin scheduling (which, for every packet sent, asked every single open stream if it had data) any more.
49 lines
1.4 KiB
Go
49 lines
1.4 KiB
Go
package quic
|
|
|
|
import (
|
|
"io"
|
|
|
|
"github.com/lucas-clemente/quic-go/internal/flowcontrol"
|
|
"github.com/lucas-clemente/quic-go/internal/protocol"
|
|
"github.com/lucas-clemente/quic-go/internal/wire"
|
|
)
|
|
|
|
type cryptoStreamI interface {
|
|
StreamID() protocol.StreamID
|
|
io.Reader
|
|
io.Writer
|
|
handleStreamFrame(*wire.StreamFrame) error
|
|
popStreamFrame(protocol.ByteCount) (*wire.StreamFrame, bool)
|
|
closeForShutdown(error)
|
|
hasDataForWriting() bool
|
|
setReadOffset(protocol.ByteCount)
|
|
// methods needed for flow control
|
|
getWindowUpdate() protocol.ByteCount
|
|
handleMaxStreamDataFrame(*wire.MaxStreamDataFrame)
|
|
}
|
|
|
|
type cryptoStream struct {
|
|
*stream
|
|
}
|
|
|
|
var _ cryptoStreamI = &cryptoStream{}
|
|
|
|
func newCryptoStream(sender streamSender, flowController flowcontrol.StreamFlowController, version protocol.VersionNumber) cryptoStreamI {
|
|
str := newStream(version.CryptoStreamID(), sender, flowController, version)
|
|
return &cryptoStream{str}
|
|
}
|
|
|
|
// SetReadOffset sets the read offset.
|
|
// It is only needed for the crypto stream.
|
|
// It must not be called concurrently with any other stream methods, especially Read and Write.
|
|
func (s *cryptoStream) setReadOffset(offset protocol.ByteCount) {
|
|
s.receiveStream.readOffset = offset
|
|
s.receiveStream.frameQueue.readPosition = offset
|
|
}
|
|
|
|
func (s *cryptoStream) hasDataForWriting() bool {
|
|
s.sendStream.mutex.Lock()
|
|
hasData := s.sendStream.dataForWriting != nil
|
|
s.sendStream.mutex.Unlock()
|
|
return hasData
|
|
}
|