Bumped ws to v2.0.0.

echo-server.js

@@ -1,3 +1,5 @@
+'use strict'
+
 var http = require('http')
 var websocket = require('./')
 var server = null

index.js

@@ -1,6 +1,6 @@
 
-var Server = require('./server.js')
+var server = require('./server.js')
 
 module.exports = require('./stream.js')
-module.exports.Server = Server
-module.exports.createServer = Server
+module.exports.Server = server.Server
+module.exports.createServer = server.createServer

package.json

@@ -26,7 +26,7 @@
     "duplexify": "^3.2.0",
     "inherits": "^2.0.1",
     "through2": "^2.0.0",
-    "ws": "^1.0.1",
+    "ws": "^2.2.3",
     "xtend": "^4.0.0"
   },
   "devDependencies": {

readme.md

@@ -40,6 +40,23 @@ Always convert to `Buffer` in Node.js before sending.
 
 Default: `true`
 
+##### `options.perMessageDeflate`
+
+We recommend disabling the [per message deflate
+extension](https://tools.ietf.org/html/rfc7692) to achieve the best
+throughput.
+
+Default: `true`
+
+Example:
+
+```js
+var websocket = require('websocket-stream')
+var ws = websocket('ws://realtimecats.com', {
+  perMessageDeflate: false
+})
+```
+
 ##### Other options
 
 When used in node.js see the [ws.WebSocket documentation](https://github.com/websockets/ws/blob/master/doc/ws.md#class-wswebsocket)
@@ -57,6 +74,22 @@ function handle(stream) {
 }
 ```
 
+We recommend disabling the [per message deflate
+extension](https://tools.ietf.org/html/rfc7692) to achieve the best
+throughput:
+
+```javascript
+var websocket = require('websocket-stream')
+var wss = websocket.createServer({
+  perMessageDeflate: false,
+  server: someHTTPServer
+}, handle)
+
+function handle(stream) {
+  fs.createReadStream('bigdata.json').pipe(stream)
+}
+```
+
 ## Run the tests
 
 ### Server-side tests

server.js

@@ -1,31 +1,29 @@
+'use strict'
 
-var inherits = require('inherits')
 var WebSocketServer = require('ws').Server
 var stream = require('./stream')
 
-module.exports = Server
-
-function Server(opts, cb) {
-  if (!(this instanceof Server)) {
-    return new Server(opts, cb)
-  }
-
-  WebSocketServer.call(this, opts)
-
-  var proxied = false
-  this.on('newListener', function(event) {
-    if (!proxied && event === 'stream') {
-      proxied = true
-      this.on('connection', function(conn) {
-        this.emit('stream', stream(conn, opts))
-      })
+class Server extends WebSocketServer{
+  constructor(opts, cb) {
+    super(opts)
+
+    var proxied = false
+    this.on('newListener', function(event) {
+      if (!proxied && event === 'stream') {
+        proxied = true
+        this.on('connection', function(conn) {
+          this.emit('stream', stream(conn, opts))
+        })
+      }
+    })
+
+    if (cb) {
+      this.on('stream', cb)
     }
-  })
-
-  if (cb) {
-    this.on('stream', cb)
   }
 }
 
-inherits(Server, WebSocketServer)
-
+module.exports.Server = Server
+module.exports.createServer = function(opts, cb) {
+  return new Server(opts, cb)
+}

stream.js

@@ -67,6 +67,13 @@ function WebSocketStream(target, protocols, options) {
   var coerceToBuffer = options.binary || options.binary === undefined
 
   function socketWriteNode(chunk, enc, next) {
+    // avoid errors, this never happens unless
+    // destroy() is called
+    if (socket.readyState !== WS.OPEN) {
+      next()
+      return
+    }
+
     if (coerceToBuffer && typeof chunk === 'string') {
       chunk = new Buffer(chunk, 'utf8')
     }