Reduce.js 971 B

12345678910111213141516171819202122232425262728293031
  1. const {Writable} = require('stream');
  2. const defaultInitial = 0;
  3. const defaultReducer = (acc, value) => value;
  4. class Reduce extends Writable {
  5. constructor(options) {
  6. super(Object.assign({}, options, {objectMode: true}));
  7. this.accumulator = defaultInitial;
  8. this._reducer = defaultReducer;
  9. if (options) {
  10. 'initial' in options && (this.accumulator = options.initial);
  11. 'reducer' in options && (this._reducer = options.reducer);
  12. }
  13. }
  14. _write(chunk, encoding, callback) {
  15. this.accumulator = this._reducer.call(this, this.accumulator, chunk);
  16. callback(null);
  17. }
  18. _writev(chunks, callback) {
  19. chunks.forEach(item => (this.accumulator = this._reducer.call(this, this.accumulator, item.chunk)));
  20. callback(null);
  21. }
  22. static make(reducer, initial) {
  23. return new Reduce(typeof reducer == 'object' ? reducer : {reducer, initial});
  24. }
  25. }
  26. Reduce.reduce = Reduce.make;
  27. Reduce.make.Constructor = Reduce;
  28. module.exports = Reduce;