-
Notifications
You must be signed in to change notification settings - Fork 109
/
request.js
79 lines (70 loc) · 2.18 KB
/
request.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
'use strict';
var http = require('http');
var https = require('https');
var parseUrl = require('url').parse;
var once = require('./once');
var joinPath = require('path').join;
var LinkedList = require('linkedlist');
function rawCopy(obj) {
var data = {};
for (var k in obj) {
if (obj.hasOwnProperty(k)) {
data[k] = obj[k];
}
}
return data;
}
module.exports = function (baseUrl, agent, agentOptions) {
var baseUrlParts = rawCopy(parseUrl(baseUrl));
var isSsl = baseUrlParts.protocol === 'https:';
if (!agent) {
agent = new (isSsl ? https : http).Agent(agentOptions);
}
var queue = new LinkedList();
var maxTasks = typeof agent.maxSockets === 'number' ? agent.maxSockets * 2 : Infinity;
var activeTasks = 0;
function drainQueue() {
if (!queue.length || activeTasks >= maxTasks) return;
var task = queue.shift();
activeTasks += 1;
task(function () {
activeTasks -= 1;
drainQueue();
});
}
return function request(_ref, cb) {
var method = _ref.method;
var url = _ref.url;
var headers = _ref.headers;
var body = _ref.body;
var path = baseUrlParts.pathname ? url.pathname ? joinPath(baseUrlParts.pathname, url.pathname) : baseUrlParts.pathname : url.pathname;
var search = url.search ? baseUrlParts.search ? baseUrlParts.search + '&' + url.search.slice(1) : url.search : baseUrlParts.search;
if (search) path += search;
var options = { path: path, method: method, headers: headers, agent: agent };
options.hostname = baseUrlParts.hostname;
options.port = baseUrlParts.port;
options.auth = baseUrlParts.auth;
queue.push(function (next) {
var callback = once(function () {
next();
cb.apply(this, arguments);
});
var req = (isSsl ? https : http).request(options, function (res) {
var data = [];
res.on('data', function (b) {
data.push(b);
}).on('end', function () {
res.body = data.join('');
callback(null, res);
});
});
req.on('error', function (err) {
err.request = req;
callback(err);
});
if (body) req.write(body);
req.end();
});
drainQueue();
};
};