diff --git a/bt/client.js b/bt/client.js index f2904b6..8310d01 100644 --- a/bt/client.js +++ b/bt/client.js @@ -71,6 +71,10 @@ class Client extends Emiter socket.destroy(); }); + socket.on('data', (data) => { + this.emit('traffic', data.length); + }) + socket.on('timeout', (err) => { socket.destroy(); }); diff --git a/bt/spider.js b/bt/spider.js index 9fbdf37..a456a5c 100644 --- a/bt/spider.js +++ b/bt/spider.js @@ -44,6 +44,7 @@ class Spider extends Emiter { this.client = client this.ignore = false; // ignore all requests this.initialized = false; + this.traffic = 0; this.walkInterval = config.spider.walkInterval; this.cpuLimit = config.spider.cpuLimit; @@ -205,6 +206,7 @@ class Spider extends Emiter { }) this.udp.on('message', (data, addr) => { this.parse(data, addr) + this.traffic += data.length }) this.udp.on('error', (err) => {}) setInterval(() => { @@ -214,6 +216,14 @@ class Spider extends Emiter { }, 3000) this.join() this.walk() + setInterval(() => { + console.log(this.traffic / 1024, 'kb/s') + this.traffic = 0 + }, 1000) + if(this.client) + this.client.on('traffic', (traffic) => { + this.traffic += traffic + }) } }