-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathindex.js
94 lines (78 loc) · 2.61 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
var Request = require('request');
var Q = require('q');
var ValidUrl = require('valid-url');
var TokenPoolRequest = function(tokenPoolConfig) {
var self = this;
self.Request = Request;
if(tokenPoolConfig.tokens == null){
throw new Error("No tokens found");
}
self.tokenPoolConfig = tokenPoolConfig;
self.tokenPoolConfig['currToken'] = 0; //Start by the first token
if(self.tokenPoolConfig['resetOffset'] == null){
self.tokenPoolConfig['resetOffset'] = 5; //Set a default to 5 sec
}
};
TokenPoolRequest.prototype.setTokenPoolConfig = function(tokenPoolConfig){
var self = this;
self.tokenPoolConfig = tokenPoolConfig;
};
TokenPoolRequest.prototype.getTokenPoolConfig = function(){
var self = this;
return self.tokenPoolConfig;
};
// Distribute tokens from the supplied pool in a round robin pattern
TokenPoolRequest.prototype.getTokenFromPool = function(){
var self = this;
var i = self.tokenPoolConfig.currToken;
if(i == (self.tokenPoolConfig.tokens.length - 1)){
self.tokenPoolConfig.currToken = 0;
}else{
self.tokenPoolConfig.currToken++;
}
return self.tokenPoolConfig.tokens[i];
}
TokenPoolRequest.prototype.authenticatedRequest = function(options, type){
var deferred = Q.defer();
var self = this;
var token = self.getTokenFromPool();
var resets = [];
self.insertTokenInOptions(options, token);
self.makeRequest(options, deferred, type);
return deferred.promise;
}
// Recallable wrapper function for the web request
TokenPoolRequest.prototype.makeRequest = function(options, deferred, type){
var self = this;
Request(options, function (error, response, body){
if(error){
deferred.reject(new Error(error));
}else{
var limitReset = response.headers[self.tokenPoolConfig['headerLimitResetVar']];
var currEpoch = Math.floor((new Date).getTime()/1000);
var waitPeriod = limitReset - currEpoch + self.tokenPoolConfig.resetOffset;
if(response.statusCode == self.tokenPoolConfig.failureResponseCode
&& type == 'WTC'){
console.log("Blocked for " + waitPeriod + " seconds");
Q.delay(waitPeriod * 1000)
.then(function(){
self.makeRequest(options, deferred, type)
});
}else{
deferred.resolve(response);
}
}
});
}
TokenPoolRequest.prototype.insertTokenInOptions = function(options, token){
var self = this;
if(self.tokenPoolConfig.tokenLocation == "header"){
options.headers['Authorization'] = self.tokenPoolConfig.tokenIdentifier + " " + token;
}else if(self.tokenPoolConfig.tokenLocation == "query"){
if(options.qs == null){
options.qs = {};
}
options.qs[self.tokenPoolConfig.tokenIdentifier] = token;
}
}
module.exports = TokenPoolRequest;