chore cleanup

This commit is contained in:
Jason
2019-08-12 15:06:14 +08:00
parent e3b1ff47bd
commit 2254731d3b
3 changed files with 12 additions and 5 deletions

View File

@@ -233,6 +233,8 @@ func (h *udpHandler) ReceiveTo(conn core.UDPConn, data []byte, addr *net.UDPAddr
}
func (h *udpHandler) Close(conn core.UDPConn) {
conn.Close()
if remoteConn, ok := h.remoteConnMap.Load(conn); ok {
remoteConn.(net.Conn).Close()
h.remoteConnMap.Delete(conn)
@@ -243,7 +245,6 @@ func (h *udpHandler) Close(conn core.UDPConn) {
h.remotePacketConnMap.Delete(conn)
}
conn.Close()
h.remoteAddrMap.Delete(conn)
if h.sessionStater != nil {