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

18
lib/spider.js
View File

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