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

destroy sessions when deleting them from the packet handler map

Browse source
This commit is contained in:
Marten Seemann 2019-09-24 20:32:53 +07:00
parent 03483d5e71
commit da0288b273
2 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
packet_handler_map.go
View file

@ -86,6 +86,9 @@ func (h *packetHandlerMap) Retire(id protocol.ConnectionID) {
func (h *packetHandlerMap) retireByConnectionIDAsString(id string) {
time.AfterFunc(h.deleteRetiredSessionsAfter, func() {
h.mutex.Lock()
if sess, ok := h.handlers[id]; ok {
sess.destroy(errors.New("deleting"))
}
h.removeByConnectionIDAsString(id)
h.mutex.Unlock()
})

4
packet_handler_map_test.go
View file

@ -137,7 +137,9 @@ var _ = Describe("Packet Handler Map", func() {
It("deletes retired session entries after a wait time", func() {
handler.deleteRetiredSessionsAfter = scaleDuration(10 * time.Millisecond)
connID := protocol.ConnectionID{1, 2, 3, 4, 5, 6, 7, 8}
handler.Add(connID, NewMockPacketHandler(mockCtrl))
sess := NewMockPacketHandler(mockCtrl)
handler.Add(connID, sess)
sess.EXPECT().destroy(gomock.Any())
handler.Retire(connID)
time.Sleep(scaleDuration(30 * time.Millisecond))
handler.handlePacket(nil, nil, getPacket(connID))