kefir/src/two-sources/buffer-by.js

Reactive Programming library for JavaScript inspired by Bacon.js and RxJS with focus on high performance and low memory usage

import {createStream, createProperty} from '../patterns/two-sources'

const mixin = {
  _init({flushOnEnd = true} = {}) {
    this._buff = []
    this._flushOnEnd = flushOnEnd
  },

  _free() {
    this._buff = null
  },

  _flush() {
    if (this._buff !== null) {
      this._emitValue(this._buff)
      this._buff = []
    }
  },

  _handlePrimaryEnd() {
    if (this._flushOnEnd) {
      this._flush()
    }
    this._emitEnd()
  },

  _onActivation() {
    this._primary.onAny(this._$handlePrimaryAny)
    if (this._alive && this._secondary !== null) {
      this._secondary.onAny(this._$handleSecondaryAny)
    }
  },

  _handlePrimaryValue(x) {
    this._buff.push(x)
  },

  _handleSecondaryValue() {
    this._flush()
  },

  _handleSecondaryEnd() {
    if (!this._flushOnEnd) {
      this._emitEnd()
    }
  },
}

const S = createStream('bufferBy', mixin)
const P = createProperty('bufferBy', mixin)

export default function bufferBy(primary, secondary, options /* optional */) {
  return new (primary._ofSameType(S, P))(primary, secondary, options)
}