something is working
This commit is contained in:
+68
@@ -0,0 +1,68 @@
|
||||
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]);
|
||||
Reference in New Issue
Block a user