@tanstack/db-ivm/dist/cjs/operators/topK.cjs

Incremental View Maintenance for TanStack DB based on Differential Dataflow

"use strict";
Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
const multiset = require("../multiset.cjs");
const reduce = require("./reduce.cjs");
function topK(comparator, options) {
  const limit = (options == null ? void 0 : options.limit) ?? Infinity;
  const offset = (options == null ? void 0 : options.offset) ?? 0;
  return (stream) => {
    const reduced = stream.pipe(
      reduce.reduce((values) => {
        const consolidated = new multiset.MultiSet(values).consolidate();
        const sortedValues = consolidated.getInner().sort((a, b) => comparator(a[0], b[0]));
        return sortedValues.slice(offset, offset + limit);
      })
    );
    return reduced;
  };
}
function topKWithIndex(comparator, options) {
  const limit = (options == null ? void 0 : options.limit) ?? Infinity;
  const offset = (options == null ? void 0 : options.offset) ?? 0;
  return (stream) => {
    const reduced = stream.pipe(
      reduce.reduce((values) => {
        const consolidated = new multiset.MultiSet(values).consolidate();
        let i = offset;
        const sortedValues = consolidated.getInner().sort((a, b) => comparator(a[0], b[0])).slice(offset, offset + limit).map(([value, multiplicity]) => [
          [value, i++],
          multiplicity
        ]);
        return sortedValues;
      })
    );
    return reduced;
  };
}
exports.topK = topK;
exports.topKWithIndex = topKWithIndex;
//# sourceMappingURL=topK.cjs.map