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

collapse packetHandlerManager.Add and AddIfNotTaken into one function

Browse Source
This commit is contained in:
Marten Seemann
2020-01-14 18:05:23 +07:00
parent 634169b061
commit 93693349f3
7 changed files with 31 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
packet_handler_map.go
View File

@@ -95,22 +95,17 @@ func (h *packetHandlerMap) logUsage() {
}
}
func (h *packetHandlerMap) Add(id protocol.ConnectionID, handler packetHandler) {
h.mutex.Lock()
h.handlers[string(id)] = handler
h.mutex.Unlock()
}
func (h *packetHandlerMap) AddIfNotTaken(id protocol.ConnectionID, handler packetHandler) bool /* was added */ {
func (h *packetHandlerMap) Add(id protocol.ConnectionID, handler packetHandler) bool /* was added */ {
sid := string(id)
h.mutex.Lock()
defer h.mutex.Unlock()
if _, ok := h.handlers[sid]; !ok {
h.handlers[sid] = handler
return true
if _, ok := h.handlers[sid]; ok {
return false
}
return false
h.handlers[sid] = handler
return true
}
func (h *packetHandlerMap) Remove(id protocol.ConnectionID) {