-
Notifications
You must be signed in to change notification settings - Fork 30k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
net: persist net.Socket options before connect
Remembers net.Socket options called before connect and retroactively applies them after the handle has been created. This change makes the following function calls more user-friendly: - setKeepAlive() - setNoDelay() - ref() - unref() Related: nodejs/node-v0.x-archive#7077 and nodejs/node-v0.x-archive#8572 Fixes: nodejs/node-v0.x-archive#7077 Fixes: nodejs/node-v0.x-archive#8572 PR-URL: #1518 Reviewed-By: Ben Noordhuis <[email protected]> Reviewed-By: Roman Reiss <[email protected]>
- Loading branch information
Showing
4 changed files
with
130 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,33 @@ | ||
'use strict'; | ||
var common = require('../common'); | ||
var assert = require('assert'); | ||
var net = require('net'); | ||
|
||
var serverConnection; | ||
var echoServer = net.createServer(function(connection) { | ||
serverConnection = connection; | ||
connection.setTimeout(0); | ||
assert.equal(typeof connection.setKeepAlive, 'function'); | ||
connection.on('end', function() { | ||
connection.end(); | ||
}); | ||
}); | ||
echoServer.listen(common.PORT); | ||
|
||
echoServer.on('listening', function() { | ||
var clientConnection = new net.Socket(); | ||
// send a keepalive packet after 1000 ms | ||
// and make sure it persists | ||
clientConnection.setKeepAlive(true, 400); | ||
clientConnection.connect(common.PORT); | ||
clientConnection.setTimeout(0); | ||
|
||
setTimeout(function() { | ||
// make sure both connections are still open | ||
assert.equal(serverConnection.readyState, 'open'); | ||
assert.equal(clientConnection.readyState, 'open'); | ||
serverConnection.end(); | ||
clientConnection.end(); | ||
echoServer.close(); | ||
}, 600); | ||
}); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,33 @@ | ||
'use strict'; | ||
var common = require('../common'); | ||
var assert = require('assert'); | ||
var net = require('net'); | ||
var TCPWrap = process.binding('tcp_wrap').TCP; | ||
|
||
var echoServer = net.createServer(function(connection) { | ||
connection.end(); | ||
}); | ||
echoServer.listen(common.PORT); | ||
|
||
var callCount = 0; | ||
|
||
var Socket = net.Socket; | ||
var setNoDelay = TCPWrap.prototype.setNoDelay; | ||
|
||
TCPWrap.prototype.setNoDelay = function(enable) { | ||
setNoDelay.call(this, enable); | ||
callCount++; | ||
}; | ||
|
||
echoServer.on('listening', function() { | ||
var sock1 = new Socket(); | ||
// setNoDelay before the handle is created | ||
// there is probably a better way to test this | ||
|
||
sock1.setNoDelay(); | ||
sock1.connect(common.PORT); | ||
sock1.on('end', function() { | ||
assert.equal(callCount, 1); | ||
echoServer.close(); | ||
}); | ||
}); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,39 @@ | ||
'use strict'; | ||
var common = require('../common'); | ||
var assert = require('assert'); | ||
var net = require('net'); | ||
var TCPWrap = process.binding('tcp_wrap').TCP; | ||
|
||
var echoServer = net.createServer(function(conn) { | ||
conn.end(); | ||
}); | ||
|
||
var ref = TCPWrap.prototype.ref; | ||
var unref = TCPWrap.prototype.unref; | ||
|
||
var refCount = 0; | ||
|
||
TCPWrap.prototype.ref = function() { | ||
ref.call(this); | ||
refCount++; | ||
assert.equal(refCount, 0); | ||
}; | ||
|
||
TCPWrap.prototype.unref = function() { | ||
unref.call(this); | ||
refCount--; | ||
assert.equal(refCount, -1); | ||
}; | ||
|
||
echoServer.listen(common.PORT); | ||
|
||
echoServer.on('listening', function() { | ||
var sock = new net.Socket(); | ||
sock.unref(); | ||
sock.ref(); | ||
sock.connect(common.PORT); | ||
sock.on('end', function() { | ||
assert.equal(refCount, 0); | ||
echoServer.close(); | ||
}); | ||
}); |
85d9983
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
\o/