diff --git a/src/Connection/Connection.py b/src/Connection/Connection.py index a0754d9f..17865946 100644 --- a/src/Connection/Connection.py +++ b/src/Connection/Connection.py @@ -344,6 +344,7 @@ class Connection(object): else: self.log("Unknown response: %s" % message) elif cmd: + self.server.num_recv += 1 if cmd == "handshake": self.handleHandshake(message) else: @@ -397,6 +398,8 @@ class Connection(object): stat_key = message.get("cmd", "unknown") if stat_key == "response": stat_key = "response: %s" % self.last_cmd_recv + else: + self.server.num_sent += 1 self.server.stat_sent[stat_key]["num"] += 1 if streaming: diff --git a/src/Connection/ConnectionServer.py b/src/Connection/ConnectionServer.py index fe81a5a9..023241d5 100644 --- a/src/Connection/ConnectionServer.py +++ b/src/Connection/ConnectionServer.py @@ -44,6 +44,8 @@ class ConnectionServer: self.stat_sent = defaultdict(lambda: defaultdict(int)) self.bytes_recv = 0 self.bytes_sent = 0 + self.num_recv = 0 + self.num_sent = 0 # Bittorrent style peerid self.peer_id = "-ZN0%s-%s" % (config.version.replace(".", ""), CryptHash.random(12, "base64"))