Files
MultiforaDB/docs/M2_DESIGN_REVIEW.md
A.Shakhmatov 8ebeb9d4ec commands: $out and $merge, written by the dispatch epilogue
The seven reachable failures of M2, and the first commit of the milestone to
move the scorecard: 194/97/196 -> 201/90/196, with `aggregate-*.json` going
9 pass / 13 fail to 16 pass / 6 fail. The seven that moved are exactly the
seven priced as reachable, and the six that remain are exactly the six
attributed to M2.5 ($addFields, the expression engine), M4 ($listLocalSessions)
and M8 (collation).

Both stages write to a collection the pipeline is not reading, and three things
stood against doing that in the handler: `aggregate` is a `.read` command,
dispatch takes locks from a static table keyed on the command name before the
handler runs, and `Collection.lock` allows exactly one collection lock at a
time. So the handler computes the output under the locks it has and leaves it
in `Context.pending_write`; the epilogue applies it with nothing held, beside
the commit and the checkpoint already there. The `.read`/`.write` contract is
amended in its own comment rather than quietly broken.

`pending_write` is cleared at the top of every dispatch, so a handler that
errors before setting one cannot leave the previous command's write to fire. A
failed write replaces the pipeline's `ok: 1` with the failure, because a client
told the aggregation succeeded would believe the collection had been written.

What the stages do not implement is refused, not ignored: `$merge`'s
`whenMatched`, `whenNotMatched`, `on` and `let` all select behaviour this
server does not have, and a `whenMatched: "fail"` that silently merged would be
the same lie Tier 0 spent three commits removing. Codes measured against
mongod 8.3.7. `$out` and `$merge` answer byte-identically to it on both the
replace and the upsert case.

NOT ATOMIC, and said out loud in the code rather than left to be discovered.
mongod replaces an `$out` target atomically; this engine has no
cross-collection atomicity and no rename to build one from, so a crash between
the drop and the last insert leaves the target holding part of the new output
where MongoDB would leave the whole of the old. The fix is
write-to-temp-and-rename and rename is a command that does not exist here.

The test's mutation is the argument for the epilogue in one line: apply the
write inside the `$out` branch and it deadlocks rather than fails.

191/191 unit tests in ReleaseFast and ReleaseSafe, 83/83 fuzz, e2e 49, e2e2
concurrent, e2e3 16, e2e4 17, e2e6 72, e2e7 86.
2026-08-09 21:02:01 +03:00

18 KiB

M2 design review — aggregation expansion

PLAN §3 gives M2 one line of scope ("pipeline stages and expression engine, tiered scope defined at M2 design review") and one line of gate ("aggregate spec suite green"), and §6 lists the three questions it deferred: the stage/expression tiers, which spec files are the gate, and whether $lookup/$unwind/$facet make the first cut.

This document answers them, and reports one finding that has to be settled before any of the rest is worth discussing.

Everything below is measured against the M1 tip (a748a3d) unless it says otherwise. Nothing here is recalled.


1. The gate named in the plan does not exist

mongodb/specifications has no aggregation suite. The full list of source/ directories at the pinned commit is: auth, bson-*, causal-consistency, change-streams, client-side-encryption, collation, collection-management, command-logging-and-monitoring, compression, connection-*, crud, gridfs, index-management, load-balancers, logging, read-write-concern, retryable-*, run-command, sessions, server-discovery-and-monitoring, server-selection, transactions, unified-test-format, versioned-api, and a handful of others. There is no aggregation.

The thirteen aggregate-*.json files we do run live inside crud, and they test the aggregate command surface — cursor and batchSize, readConcern and readPreference routing, the write semantics of $out/$merge, collation, let, allowDiskUse, rawData. They barely touch stages, and they touch expressions not at all: $lookup, $unwind, $facet, $addFields, $replaceRoot do not appear anywhere in the pinned corpus.

So "aggregate spec suite green" is not a gate for the milestone the scope line describes. It is a gate for a much smaller milestone that happens to share the command name. Whichever option §4 picks, PLAN's M2 gate line has to be rewritten, and that is the first decision this review asks for.

Where the current corpus stands, for reference:

aggregate-*.json overall     9 pass   13 fail   19 skip
failure cases
$merge unimplemented 5
$out unimplemented 3
db.aggregate(){aggregate: 1}, no collection name 3
collation in aggregate 1
let 1

