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

make it easier to test the timer in the zeroRTTQueue

Browse source
This commit is contained in:
Marten Seemann 2020-08-28 18:03:45 +07:00
parent 7c204d4d9e
commit b21822ffc7
2 changed files with 13 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
zero_rtt_queue.go
View file

@ -15,10 +15,14 @@ type zeroRTTQueueEntry struct {
type zeroRTTQueue struct { type zeroRTTQueue struct {
mutex sync.Mutex mutex sync.Mutex
queue map[string]*zeroRTTQueueEntry queue map[string]*zeroRTTQueueEntry
queueDuration time.Duration // so we can set it in tests
} }
func newZeroRTTQueue() *zeroRTTQueue { func newZeroRTTQueue() *zeroRTTQueue {
return &zeroRTTQueue{queue: make(map[string]*zeroRTTQueueEntry)} return &zeroRTTQueue{
queue: make(map[string]*zeroRTTQueueEntry),
queueDuration: protocol.Max0RTTQueueingDuration,
}
} }
func (h *zeroRTTQueue) Enqueue(connID protocol.ConnectionID, p *receivedPacket) { func (h *zeroRTTQueue) Enqueue(connID protocol.ConnectionID, p *receivedPacket) {
@ -30,7 +34,9 @@ func (h *zeroRTTQueue) Enqueue(connID protocol.ConnectionID, p *receivedPacket)
if len(h.queue) >= protocol.Max0RTTQueues { if len(h.queue) >= protocol.Max0RTTQueues {
return return
} }
h.queue[cid] = &zeroRTTQueueEntry{timer: time.AfterFunc(protocol.Max0RTTQueueingDuration, func() { h.deleteQueue(connID) })} h.queue[cid] = &zeroRTTQueueEntry{timer: time.AfterFunc(h.queueDuration, func() {
h.deleteQueue(connID)
})}
} }
entry := h.queue[cid] entry := h.queue[cid]
if len(entry.packets) >= protocol.Max0RTTQueueLen { if len(entry.packets) >= protocol.Max0RTTQueueLen {

4
zero_rtt_queue_test.go
View file

@ -12,9 +12,11 @@ import (
var _ = Describe("0-RTT queue", func() { var _ = Describe("0-RTT queue", func() {
var q *zeroRTTQueue var q *zeroRTTQueue
queueDuration := scaleDuration(20 * time.Millisecond)
BeforeEach(func() { BeforeEach(func() {
q = newZeroRTTQueue() q = newZeroRTTQueue()
q.queueDuration = queueDuration
}) })
AfterEach(func() { AfterEach(func() {
@ -107,7 +109,7 @@ var _ = Describe("0-RTT queue", func() {
connID := protocol.ConnectionID{0xde, 0xad, 0xbe, 0xef} connID := protocol.ConnectionID{0xde, 0xad, 0xbe, 0xef}
p := &receivedPacket{data: []byte("foobar"), buffer: getPacketBuffer()} p := &receivedPacket{data: []byte("foobar"), buffer: getPacketBuffer()}
q.Enqueue(connID, p) q.Enqueue(connID, p)
time.Sleep(protocol.Max0RTTQueueingDuration * 3 / 2) time.Sleep(queueDuration * 3 / 2)
Expect(q.Dequeue(connID)).To(BeNil()) Expect(q.Dequeue(connID)).To(BeNil())
}) })
}) })