pondjs/src/pond/lib/io/collectionout.js

A timeseries library build on top of immutable.js

/**
 *  Copyright (c) 2016-2017, The Regents of the University of California,
 *  through Lawrence Berkeley National Laboratory (subject to receipt
 *  of any required approvals from the U.S. Dept. of Energy).
 *  All rights reserved.
 *
 *  This source code is licensed under the BSD-style license found in the
 *  LICENSE file in the root directory of this source tree.
 */

import Collector from "../collector";
import PipelineOut from "./pipelineout";

class CollectionOut extends PipelineOut {
    constructor(pipeline, options, callback) {
        super(pipeline);

        this._callback = callback;
        this._collector = new Collector(
            {
                windowType: pipeline.getWindowType(),
                windowDuration: pipeline.getWindowDuration(),
                groupBy: pipeline.getGroupBy(),
                emitOn: pipeline.getEmitOn()
            },
            (collection, windowKey, groupByKey) => {
                const groupBy = groupByKey ? groupByKey : "all";
                if (this._callback) {
                    this._callback(collection, windowKey, groupBy);
                } else {
                    let keys = [];
                    if (windowKey !== "global") {
                        keys.push(windowKey);
                    }
                    if (groupBy !== "all") {
                        keys.push(groupBy);
                    }
                    const k = keys.length > 0 ? keys.join("--") : "all";
                    this._pipeline.addResult(k, collection);
                }
            }
        );
    }

    addEvent(event) {
        this._collector.addEvent(event);
    }

    onEmit(cb) {
        this._callback = cb;
    }

    flush() {
        this._collector.flushCollections();
        if (!this._callback) {
            this._pipeline.resultsDone();
        }
    }
}

export default CollectionOut;