123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147 |
- 'use strict';
-
- /**
-
- Streams in a WebSocket connection
- ---------------------------------
-
- We model a WebSocket as two duplex streams: one stream is for the wire protocol
- over an I/O socket, and the other is for incoming/outgoing messages.
-
-
- +----------+ +---------+ +----------+
- [1] write(chunk) -->| ~~~~~~~~ +----->| parse() +----->| ~~~~~~~~ +--> emit('data') [2]
- | | +----+----+ | |
- | | | | |
- | IO | | [5] | Messages |
- | | V | |
- | | +---------+ | |
- [4] emit('data') <--+ ~~~~~~~~ |<-----+ frame() |<-----+ ~~~~~~~~ |<-- write(chunk) [3]
- +----------+ +---------+ +----------+
-
-
- Message transfer in each direction is simple: IO receives a byte stream [1] and
- sends this stream for parsing. The parser will periodically emit a complete
- message text on the Messages stream [2]. Similarly, when messages are written
- to the Messages stream [3], they are framed using the WebSocket wire format and
- emitted via IO [4].
-
- There is a feedback loop via [5] since some input from [1] will be things like
- ping, pong and close frames. In these cases the protocol responds by emitting
- responses directly back to [4] rather than emitting messages via [2].
-
- For the purposes of flow control, we consider the sources of each Readable
- stream to be as follows:
-
- * [2] receives input from [1]
- * [4] receives input from [1] and [3]
-
- The classes below express the relationships described above without prescribing
- anything about how parse() and frame() work, other than assuming they emit
- 'data' events to the IO and Messages streams. They will work with any protocol
- driver having these two methods.
- **/
-
-
- var Stream = require('stream').Stream,
- util = require('util');
-
-
- var IO = function(driver) {
- this.readable = this.writable = true;
- this._paused = false;
- this._driver = driver;
- };
- util.inherits(IO, Stream);
-
- // The IO pause() and resume() methods will be called when the socket we are
- // piping to gets backed up and drains. Since IO output [4] comes from IO input
- // [1] and Messages input [3], we need to tell both of those to return false
- // from write() when this stream is paused.
-
- IO.prototype.pause = function() {
- this._paused = true;
- this._driver.messages._paused = true;
- };
-
- IO.prototype.resume = function() {
- this._paused = false;
- this.emit('drain');
-
- var messages = this._driver.messages;
- messages._paused = false;
- messages.emit('drain');
- };
-
- // When we receive input from a socket, send it to the parser and tell the
- // source whether to back off.
- IO.prototype.write = function(chunk) {
- if (!this.writable) return false;
- this._driver.parse(chunk);
- return !this._paused;
- };
-
- // The IO end() method will be called when the socket piping into it emits
- // 'close' or 'end', i.e. the socket is closed. In this situation the Messages
- // stream will not emit any more data so we emit 'end'.
- IO.prototype.end = function(chunk) {
- if (!this.writable) return;
- if (chunk !== undefined) this.write(chunk);
- this.writable = false;
-
- var messages = this._driver.messages;
- if (messages.readable) {
- messages.readable = messages.writable = false;
- messages.emit('end');
- }
- };
-
- IO.prototype.destroy = function() {
- this.end();
- };
-
-
- var Messages = function(driver) {
- this.readable = this.writable = true;
- this._paused = false;
- this._driver = driver;
- };
- util.inherits(Messages, Stream);
-
- // The Messages pause() and resume() methods will be called when the app that's
- // processing the messages gets backed up and drains. If we're emitting
- // messages too fast we should tell the source to slow down. Message output [2]
- // comes from IO input [1].
-
- Messages.prototype.pause = function() {
- this._driver.io._paused = true;
- };
-
- Messages.prototype.resume = function() {
- this._driver.io._paused = false;
- this._driver.io.emit('drain');
- };
-
- // When we receive messages from the user, send them to the formatter and tell
- // the source whether to back off.
- Messages.prototype.write = function(message) {
- if (!this.writable) return false;
- if (typeof message === 'string') this._driver.text(message);
- else this._driver.binary(message);
- return !this._paused;
- };
-
- // The Messages end() method will be called when a stream piping into it emits
- // 'end'. Many streams may be piped into the WebSocket and one of them ending
- // does not mean the whole socket is done, so just process the input and move
- // on leaving the socket open.
- Messages.prototype.end = function(message) {
- if (message !== undefined) this.write(message);
- };
-
- Messages.prototype.destroy = function() {};
-
-
- exports.IO = IO;
- exports.Messages = Messages;
|