Skip to content
This repository has been archived by the owner on Apr 22, 2023. It is now read-only.

Commit

Permalink
Browse files Browse the repository at this point in the history
Simplify writeReq handling in net_uv
  • Loading branch information
ry committed Oct 6, 2011
1 parent a50c282 commit 065c6f4
Showing 1 changed file with 6 additions and 20 deletions.
26 changes: 6 additions & 20 deletions lib/net_uv.js
Expand Up @@ -57,8 +57,7 @@ exports.connect = exports.createConnection = function(port /* [host], [cb] */) {

/* called when creating new Socket, or when re-using a closed Socket */
function initSocketHandle(self) {
self._writeRequestsHead = null;
self._writeRequestsTail = null;
self._pendingWriteReqs = 0;

self._flags = 0;
self._connectQueueSize = 0;
Expand Down Expand Up @@ -236,7 +235,7 @@ Socket.prototype.destroySoon = function() {
this.writable = false;
this._flags |= FLAG_DESTROY_SOON;

if (!this._writeRequestsHead) {
if (this._pendingWriteReqs == 0) {
this.destroy();
}
};
Expand Down Expand Up @@ -393,14 +392,7 @@ Socket.prototype.write = function(data /* [encoding], [fd], [cb] */) {

writeReq.oncomplete = afterWrite;
writeReq.cb = cb;
writeReq.next = null;

if (this._writeRequestsTail) {
this._writeRequestsTail.next = writeReq;
this._writeRequestsTail = writeReq;
} else {
this._writeRequestsHead = this._writeRequestsTail = writeReq;
}
this._pendingWriteReqs++;

return this._handle.writeQueueSize == 0;
};
Expand All @@ -415,23 +407,17 @@ function afterWrite(status, handle, req, buffer) {
}
// TODO check status.

var req_ = self._writeRequestsHead;
if (req_.next) {
self._writeRequestsHead = req_.next;
} else {
self._writeRequestsHead = self._writeRequestsTail = null;
}
assert.equal(req, req_);
self._pendingWriteReqs--;

if (!self._writeRequestsHead) {
if (self._pendingWriteReqs == 0) {
// TODO remove all uses of ondrain - this is not a good hack.
if (self.ondrain) self.ondrain();
self.emit('drain');
}

if (req.cb) req.cb();

if (!self._writeRequestsHead && self._flags & FLAG_DESTROY_SOON) {
if (self._pendingWriteReqs == 0 && self._flags & FLAG_DESTROY_SOON) {
self.destroy();
}
}
Expand Down

0 comments on commit 065c6f4

Please sign in to comment.