-
Notifications
You must be signed in to change notification settings - Fork 1.8k
/
run_command.ts
55 lines (50 loc) · 1.76 KB
/
run_command.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
import type { BSONSerializeOptions, Document } from '../bson';
import { type Db } from '../db';
import { type TODO_NODE_3286 } from '../mongo_types';
import type { ReadPreferenceLike } from '../read_preference';
import type { Server } from '../sdam/server';
import type { ClientSession } from '../sessions';
import { MongoDBNamespace } from '../utils';
import { AbstractOperation } from './operation';
/** @public */
export type RunCommandOptions = {
/** Specify ClientSession for this command */
session?: ClientSession;
/** The read preference */
readPreference?: ReadPreferenceLike;
} & BSONSerializeOptions;
/** @internal */
export class RunCommandOperation<T = Document> extends AbstractOperation<T> {
constructor(parent: Db, public command: Document, public override options: RunCommandOptions) {
super(options);
this.ns = parent.s.namespace.withCollection('$cmd');
}
override async execute(server: Server, session: ClientSession | undefined): Promise<T> {
this.server = server;
return server.commandAsync(this.ns, this.command, {
...this.options,
readPreference: this.readPreference,
session
}) as TODO_NODE_3286;
}
}
export class RunAdminCommandOperation<T = Document> extends AbstractOperation<T> {
constructor(
public command: Document,
public override options: RunCommandOptions & {
noResponse?: boolean;
bypassPinningCheck?: boolean;
}
) {
super(options);
this.ns = new MongoDBNamespace('admin', '$cmd');
}
override async execute(server: Server, session: ClientSession | undefined): Promise<T> {
this.server = server;
return server.commandAsync(this.ns, this.command, {
...this.options,
readPreference: this.readPreference,
session
}) as TODO_NODE_3286;
}
}