Use ThrottleGroup

tasks/throttle.js

@@ -1,19 +1,22 @@
 'use strict';
 
 var net = require('net');
-var Throttle = require('stream-throttle');
+var ThrottleGroup = require('stream-throttle').ThrottleGroup;
 
 module.exports = function(grunt) {
 
   grunt.registerMultiTask('throttle', 'Grunt plugin for testing under a throttled connection.', function() {
 
     var self = this;
 
+    var upThrottle = new ThrottleGroup({rate: self.data.upstream});
+    var downThrottle = new ThrottleGroup({rate: self.data.downstream});
+
     var server = net.createServer(function (local) {
       var remote = net.createConnection({host: self.data.remote_host, port: self.data.remote_port});
 
-      var localThrottle = new Throttle({rate: self.data.upstream});
-      var remoteThrottle = new Throttle({rate: self.data.downstream});
+      var localThrottle = upThrottle.throttle();
+      var remoteThrottle = downThrottle.throttle();
 
       local.pipe(localThrottle).pipe(remote);
       remote.pipe(remoteThrottle).pipe(local);