-
Notifications
You must be signed in to change notification settings - Fork 29.6k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
streams: introduce StreamWrap and JSStream
Introduce a way to wrap plain-js `stream.Duplex` streams into C++ StreamBase's child class. With such method at hand it is now possible to pass `stream.Duplex` instance as a `socket` parameter to `tls.connect()`. PR-URL: #926 Reviewed-By: Chris Dickinson <[email protected]>
- Loading branch information
Showing
15 changed files
with
498 additions
and
52 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,118 @@ | ||
| const util = require('util'); | ||
| const Socket = require('net').Socket; | ||
| const JSStream = process.binding('js_stream').JSStream; | ||
| const uv = process.binding('uv'); | ||
|
|
||
| function StreamWrap(stream) { | ||
| var handle = new JSStream(); | ||
|
|
||
| this.stream = stream; | ||
|
|
||
| var self = this; | ||
| handle.close = function(cb) { | ||
| cb(); | ||
| }; | ||
| handle.isAlive = function() { | ||
| return self.isAlive(); | ||
| }; | ||
| handle.isClosing = function() { | ||
| return self.isClosing(); | ||
| }; | ||
| handle.onreadstart = function() { | ||
| return self.readStart(); | ||
| }; | ||
| handle.onreadstop = function() { | ||
| return self.readStop(); | ||
| }; | ||
| handle.onshutdown = function(req) { | ||
| return self.shutdown(req); | ||
| }; | ||
| handle.onwrite = function(req, bufs) { | ||
| return self.write(req, bufs); | ||
| }; | ||
|
|
||
| this.stream.pause(); | ||
| this.stream.on('data', function(chunk) { | ||
| self._handle.readBuffer(chunk); | ||
| }); | ||
| this.stream.once('end', function() { | ||
| self._handle.emitEOF(); | ||
| }); | ||
| this.stream.on('error', function(err) { | ||
| self.emit('error', err); | ||
| }); | ||
|
|
||
| Socket.call(this, { | ||
| handle: handle | ||
| }); | ||
| } | ||
| util.inherits(StreamWrap, Socket); | ||
| module.exports = StreamWrap; | ||
|
|
||
| // require('_stream_wrap').StreamWrap | ||
| StreamWrap.StreamWrap = StreamWrap; | ||
|
|
||
| StreamWrap.prototype.isAlive = function isAlive() { | ||
| return this.readable && this.writable; | ||
| }; | ||
|
|
||
| StreamWrap.prototype.isClosing = function isClosing() { | ||
| return !this.isAlive(); | ||
| }; | ||
|
|
||
| StreamWrap.prototype.readStart = function readStart() { | ||
| this.stream.resume(); | ||
| return 0; | ||
| }; | ||
|
|
||
| StreamWrap.prototype.readStop = function readStop() { | ||
| this.stream.pause(); | ||
| return 0; | ||
| }; | ||
|
|
||
| StreamWrap.prototype.shutdown = function shutdown(req) { | ||
| var self = this; | ||
|
|
||
| this.stream.end(function() { | ||
| // Ensure that write was dispatched | ||
| setImmediate(function() { | ||
| self._handle.finishShutdown(req, 0); | ||
| }); | ||
| }); | ||
| return 0; | ||
| }; | ||
|
|
||
| StreamWrap.prototype.write = function write(req, bufs) { | ||
| var pending = bufs.length; | ||
| var self = this; | ||
|
|
||
| self.stream.cork(); | ||
| bufs.forEach(function(buf) { | ||
| self.stream.write(buf, done); | ||
| }); | ||
| self.stream.uncork(); | ||
|
|
||
| function done(err) { | ||
| if (!err && --pending !== 0) | ||
| return; | ||
|
|
||
| // Ensure that this is called once in case of error | ||
| pending = 0; | ||
|
|
||
| // Ensure that write was dispatched | ||
| setImmediate(function() { | ||
| var errCode = 0; | ||
| if (err) { | ||
| if (err.code && uv['UV_' + err.code]) | ||
| errCode = uv['UV_' + err.code]; | ||
| else | ||
| errCode = uv.UV_EPIPE; | ||
| } | ||
|
|
||
| self._handle.doAfterWrite(req); | ||
| self._handle.finishWrite(req, errCode); | ||
| }); | ||
| } | ||
|
|
||
| return 0; | ||
| }; |
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
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
Oops, something went wrong.