You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
1.9 KiB
69 lines
1.9 KiB
import type { Document } from '../bson';
|
|
import type { Collection } from '../collection';
|
|
import type { Server } from '../sdam/server';
|
|
import type { ClientSession } from '../sessions';
|
|
import type { Callback, MongoDBNamespace } from '../utils';
|
|
import { CommandOperation, CommandOperationOptions } from './command';
|
|
import { Aspect, defineAspects } from './operation';
|
|
|
|
/** @public */
|
|
export interface CountOptions extends CommandOperationOptions {
|
|
/** The number of documents to skip. */
|
|
skip?: number;
|
|
/** The maximum amounts to count before aborting. */
|
|
limit?: number;
|
|
/** Number of milliseconds to wait before aborting the query. */
|
|
maxTimeMS?: number;
|
|
/** An index name hint for the query. */
|
|
hint?: string | Document;
|
|
}
|
|
|
|
/** @internal */
|
|
export class CountOperation extends CommandOperation<number> {
|
|
override options: CountOptions;
|
|
collectionName?: string;
|
|
query: Document;
|
|
|
|
constructor(namespace: MongoDBNamespace, filter: Document, options: CountOptions) {
|
|
super({ s: { namespace: namespace } } as unknown as Collection, options);
|
|
|
|
this.options = options;
|
|
this.collectionName = namespace.collection;
|
|
this.query = filter;
|
|
}
|
|
|
|
override execute(
|
|
server: Server,
|
|
session: ClientSession | undefined,
|
|
callback: Callback<number>
|
|
): void {
|
|
const options = this.options;
|
|
const cmd: Document = {
|
|
count: this.collectionName,
|
|
query: this.query
|
|
};
|
|
|
|
if (typeof options.limit === 'number') {
|
|
cmd.limit = options.limit;
|
|
}
|
|
|
|
if (typeof options.skip === 'number') {
|
|
cmd.skip = options.skip;
|
|
}
|
|
|
|
if (options.hint != null) {
|
|
cmd.hint = options.hint;
|
|
}
|
|
|
|
if (typeof options.maxTimeMS === 'number') {
|
|
cmd.maxTimeMS = options.maxTimeMS;
|
|
}
|
|
|
|
super.executeCommand(server, session, cmd, (err, result) => {
|
|
callback(err, result ? result.n : 0);
|
|
});
|
|
}
|
|
}
|
|
|
|
defineAspects(CountOperation, [Aspect.READ_OPERATION, Aspect.RETRYABLE]);
|