address review comments

DataDog · Jun 27, 2016 · 8a69267 · 8a69267
1 parent 9b68dca
commit 8a69267
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/checks.d/network.py b/checks.d/network.py
@@ -8,6 +8,7 @@
 # stdlib
 import re
 import socket
+from collections import defaultdict
 
 # project
 from checks import AgentCheck
@@ -558,13 +559,12 @@ def _cx_state_psutil(self):
         """
         Collect metrics about connections state using psutil
         """
-        metrics = dict.fromkeys(self.CX_STATE_GAUGE.values(), 0)
+        metrics = defaultdict(int)
         for conn in psutil.net_connections():
             protocol = self._parse_protocol_psutil(conn)
             status = self.TCP_STATES['psutil'].get(conn.status)
             metric = self.CX_STATE_GAUGE.get((protocol, status))
             if metric is None:
-                print(protocol, status)
                 self.log.warning('Metric not found for: %s,%s', protocol, status)
             else:
                 metrics[metric] += 1