diff --git a/src/stats.py b/src/stats.py new file mode 100644 --- /dev/null +++ b/src/stats.py @@ -0,0 +1,28 @@ +class Stats: + received=0 + sent=0 + exchangedNodes=0 + transferredBlocks=0 + + +def logReceived(data): + Stats.received+=len(data) + + +def logSent(data): + Stats.sent+=len(data) + + +def logExchangedNode(): + Stats.exchangedNodes+=1 + + +def logTransferredBlock(): + Stats.transferredBlocks+=1 + + +def report(): + return """received {r}B +sent {s}B +exchanged {nodes} hash tree nodes +transferred {blocks} blocks""".format(r=Stats.received, s=Stats.sent, nodes=Stats.exchangedNodes, blocks=Stats.transferredBlocks)