evan/rats-search
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

возможность установить лимит использования cpu

Browse Source
This commit is contained in:
Alexey Kasyanchuk 2017-01-17 23:53:41 +03:00
parent d1137149c1
commit 731ce1f7c8
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
lib/spider.js
View File

@ -5,6 +5,7 @@ const Emiter = require('events')
const bencode = require('bencode')
const {Table, Node} = require('./table')
const Token = require('./token')
const cpuUsage = require('./cpu-usage')
const bootstraps = [{
address: 'router.bittorrent.com',
@ -40,6 +41,8 @@ class Spider extends Emiter {
this.ignore = false; // ignore all requests
this.walkInterval = 5;
this.cpuLimit = 0;
this.cpuInterval = 10;
}
send(message, address) {
@ -68,7 +71,7 @@ class Spider extends Emiter {
walk() {
if(!this.client || this.client.isIdle()) {
if(!this.ignore)
if(!this.ignore && (this.cpuLimit <= 0 || cpuUsage() < this.cpuLimit + this.cpuInterval))
{
const node = this.table.shift()
if (node) {
@ -90,6 +93,10 @@ class Spider extends Emiter {
}
onFindNodeRequest(message, address) {
if(this.cpuLimit > 0 && cpuUsage() > this.cpuLimit) {
return
}
const {t: tid, a: {id: nid, target: infohash}} = message
if (tid === undefined || target.length != 20 || nid.length != 20) {
@ -107,6 +114,10 @@ class Spider extends Emiter {
}
onGetPeersRequest(message, address) {
if(this.cpuLimit > 0 && cpuUsage() > this.cpuLimit) {
return
}
const {t: tid, a: {id: nid, info_hash: infohash}} = message
if (tid === undefined || infohash.length != 20 || nid.length != 20) {
@ -143,9 +154,12 @@ class Spider extends Emiter {
};
this.emit('ensureHash', infohash.toString('hex').toUpperCase(), addressPair)
if(this.client && !this.ignore) {
console.log('cpu usage:' + cpuUsage())
if(this.cpuLimit <= 0 || cpuUsage() <= this.cpuLimit + this.cpuInterval) {
this.client.add(addressPair, infohash);
}
}
}
onPingRequest(message, address) {
this.send({ t: message.t, y: 'r', r: { id: Node.neighbor(message.a.id, this.table.id) } }, address)