mingo/esm/operators/pipeline/fill.js

MongoDB query language for in-memory objects

import { assert, has, isObject } from "../../util";
import { $ifNull } from "../expression/conditional/ifNull";
import { $linearFill } from "../window/linearFill";
import { $locf } from "../window/locf";
import { $addFields } from "./addFields";
import { $setWindowFields } from "./setWindowFields";
const FILL_METHODS = { locf: "$locf", linear: "$linearFill" };
function $fill(coll, expr, options) {
  assert(!expr.sortBy || isObject(expr.sortBy), "sortBy must be an object.");
  assert(
    !!expr.sortBy || Object.values(expr.output).every((m) => has(m, "value")),
    "sortBy required if any output field specifies a 'method'."
  );
  assert(
    !(expr.partitionBy && expr.partitionByFields),
    "specify either partitionBy or partitionByFields."
  );
  assert(
    !expr.partitionByFields || expr?.partitionByFields?.every((s) => s[0] !== "$"),
    "fields in partitionByFields cannot begin with '$'."
  );
  options.context.addExpressionOps({ $ifNull });
  options.context.addWindowOps({ $locf, $linearFill });
  const partitionExpr = expr.partitionBy || expr?.partitionByFields?.map((s) => "$" + s);
  const valueExpr = {};
  const methodExpr = {};
  for (const k of Object.keys(expr.output)) {
    const m = expr.output[k];
    if (has(m, "value")) {
      const out = m;
      valueExpr[k] = { $ifNull: [`$$CURRENT.${k}`, out.value] };
    } else {
      const out = m;
      const fillOp = FILL_METHODS[out.method];
      assert(!!fillOp, `invalid fill method '${out.method}'.`);
      methodExpr[k] = { [fillOp]: "$" + k };
    }
  }
  if (Object.keys(methodExpr).length > 0) {
    coll = $setWindowFields(
      coll,
      {
        sortBy: expr.sortBy || {},
        partitionBy: partitionExpr,
        output: methodExpr
      },
      options
    );
  }
  if (Object.keys(valueExpr).length > 0) {
    coll = $addFields(coll, valueExpr, options);
  }
  return coll;
}
export {
  $fill
};