fixup! WIP compressed write

Author: dougwilson

lib/protocol/PacketWriter.js

@@ -27,50 +27,31 @@ PacketWriter.prototype.finalize = function finalize(parser) {
     this._offset = 0;
   }
 
+  var maxPacketLength = parser._compressed
+    ? MAX_PACKET_LENGTH - PACKET_HEADER_LENGTH
+    : MAX_PACKET_LENGTH;
+
   var buffer  = this._buffer;
   var length  = this._offset;
-  var packets = Math.floor(length / MAX_PACKET_LENGTH) + 1;
-  var size    = length + (packets * PACKET_HEADER_LENGTH);
-
-  this._buffer = Buffer.allocUnsafe(size);
-  this._offset = 0;
+  var packets = Math.floor(length / maxPacketLength) + 1;
 
   for (var packet = 0; packet < packets; packet++) {
     var isLast = (packet + 1 === packets);
     var packetLength = (isLast)
-      ? length % MAX_PACKET_LENGTH
-      : MAX_PACKET_LENGTH;
+      ? length % maxPacketLength
+      : maxPacketLength;
 
-    var packetNumber = parser.incrementPacketNumber();
-    var packetOffset = this._offset;
-    var start        = packet * MAX_PACKET_LENGTH;
-    var end          = start + packetLength;
+    var num   = parser.incrementPacketNumber();
+    var start = packet * maxPacketLength;
+    var end   = start + packetLength;
+    var buf   = this._toPacket(num, buffer.slice(start, end));
 
-    this.writeBuffer(this._toPacket(packetNumber, buffer.slice(start, end)));
-
-    if (!parser._compressed) {
-      this.emit('data', this._buffer.slice(packetOffset, this._offset));
+    if (parser._compressed) {
+      num = parser.incrementCompressedPacketNumber();
+      buf = this._toCompressedPacket(num, buf);
     }
-  }
 
-  if (parser._compressed) {
-    buffer  = this._buffer;
-    length  = this._offset;
-    packets = Math.floor(length / MAX_PACKET_LENGTH) + 1;
-    size    = length + (packets * COMPRESSED_PACKET_HEADER_LENGTH);
-
-    for (packet = 0; packet < packets; packet++) {
-      isLast = (packet + 1 === packets);
-      packetLength = (isLast)
-        ? length % MAX_PACKET_LENGTH
-        : MAX_PACKET_LENGTH;
-
-      var num   = parser.incrementCompressedPacketNumber();
-      var start = packet * MAX_PACKET_LENGTH;
-      var end   = start + packetLength;
-
-      this.emit('data', this._toCompressedPacket(num, buffer.slice(start, end)));
-    }
+    this.emit('data', buf);
   }
 };