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

add a function to tell if there's app data in the retransmission queue

Browse Source
This commit is contained in:
Marten Seemann
2020-05-13 11:58:24 +07:00
parent 77b1021a54
commit df55c30c86
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
retransmission_queue.go
View File

@@ -47,6 +47,10 @@ func (q *retransmissionQueue) HasHandshakeData() bool {
return len(q.handshakeCryptoData) > 0 || len(q.handshake) > 0 return len(q.handshakeCryptoData) > 0 || len(q.handshake) > 0
} }
func (q *retransmissionQueue) HasAppData() bool {
return len(q.appData) > 0
}
func (q *retransmissionQueue) AddAppData(f wire.Frame) { func (q *retransmissionQueue) AddAppData(f wire.Frame) {
if _, ok := f.(*wire.StreamFrame); ok { if _, ok := f.(*wire.StreamFrame); ok {
panic("STREAM frames are handled with their respective streams.") panic("STREAM frames are handled with their respective streams.")

3
retransmission_queue_test.go
View File

@@ -176,9 +176,12 @@ var _ = Describe("Retransmission queue", func() {
It("queues and retrieves a control frame", func() { It("queues and retrieves a control frame", func() {
f := &wire.MaxDataFrame{ByteOffset: 0x42} f := &wire.MaxDataFrame{ByteOffset: 0x42}
Expect(q.HasAppData()).To(BeFalse())
q.AddAppData(f) q.AddAppData(f)
Expect(q.HasAppData()).To(BeTrue())
Expect(q.GetAppDataFrame(f.Length(version) - 1)).To(BeNil()) Expect(q.GetAppDataFrame(f.Length(version) - 1)).To(BeNil())
Expect(q.GetAppDataFrame(f.Length(version))).To(Equal(f)) Expect(q.GetAppDataFrame(f.Length(version))).To(Equal(f))
Expect(q.HasAppData()).To(BeFalse())
}) })
}) })
}) })