102 lines
3.6 KiB
JavaScript
102 lines
3.6 KiB
JavaScript
"use strict";
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.AbstractOperation = exports.Aspect = void 0;
|
|
exports.defineAspects = defineAspects;
|
|
const bson_1 = require("../bson");
|
|
const read_preference_1 = require("../read_preference");
|
|
exports.Aspect = {
|
|
READ_OPERATION: Symbol('READ_OPERATION'),
|
|
WRITE_OPERATION: Symbol('WRITE_OPERATION'),
|
|
RETRYABLE: Symbol('RETRYABLE'),
|
|
EXPLAINABLE: Symbol('EXPLAINABLE'),
|
|
SKIP_COLLATION: Symbol('SKIP_COLLATION'),
|
|
CURSOR_CREATING: Symbol('CURSOR_CREATING'),
|
|
MUST_SELECT_SAME_SERVER: Symbol('MUST_SELECT_SAME_SERVER'),
|
|
COMMAND_BATCHING: Symbol('COMMAND_BATCHING'),
|
|
SUPPORTS_RAW_DATA: Symbol('SUPPORTS_RAW_DATA')
|
|
};
|
|
/**
|
|
* This class acts as a parent class for any operation and is responsible for setting this.options,
|
|
* as well as setting and getting a session.
|
|
* Additionally, this class implements `hasAspect`, which determines whether an operation has
|
|
* a specific aspect.
|
|
* @internal
|
|
*/
|
|
class AbstractOperation {
|
|
constructor(options = {}) {
|
|
this.readPreference = this.hasAspect(exports.Aspect.WRITE_OPERATION)
|
|
? read_preference_1.ReadPreference.primary
|
|
: (read_preference_1.ReadPreference.fromOptions(options) ?? read_preference_1.ReadPreference.primary);
|
|
// Pull the BSON serialize options from the already-resolved options
|
|
this.bsonOptions = (0, bson_1.resolveBSONOptions)(options);
|
|
this._session = options.session != null ? options.session : undefined;
|
|
this.options = options;
|
|
this.bypassPinningCheck = !!options.bypassPinningCheck;
|
|
}
|
|
hasAspect(aspect) {
|
|
const ctor = this.constructor;
|
|
if (ctor.aspects == null) {
|
|
return false;
|
|
}
|
|
return ctor.aspects.has(aspect);
|
|
}
|
|
// Make sure the session is not writable from outside this class.
|
|
get session() {
|
|
return this._session;
|
|
}
|
|
set session(session) {
|
|
this._session = session;
|
|
}
|
|
clearSession() {
|
|
this._session = undefined;
|
|
}
|
|
resetBatch() {
|
|
return true;
|
|
}
|
|
get canRetryRead() {
|
|
return this.hasAspect(exports.Aspect.RETRYABLE) && this.hasAspect(exports.Aspect.READ_OPERATION);
|
|
}
|
|
get canRetryWrite() {
|
|
return this.hasAspect(exports.Aspect.RETRYABLE) && this.hasAspect(exports.Aspect.WRITE_OPERATION);
|
|
}
|
|
/**
|
|
* Given an instance of a MongoDBResponse, map the response to the correct result type. For
|
|
* example, a `CountOperation` might map the response as follows:
|
|
*
|
|
* ```typescript
|
|
* override handleOk(response: InstanceType<typeof this.SERVER_COMMAND_RESPONSE_TYPE>): TResult {
|
|
* return response.toObject(this.bsonOptions).n ?? 0;
|
|
* }
|
|
*
|
|
* // or, with type safety:
|
|
* override handleOk(response: InstanceType<typeof this.SERVER_COMMAND_RESPONSE_TYPE>): TResult {
|
|
* return response.getNumber('n') ?? 0;
|
|
* }
|
|
* ```
|
|
*/
|
|
handleOk(response) {
|
|
return response.toObject(this.bsonOptions);
|
|
}
|
|
/**
|
|
* Optional.
|
|
*
|
|
* If the operation performs error handling, such as wrapping, renaming the error, or squashing errors
|
|
* this method can be overridden.
|
|
*/
|
|
handleError(error) {
|
|
throw error;
|
|
}
|
|
}
|
|
exports.AbstractOperation = AbstractOperation;
|
|
function defineAspects(operation, aspects) {
|
|
if (!Array.isArray(aspects) && !(aspects instanceof Set)) {
|
|
aspects = [aspects];
|
|
}
|
|
aspects = new Set(aspects);
|
|
Object.defineProperty(operation, 'aspects', {
|
|
value: aspects,
|
|
writable: false
|
|
});
|
|
return aspects;
|
|
}
|
|
//# sourceMappingURL=operation.js.map
|