Commit

node_modules/mongodb/lib/operations/operation.js

@@ -0,0 +1,102 @@
+"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