diff --git a/network/connection.go b/network/connection.go index dda54416..d9172150 100644 --- a/network/connection.go +++ b/network/connection.go @@ -548,10 +548,10 @@ func (conn *Connection) delete() { // A connection without an ID has been created from // a DNS request rather than a packet. Choose the correct // connection store here. - if conn.ID == "" { - dnsConns.delete(conn) - } else { + if conn.Type == IPConnection { conns.delete(conn) + } else { + dnsConns.delete(conn) } conn.Meta().Delete() diff --git a/network/metrics.go b/network/metrics.go index d8972bda..047ad19c 100644 --- a/network/metrics.go +++ b/network/metrics.go @@ -132,7 +132,7 @@ func (conn *Connection) addToMetrics() { } // Only count successful connections, not DNS requests. - if conn.ID == "" { + if conn.Type == DNSRequest { return }