40 lines
1.5 KiB
JavaScript
40 lines
1.5 KiB
JavaScript
/**
|
|
* Tests that the $setUnion accumulator does not allow spilling to disk.
|
|
* @tags: [requires_fcv_81, requires_collstats]
|
|
*/
|
|
import {FixtureHelpers} from "jstests/libs/fixture_helpers.js";
|
|
|
|
const coll = db["set_union_spill_to_disk"];
|
|
coll.drop();
|
|
|
|
const sharded = FixtureHelpers.isSharded(coll);
|
|
|
|
const memoryLimitMB = sharded ? 200 : 100;
|
|
const bigStr = Array(1024 * 1024 + 1).toString(); // 1MB of ','
|
|
|
|
const bulk = coll.initializeUnorderedBulkOp();
|
|
for (let i = 0; i < memoryLimitMB + 10; i++) {
|
|
bulk.insert({_id: i, bigArr: [i + bigStr], sortKey: i});
|
|
}
|
|
assert.commandWorked(bulk.execute());
|
|
|
|
assert.gt(coll.stats().size, memoryLimitMB * 1024 * 1024);
|
|
|
|
// Test accumulating all values into one array. On debug builds we will spill to disk for $group and
|
|
// so may hit the group error code before we hit ExceededMemoryLimit.
|
|
const pipeline = [{$sort: {sortKey: 1}}, {$group: {_id: null, bigArray: {$setUnion: "$bigArr"}}}];
|
|
const expectedCodes = [ErrorCodes.QueryExceededMemoryLimitNoDiskUseAllowed, ErrorCodes.ExceededMemoryLimit];
|
|
|
|
// Test that 'allowDiskUse: false' does indeed prevent spilling to disk.
|
|
assert.commandFailedWithCode(
|
|
db.runCommand({aggregate: coll.getName(), pipeline: pipeline, cursor: {}, allowDiskUse: false}),
|
|
expectedCodes,
|
|
);
|
|
|
|
// The $setUnion accumulator does not support spilling to disk, so ensure that it will fail even
|
|
// when disk use is allowed.
|
|
assert.commandFailedWithCode(
|
|
db.runCommand({aggregate: coll.getName(), pipeline: pipeline, cursor: {}, allowDiskUse: true}),
|
|
expectedCodes,
|
|
);
|