All 19 skips are environmental -- 10 distinct requirements, some covering a whole file at once (needs topology replicaset, needs server >= 5.0, needs server >= 8.2.0, needs server <= 4.2.99). Not one can turn green on a standalone, so a literal reading of "green" is unreachable for that reason alone, and the gate has to be stated as 0 fail rather than all pass.


2. Where the implementation actually is

Seven stages, all in one if/else if chain in cmd_aggregate: $match, $sort, $skip, $limit, $project, $group, $count. Anything else answers Location40324, with the code verified against a real mongod.

($set and $unset exist in the tree as update operators in src/update.zig. They are not pipeline stages. An earlier count of nine stages in this project's notes was wrong.)

There is no expression engine. $field path resolution is open-coded in two places inside run_group, and nowhere else. $sum is the only accumulator.

That last sentence is the important one, because of how the missing pieces fail. Probed against a live server on three documents ({g:'a',x:10}, {g:'a',x:20}, {g:'b',x:30}):

pipeline answer correct answer
$group: {_id:'$g', s:{$sum:'$x'}} [{a,30},{b,30}]
$group: {_id:'$g', a:{$avg:'$x'}} [{a,0},{b,0}] [{a,15},{b,30}]
$group: {_id:'$g', m:{$max:'$x'}} [{a,0},{b,0}] [{a,20},{b,30}]
$group: {_id:'$g', p:{$push:'$x'}} [{a,0},{b,0}] arrays
$group: {_id:{g:'$g'}, n:{$sum:1}} one group, _id = {"g":"$g"} two groups
$project:{g:1} then $match:{x:10} matches matches nothing
$group: {_id:{$literal:1}, ...} _id = {"$literal":1} _id = 1
$addFields Location40324 (unimplemented)

Six of those eight are silent wrong answers with ok: 1. Only $addFields fails honestly. That is the shape of the risk in this area and it should drive the milestone: an unrecognised stage is a bug report, an $avg that returns 0 is a corrupted report that nobody files. It is also precisely the class the M1 spec-runner work was about — and the corpus cannot see any of it, because the corpus has no stage tests.

The pipeline model will not carry the milestone as it stands. The stream is a materialized window [start, end) over either slab offsets or generated documents, and each stage moves the bounds. That works for the filtering and reordering stages and nothing else:

  • $project is not a stage transform at all. It sets a variable that is applied once, at emit — so only the last $project in a pipeline has any effect, and a $match after a $project sees the unprojected document.
  • $unwind is 1→N. There is no way to express that in a window over the input.
  • $lookup needs a second collection under a second lock, inside a stage.
  • $facet needs the same stream fanned into several independent sub-pipelines.
  • allowDiskUse is accepted and ignored. $group and $sort are unbounded in memory; the three passing allowdiskuse cases pass because the option is parsed, not because anything spills. src/spill.zig exists and is used by index builds, not here.

So M2 is not "add stages to the chain". It is: build an expression evaluator, change the stream model to a per-stage document iterator, and then stages become cheap. Anything that adds stages before those two exist adds more silent wrong answers.


3. What that implies for tiers

The dependency order is forced, not chosen:

Tier 0 — stop lying (no new features). Refuse what is not implemented instead of answering 0: unknown accumulators, non-path expressions where a path is expected, $project used anywhere but last. Every one of these becomes an error with a measured mongod code. This is small, and it converts six silent wrong answers into six honest ones. It is also the only tier that makes the later ones measurable, for the same reason Stage 1 preceded Stage 3 in M1.

Tier 1 — the expression engine. A real evaluator over bson.Value: field paths, $literal, comparison and boolean operators, arithmetic, $cond/$ifNull/$switch, string and date operators. Everything above depends on it, including $project's computed fields and $group's _id.

Tier 2 — the stage iterator. Replace the window with a per-stage pull model, so a stage can emit more or fewer documents than it consumed. $project becomes a real transform. Carries $addFields/$set, $unset, $replaceRoot, $unwind, $sortByCount.

Tier 3 — the accumulators. $avg, $min, $max, $first, $last, $push, $addToSet, $count, $mergeObjects. Cheap once Tier 1 exists.

Tier 4 — the expensive stages. $lookup (second collection, lock order), $facet (sub-pipelines), $unionWith, $graphLookup. Each is a design question of its own.

Tier 5 — the write stages. $out, $merge. These are not aggregation work at all — they are a write path that happens to be spelled as a stage, with their own durability and atomicity questions. They are also 8 of the 13 current aggregate-* failures.

Orthogonal, and cheap: db.aggregate() ({aggregate: 1}), 3 failures. Orthogonal, and not cheap: allowDiskUse meaning something, which is a memory-bound question for $group/$sort and belongs with whichever tier first makes those able to hold a lot.


4. Gate options

Each option is stated with what it would actually prove and what it costs.

Option A — "the aggregate command surface"

Gate: the 13 aggregate-*.json files in the crud corpus, minus the 13 environmental skips. Scope: Tier 5 + db.aggregate() + collation + let.

  • Proves: the aggregate command behaves — cursors, read concern, write stages, database-level form.
  • Does not prove: any stage or expression is correct. All six silent wrong answers survive this gate untouched.
  • Cost: small. Two write stages and three small fixes.
  • Honest name: this is not "aggregation expansion". If chosen, M2 should be renamed and the aggregation engine deferred to its own milestone.

Option B — a hand-built stage corpus, modelled on mongod's own

Gate: a new tests/spec/aggregate/ corpus in the unified format the runner already reads, with cases ported from mongod's jstests/aggregation and every expected result measured against mongod 8.3.7, the way the cursor and lsid error codes were.

  • Proves: exactly what the milestone claims to build, at whatever tier depth the corpus is written to.
  • Cost: the corpus is the milestone's second deliverable and a real one — but the runner, the entity model, the matchers and expectEvents all exist already, so the marginal cost is writing cases, not machinery.
  • Risk, stated plainly: a corpus we author is a corpus we can accidentally write to match our own bugs. The mitigation is the one this project already uses: every expectation measured against a real mongod before it is committed, never recalled.

Option C — differential against mongod

Gate: a harness that runs the same pipelines against MultiforaDB and mongod 8.3.7 and diffs the results, over generated pipelines and a fixed corpus.

  • Proves: the most, by a distance, and it finds the silent-wrong-answer class by construction rather than by someone thinking to test for it.
  • Cost: mongod becomes a test dependency (it already is one for measurement, but not for the gate). Result ordering and floating-point formatting need normalising, and $group output order is genuinely unspecified.
  • Note: this is a harness, not a corpus, so it composes with B rather than competing with it — B gives regression cases with committed expectations, C finds new ones.

Option D — keep the plan's line and reinterpret "green"

Gate: aggregate-*.json at 22 pass / 0 fail / 13 skip, and declare the skips out of scope.

  • Proves: the same as A.
  • Why it is listed: only to be rejected explicitly. It reads as if the aggregation milestone were gated on aggregation tests, and it is not. If A is the decision, say A.

5. Decision

Taken: the command surface first (Option A as M2), the engine second (M2.5), with Tier 0 folded into M2. Recorded as PLAN amendment A6; the §3 rows are rewritten and §6's deferred entry now points here.

The cost of that order is on the record: the engine is what gives wrong answers today, and this leaves them alive for a further milestone. Tier 0 is what makes the trade acceptable rather than merely convenient — every unimplemented construct stops answering 0 and starts answering an error, so the wrong answers become visible bug reports instead of quiet ones while the command surface is built. It is not optional under this sequencing.

The recommendation the review arrived at independently is kept below, since a decision is easier to revisit when the argument it overrode is still legible.

5.1 Recommendation as written before the decision

Tier 0 first, alone, before any scoping decision is final — the same argument that put expectEvents before the free list in M1. Six silent wrong answers are six results nobody can measure past, and turning them into errors is a day's work that changes what every later number means. It will move the scorecard down, and that is the point.

Then B as the gate, with C as the harness that feeds it. Option A is a real piece of work and should happen — the $merge/$out failures are 8 of 13 — but it should be named for what it is and probably belongs in M6 (admin/ops) or its own small milestone, not as the gate for an engine.

Tiers 1→3 as M2 proper, with Tier 4 explicitly out of the first cut. That answers §6's third question: $unwind yes (it falls out of Tier 2 for free), $lookup and $facet no$lookup because a stage that takes a second collection lock while the pipeline holds one is a lock-order design question this milestone should not be carrying, and $facet because it needs sub-pipeline execution that Tier 2's iterator should be allowed to settle first.

One thing to decide that this review cannot decide alone: whether allowDiskUse must mean something by the end of M2. It is currently accepted and ignored, which is the same category of lie as $avg returning 0 — a client that passes it believes it is protected. If the answer is yes, the memory bound on $group/$sort joins Tier 2 rather than trailing it.


6. What this review does not cover

  • $out/$merge durability semantics (atomic swap? logged? crash behaviour?). Needed before Tier 5 is scoped.
  • Whether the expression engine is shared with update's pipeline-update form, which M3 will need. Probably yes; it changes where the code lives.
  • Index use inside a pipeline beyond the existing leading-$match pushdown.

7. Post-decision measurement: the gate is not reachable either

Written after Tier 0 landed, when the 13 failures were priced one by one instead of counted. Section 1 established that the corpus named in the plan was the wrong instrument; this establishes that the target set on the corpus we kept is unreachable too. Both were found by looking rather than by assuming, and this one only by opening each case.

cases what they need milestone
5 aggregate-merge.json $merge with a plain into M2, blocked below
2 aggregate-out.json $out with a plain target M2, blocked below
3 db-aggregate*.json $listLocalSessions and $addFields M4 + M2.5
2 aggregate-let.json let, $$var, $expr M2.5
1 aggregate-collation.json a case-insensitive collation M8

So 6 of 13 can never turn green in M2, whatever is built, and the gate has to read 0 fail among the seven this milestone can reach, with the other six named and attributed. Note what the three db.aggregate() cases really need: implementing {aggregate: 1} moves none of them, because each then fails on $listLocalSessions instead. The design review priced that line "orthogonal, and cheap"; it was cheap and it was worth nothing.

And the seven need a decision this review has no authority to take

$out and $merge write to a collection the pipeline is not reading. Three things in the current design stand against that, and none of them is a detail:

  • aggregate is declared .kind = .read, and the dispatch contract is explicit that "only .write commands may call engine mutation functions".
  • dispatch acquires locks from a static table keyed on the command name, before the handler runs, for the one collection named by the command.
  • Collection.lock's own comment states the invariant that decides this: never more than one collection lock at a time. Taking the target's exclusive lock while holding the source's shared one breaks it directly.

This is the same shape as the objection that kept $lookup out of M2.5's first cut — "a stage that takes a second collection's lock while the pipeline holds one is a lock-order question that should not ride along" — and the review should have applied it to the write stages in the same breath. It did not, which is the review's own error and is recorded here rather than quietly corrected.

Two ways out, both real:

(a) A pipeline that ends in a write stage is a write command. Compute the effective lock shape from the pipeline rather than the name: catalog exclusive, no collection lock from dispatch, and the handler takes source-shared → release → target-exclusive in that order. Heavier than mongod, and it serializes such an aggregate against the whole catalog, but it keeps every existing invariant intact and is honest about what the command is.

(b) The write stages run in the dispatch epilogue, after every lock is released, next to the commit and the checkpoint that already live there. The handler stashes what to write. Cheaper and better-behaved under concurrency, but it amends the .read/.write contract, so the contract's comment has to change with it and say why.

Either way the durability question is open and belongs with whichever is chosen: mongod's $out replaces the target collection atomically, and this engine has no cross-collection atomicity. A pipeline that fails after writing half its output must not leave the target half-replaced.

Decision: (b), the epilogue. Taken after the options were set out. The handler computes the output under the locks it already holds and leaves it in Context.pending_write; the epilogue applies it once every lock is released, beside the commit and the checkpoint that already live there. The .read/.write contract's comment was amended to say so and why.

Measured after: aggregate-*.json went 9 pass / 13 fail to 16 pass / 6 fail, and the whole corpus 194/97/196 to 201/90/196. The seven that moved are exactly the seven priced as reachable above, and the six that remain are exactly the six attributed to M2.5, M4 and M8.

The durability question is answered honestly rather than solved. $out is not atomic here: it drops the target and inserts, so a crash in between leaves part of the new output where mongod would leave the whole of the old. The shape that fixes it is write-to-temp-and-rename, and this server has no rename command. Recorded in apply_pending_write's own comment as well, since that is where somebody will be standing when it matters.