browser/libs/utils/stream-byte-counter.js

// Copyright 2015 The Vanadium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

import { default as Stream } from "stream"

var Transform = Stream.Transform;
/*
 * A through transform stream that counts number of bytes being piped to it
 * @param {function} onUpdate Callback function that gets called with number of
 * bytes read when a chunk is read
 * @class
 */
export class StreamByteCounter extends Transform {
  constructor(onUpdate) {
    super();
    this._onUpdate = onUpdate;
  }

  _transform(chunk, encoding, cb) {
    this._onUpdate(chunk.length);
    this.push(chunk)
    cb();
  }
}