Files
mongo/jstests/core/profile_sampling.js
2022-05-06 22:15:18 +00:00

81 lines
2.8 KiB
JavaScript

// Confirms that the number of profiled operations is consistent with the sampleRate, if set.
// @tags: [
// does_not_support_stepdowns,
// requires_fastcount,
// requires_profiling,
// # Tenant migrations passthrough suites automatically retry operations on TenantMigrationAborted
// # errors.
// tenant_migration_incompatible,
// ]
(function() {
"use strict";
// Use a special db to support running other tests in parallel.
const profileDB = db.getSiblingDB("profile_sampling");
const coll = profileDB.profile_sampling;
profileDB.dropDatabase();
let originalProfilingSettings;
try {
originalProfilingSettings = assert.commandWorked(profileDB.setProfilingLevel(0));
profileDB.system.profile.drop();
assert.eq(0, profileDB.system.profile.count());
profileDB.createCollection(coll.getName());
assert.commandWorked(coll.insert({x: 1}));
assert.commandWorked(profileDB.setProfilingLevel(1, {sampleRate: 0, slowms: -1}));
assert.neq(null, coll.findOne({x: 1}));
assert.eq(1, coll.find({x: 1}).count());
assert.commandWorked(coll.update({x: 1}, {$inc: {a: 1}}));
assert.commandWorked(profileDB.setProfilingLevel(0));
assert.eq(0, profileDB.system.profile.count());
profileDB.system.profile.drop();
assert.commandWorked(profileDB.setProfilingLevel(1, {sampleRate: 0.5, slowms: -1}));
// This should generate about 500 profile log entries.
for (let i = 0; i < 500; i++) {
assert.neq(null, coll.findOne({x: 1}));
assert.commandWorked(coll.update({x: 1}, {$inc: {a: 1}}));
}
assert.commandWorked(profileDB.setProfilingLevel(0));
assert.between(10, profileDB.system.profile.count(), 990);
profileDB.system.profile.drop();
// Profiling level of 2 should log all operations, regardless of sample rate setting.
assert.commandWorked(profileDB.setProfilingLevel(2, {sampleRate: 0}));
// This should generate exactly 1000 profile log entries.
for (let i = 0; i < 5; i++) {
assert.neq(null, coll.findOne({x: 1}));
assert.commandWorked(coll.update({x: 1}, {$inc: {a: 1}}));
}
assert.commandWorked(profileDB.setProfilingLevel(0));
assert.eq(10, profileDB.system.profile.count());
profileDB.system.profile.drop();
} finally {
let profileCmd = {};
profileCmd.profile = originalProfilingSettings.was;
// The profile command will reject unknown fields, so wee need to populate only the fields
// accepted from parser
if ('slowms' in originalProfilingSettings)
profileCmd.slowms = originalProfilingSettings.slowms;
if ('sampleRate' in originalProfilingSettings)
profileCmd.sampleRate = originalProfilingSettings.sampleRate;
if ('filter' in originalProfilingSettings)
profileCmd.filter = originalProfilingSettings.filter;
assert.commandWorked(profileDB.runCommand(profileCmd));
}
}());