add garbage collector for ReceivedPacketHandler

fixes #7
This commit is contained in:
Marten Seemann
2016-05-20 17:09:47 +07:00
parent 283f75aa98
commit a1ba23aebd
2 changed files with 39 additions and 3 deletions

View File

@@ -22,9 +22,11 @@ type receivedPacketHandler struct {
highestInOrderObserved protocol.PacketNumber
highestInOrderObservedEntropy EntropyAccumulator
largestObserved protocol.PacketNumber
packetHistory map[protocol.PacketNumber]packetHistoryEntry
currentAckFrame *frames.AckFrame
stateChanged bool // has an ACK for this state already been sent? Will be set to false every time a new packet arrives, and to false every time an ACK is sent
packetHistory map[protocol.PacketNumber]packetHistoryEntry
smallestInPacketHistory protocol.PacketNumber
}
// NewReceivedPacketHandler creates a new receivedPacketHandler
@@ -59,6 +61,9 @@ func (h *receivedPacketHandler) ReceivedPacket(packetNumber protocol.PacketNumbe
EntropyBit: entropyBit,
TimeReceived: time.Now(),
}
h.garbageCollect()
return nil
}
@@ -72,6 +77,8 @@ func (h *receivedPacketHandler) ReceivedStopWaiting(f *frames.StopWaitingFrame)
h.highestInOrderObserved = f.LeastUnacked - 1
h.highestInOrderObservedEntropy = EntropyAccumulator(f.Entropy)
h.garbageCollect()
return nil
}
@@ -130,3 +137,10 @@ func (h *receivedPacketHandler) GetAckFrame(dequeue bool) (*frames.AckFrame, err
}
return h.currentAckFrame, nil
}
func (h *receivedPacketHandler) garbageCollect() {
for i := h.smallestInPacketHistory; i < h.highestInOrderObserved; i++ {
delete(h.packetHistory, i)
}
h.smallestInPacketHistory = h.highestInOrderObserved
}