aboutsummaryrefslogtreecommitdiff
path: root/node_modules/agentkeepalive/lib/agent.js
blob: ad4917f0ae08daa163a82710b71ab6954370facf (plain)
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
95
96
97
/**
 * refer:
 *   * @atimb "Real keep-alive HTTP agent": https://gist.github.com/2963672
 *   * https://github.com/joyent/node/blob/master/lib/http.js
 *   * https://github.com/joyent/node/blob/master/lib/https.js
 *   * https://github.com/joyent/node/blob/master/lib/_http_agent.js
 *
 * Copyright(c) 2012 - 2014 fengmk2 <fengmk2@gmail.com>
 * Copyright(c) node-modules
 * MIT Licensed
 */

'use strict';

/**
 * Module dependencies.
 */

var https = require('https');
var utils = require('./utils');
var OriginalAgent = require('./_http_agent').Agent;
var OriginalHttpsAgent = https.Agent;

module.exports = Agent;

function Agent(options) {
  if (!(this instanceof Agent)) {
    return new Agent(options);
  }

  options = options || {};
  options.keepAlive = options.keepAlive !== false;
  // default is keep-alive and 15s free socket timeout
  if (options.keepAliveTimeout === undefined) {
    options.keepAliveTimeout = 15000;
  }
  // default timeout is double keepalive timeout
  if (options.timeout === undefined) {
    options.timeout = options.keepAliveTimeout * 2;
  }

  OriginalAgent.call(this, options);

  var self = this;
  self.createSocketCount = 0;
  self.closeSocketCount = 0;
  // socket error event count
  self.errorSocketCount = 0;
  self.requestCount = 0;
  self.timeoutSocketCount = 0;
  self.on('free', function () {
    self.requestCount++;
  });
  self.on('timeout', function () {
    self.timeoutSocketCount++;
  });
  self.on('close', function () {
    self.closeSocketCount++;
  });
  self.on('error', function () {
    self.errorSocketCount++;
  });
}

utils.inherits(Agent, OriginalAgent);

Agent.prototype.createSocket = function (req, options) {
  var socket = OriginalAgent.prototype.createSocket.call(this, req, options);
  if (this.keepAlive) {
    // Disable Nagle's algorithm: http://blog.caustik.com/2012/04/08/scaling-node-js-to-100k-concurrent-connections/
    // http://fengmk2.com/benchmark/nagle-algorithm-delayed-ack-mock.html
    socket.setNoDelay(true);
  }
  this.createSocketCount++;
  return socket;
};

Agent.prototype.getCurrentStatus = function () {
  return {
    createSocketCount: this.createSocketCount,
    closeSocketCount: this.closeSocketCount,
    errorSocketCount: this.errorSocketCount,
    timeoutSocketCount: this.timeoutSocketCount,
    requestCount: this.requestCount,
    freeSockets: inspect(this.freeSockets),
    sockets: inspect(this.sockets),
    requests: inspect(this.requests)
  };
};

function inspect(obj) {
  var res = {};
  for (var key in obj) {
    res[key] = obj[key].length;
  }
  return res;
}