You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
2.0 KiB
TypeScript
75 lines
2.0 KiB
TypeScript
import type { Db } from '../db';
|
|
import { MongoInvalidArgumentError, MongoRuntimeError } from '../error';
|
|
import type { Server } from '../sdam/server';
|
|
import type { ClientSession } from '../sessions';
|
|
import type { Callback } from '../utils';
|
|
import { enumToString } from '../utils';
|
|
import { CommandOperation, CommandOperationOptions } from './command';
|
|
|
|
const levelValues = new Set(['off', 'slow_only', 'all']);
|
|
|
|
/** @public */
|
|
export const ProfilingLevel = Object.freeze({
|
|
off: 'off',
|
|
slowOnly: 'slow_only',
|
|
all: 'all'
|
|
} as const);
|
|
|
|
/** @public */
|
|
export type ProfilingLevel = (typeof ProfilingLevel)[keyof typeof ProfilingLevel];
|
|
|
|
/** @public */
|
|
export type SetProfilingLevelOptions = CommandOperationOptions;
|
|
|
|
/** @internal */
|
|
export class SetProfilingLevelOperation extends CommandOperation<ProfilingLevel> {
|
|
override options: SetProfilingLevelOptions;
|
|
level: ProfilingLevel;
|
|
profile: 0 | 1 | 2;
|
|
|
|
constructor(db: Db, level: ProfilingLevel, options: SetProfilingLevelOptions) {
|
|
super(db, options);
|
|
this.options = options;
|
|
switch (level) {
|
|
case ProfilingLevel.off:
|
|
this.profile = 0;
|
|
break;
|
|
case ProfilingLevel.slowOnly:
|
|
this.profile = 1;
|
|
break;
|
|
case ProfilingLevel.all:
|
|
this.profile = 2;
|
|
break;
|
|
default:
|
|
this.profile = 0;
|
|
break;
|
|
}
|
|
|
|
this.level = level;
|
|
}
|
|
|
|
override execute(
|
|
server: Server,
|
|
session: ClientSession | undefined,
|
|
callback: Callback<ProfilingLevel>
|
|
): void {
|
|
const level = this.level;
|
|
|
|
if (!levelValues.has(level)) {
|
|
return callback(
|
|
new MongoInvalidArgumentError(
|
|
`Profiling level must be one of "${enumToString(ProfilingLevel)}"`
|
|
)
|
|
);
|
|
}
|
|
|
|
// TODO(NODE-3483): Determine error to put here
|
|
super.executeCommand(server, session, { profile: this.profile }, (err, doc) => {
|
|
if (err == null && doc.ok === 1) return callback(undefined, level);
|
|
return err != null
|
|
? callback(err)
|
|
: callback(new MongoRuntimeError('Error with profile command'));
|
|
});
|
|
}
|
|
}
|