1berry/nodejs
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

net: Add hook for when writeQueue changes

Browse Source
This commit is contained in:
Ryan Dahl 2011-01-31 11:01:57 -08:00
parent ef123600ce
commit 9f3a20c76d
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lib/net.js
View File

@ -345,6 +345,8 @@ Socket.prototype.write = function(data /* [encoding], [fd], [cb] */) {
this._writeQueueFD.push(fd);
}
this._onBufferChange();
return false;
} else {
// Fast.
@ -417,6 +419,7 @@ Socket.prototype._writeOut = function(data, encoding, fd, cb) {
this._writeQueueEncoding.unshift(encoding);
this._writeQueueCallbacks.unshift(cb);
this._writeWatcher.start();
this._onBufferChange();
queuedData = true;
}
}
@ -465,6 +468,7 @@ Socket.prototype._writeOut = function(data, encoding, fd, cb) {
this._writeQueue.unshift(leftOver);
this._writeQueueEncoding.unshift(null);
this._writeQueueCallbacks.unshift(cb);
this._onBufferChange();
// If didn't successfully write any bytes, enqueue our fd and try again
if (!bytesWritten) {
@ -475,6 +479,12 @@ Socket.prototype._writeOut = function(data, encoding, fd, cb) {
};
Socket.prototype._onBufferChange = function() {
// Put DTrace hooks here.
;
};
// Flushes the write buffer out.
// Returns true if the entire buffer was flushed.
Socket.prototype.flush = function() {
@ -491,6 +501,7 @@ Socket.prototype.flush = function() {
// Only decrement if it's not the END_OF_FILE object...
this.bufferSize -= data.length;
this._onBufferChange();
var flushed = this._writeOut(data, encoding, fd, cb);
if (!flushed) return false;