12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022 |
- "use strict";
- Object.defineProperty(exports, "__esModule", { value: true });
- exports.TimeoutController = exports.request = exports.matchesParentDomain = exports.parseUnsignedInteger = exports.parseInteger = exports.compareObjectId = exports.commandSupportsReadConcern = exports.shuffle = exports.supportsRetryableWrites = exports.enumToString = exports.emitWarningOnce = exports.emitWarning = exports.MONGODB_WARNING_CODE = exports.DEFAULT_PK_FACTORY = exports.HostAddress = exports.BufferPool = exports.List = exports.deepCopy = exports.isRecord = exports.setDifference = exports.isHello = exports.isSuperset = exports.resolveOptions = exports.hasAtomicOperators = exports.calculateDurationInMs = exports.now = exports.makeStateMachine = exports.errorStrictEqual = exports.arrayStrictEqual = exports.eachAsync = exports.maxWireVersion = exports.uuidV4 = exports.maybeCallback = exports.makeCounter = exports.MongoDBCollectionNamespace = exports.MongoDBNamespace = exports.ns = exports.getTopology = exports.decorateWithExplain = exports.decorateWithReadConcern = exports.decorateWithCollation = exports.isPromiseLike = exports.applyRetryableWrites = exports.filterOptions = exports.mergeOptions = exports.isObject = exports.normalizeHintField = exports.hostMatchesWildcards = exports.ByteUtils = void 0;
- const crypto = require("crypto");
- const http = require("http");
- const timers_1 = require("timers");
- const url = require("url");
- const url_1 = require("url");
- const bson_1 = require("./bson");
- const constants_1 = require("./cmap/wire_protocol/constants");
- const constants_2 = require("./constants");
- const error_1 = require("./error");
- const read_concern_1 = require("./read_concern");
- const read_preference_1 = require("./read_preference");
- const common_1 = require("./sdam/common");
- const write_concern_1 = require("./write_concern");
- exports.ByteUtils = {
- toLocalBufferType(buffer) {
- return Buffer.isBuffer(buffer)
- ? buffer
- : Buffer.from(buffer.buffer, buffer.byteOffset, buffer.byteLength);
- },
- equals(seqA, seqB) {
- return exports.ByteUtils.toLocalBufferType(seqA).equals(seqB);
- },
- compare(seqA, seqB) {
- return exports.ByteUtils.toLocalBufferType(seqA).compare(seqB);
- },
- toBase64(uint8array) {
- return exports.ByteUtils.toLocalBufferType(uint8array).toString('base64');
- }
- };
- /**
- * Determines if a connection's address matches a user provided list
- * of domain wildcards.
- */
- function hostMatchesWildcards(host, wildcards) {
- for (const wildcard of wildcards) {
- if (host === wildcard ||
- (wildcard.startsWith('*.') && host?.endsWith(wildcard.substring(2, wildcard.length))) ||
- (wildcard.startsWith('*/') && host?.endsWith(wildcard.substring(2, wildcard.length)))) {
- return true;
- }
- }
- return false;
- }
- exports.hostMatchesWildcards = hostMatchesWildcards;
- /**
- * Ensure Hint field is in a shape we expect:
- * - object of index names mapping to 1 or -1
- * - just an index name
- * @internal
- */
- function normalizeHintField(hint) {
- let finalHint = undefined;
- if (typeof hint === 'string') {
- finalHint = hint;
- }
- else if (Array.isArray(hint)) {
- finalHint = {};
- hint.forEach(param => {
- finalHint[param] = 1;
- });
- }
- else if (hint != null && typeof hint === 'object') {
- finalHint = {};
- for (const name in hint) {
- finalHint[name] = hint[name];
- }
- }
- return finalHint;
- }
- exports.normalizeHintField = normalizeHintField;
- const TO_STRING = (object) => Object.prototype.toString.call(object);
- /**
- * Checks if arg is an Object:
- * - **NOTE**: the check is based on the `[Symbol.toStringTag]() === 'Object'`
- * @internal
- */
- function isObject(arg) {
- return '[object Object]' === TO_STRING(arg);
- }
- exports.isObject = isObject;
- /** @internal */
- function mergeOptions(target, source) {
- return { ...target, ...source };
- }
- exports.mergeOptions = mergeOptions;
- /** @internal */
- function filterOptions(options, names) {
- const filterOptions = {};
- for (const name in options) {
- if (names.includes(name)) {
- filterOptions[name] = options[name];
- }
- }
- // Filtered options
- return filterOptions;
- }
- exports.filterOptions = filterOptions;
- /**
- * Applies retryWrites: true to a command if retryWrites is set on the command's database.
- * @internal
- *
- * @param target - The target command to which we will apply retryWrites.
- * @param db - The database from which we can inherit a retryWrites value.
- */
- function applyRetryableWrites(target, db) {
- if (db && db.s.options?.retryWrites) {
- target.retryWrites = true;
- }
- return target;
- }
- exports.applyRetryableWrites = applyRetryableWrites;
- /**
- * Applies a write concern to a command based on well defined inheritance rules, optionally
- * detecting support for the write concern in the first place.
- * @internal
- *
- * @param target - the target command we will be applying the write concern to
- * @param sources - sources where we can inherit default write concerns from
- * @param options - optional settings passed into a command for write concern overrides
- */
- /**
- * Checks if a given value is a Promise
- *
- * @typeParam T - The resolution type of the possible promise
- * @param value - An object that could be a promise
- * @returns true if the provided value is a Promise
- */
- function isPromiseLike(value) {
- return !!value && typeof value.then === 'function';
- }
- exports.isPromiseLike = isPromiseLike;
- /**
- * Applies collation to a given command.
- * @internal
- *
- * @param command - the command on which to apply collation
- * @param target - target of command
- * @param options - options containing collation settings
- */
- function decorateWithCollation(command, target, options) {
- const capabilities = getTopology(target).capabilities;
- if (options.collation && typeof options.collation === 'object') {
- if (capabilities && capabilities.commandsTakeCollation) {
- command.collation = options.collation;
- }
- else {
- throw new error_1.MongoCompatibilityError(`Current topology does not support collation`);
- }
- }
- }
- exports.decorateWithCollation = decorateWithCollation;
- /**
- * Applies a read concern to a given command.
- * @internal
- *
- * @param command - the command on which to apply the read concern
- * @param coll - the parent collection of the operation calling this method
- */
- function decorateWithReadConcern(command, coll, options) {
- if (options && options.session && options.session.inTransaction()) {
- return;
- }
- const readConcern = Object.assign({}, command.readConcern || {});
- if (coll.s.readConcern) {
- Object.assign(readConcern, coll.s.readConcern);
- }
- if (Object.keys(readConcern).length > 0) {
- Object.assign(command, { readConcern: readConcern });
- }
- }
- exports.decorateWithReadConcern = decorateWithReadConcern;
- /**
- * Applies an explain to a given command.
- * @internal
- *
- * @param command - the command on which to apply the explain
- * @param options - the options containing the explain verbosity
- */
- function decorateWithExplain(command, explain) {
- if (command.explain) {
- return command;
- }
- return { explain: command, verbosity: explain.verbosity };
- }
- exports.decorateWithExplain = decorateWithExplain;
- /**
- * A helper function to get the topology from a given provider. Throws
- * if the topology cannot be found.
- * @throws MongoNotConnectedError
- * @internal
- */
- function getTopology(provider) {
- // MongoClient or ClientSession or AbstractCursor
- if ('topology' in provider && provider.topology) {
- return provider.topology;
- }
- else if ('client' in provider && provider.client.topology) {
- return provider.client.topology;
- }
- throw new error_1.MongoNotConnectedError('MongoClient must be connected to perform this operation');
- }
- exports.getTopology = getTopology;
- /** @internal */
- function ns(ns) {
- return MongoDBNamespace.fromString(ns);
- }
- exports.ns = ns;
- /** @public */
- class MongoDBNamespace {
- /**
- * Create a namespace object
- *
- * @param db - database name
- * @param collection - collection name
- */
- constructor(db, collection) {
- this.db = db;
- this.collection = collection;
- this.collection = collection === '' ? undefined : collection;
- }
- toString() {
- return this.collection ? `${this.db}.${this.collection}` : this.db;
- }
- withCollection(collection) {
- return new MongoDBCollectionNamespace(this.db, collection);
- }
- static fromString(namespace) {
- if (typeof namespace !== 'string' || namespace === '') {
- // TODO(NODE-3483): Replace with MongoNamespaceError
- throw new error_1.MongoRuntimeError(`Cannot parse namespace from "${namespace}"`);
- }
- const [db, ...collectionParts] = namespace.split('.');
- const collection = collectionParts.join('.');
- return new MongoDBNamespace(db, collection === '' ? undefined : collection);
- }
- }
- exports.MongoDBNamespace = MongoDBNamespace;
- /**
- * @public
- *
- * A class representing a collection's namespace. This class enforces (through Typescript) that
- * the `collection` portion of the namespace is defined and should only be
- * used in scenarios where this can be guaranteed.
- */
- class MongoDBCollectionNamespace extends MongoDBNamespace {
- constructor(db, collection) {
- super(db, collection);
- this.collection = collection;
- }
- static fromString(namespace) {
- return super.fromString(namespace);
- }
- }
- exports.MongoDBCollectionNamespace = MongoDBCollectionNamespace;
- /** @internal */
- function* makeCounter(seed = 0) {
- let count = seed;
- while (true) {
- const newCount = count;
- count += 1;
- yield newCount;
- }
- }
- exports.makeCounter = makeCounter;
- function maybeCallback(promiseFn, callback) {
- const promise = promiseFn();
- if (callback == null) {
- return promise;
- }
- promise.then(result => callback(undefined, result), error => callback(error));
- return;
- }
- exports.maybeCallback = maybeCallback;
- /**
- * Synchronously Generate a UUIDv4
- * @internal
- */
- function uuidV4() {
- const result = crypto.randomBytes(16);
- result[6] = (result[6] & 0x0f) | 0x40;
- result[8] = (result[8] & 0x3f) | 0x80;
- return result;
- }
- exports.uuidV4 = uuidV4;
- /**
- * A helper function for determining `maxWireVersion` between legacy and new topology instances
- * @internal
- */
- function maxWireVersion(topologyOrServer) {
- if (topologyOrServer) {
- if (topologyOrServer.loadBalanced) {
- // Since we do not have a monitor, we assume the load balanced server is always
- // pointed at the latest mongodb version. There is a risk that for on-prem
- // deployments that don't upgrade immediately that this could alert to the
- // application that a feature is available that is actually not.
- return constants_1.MAX_SUPPORTED_WIRE_VERSION;
- }
- if (topologyOrServer.hello) {
- return topologyOrServer.hello.maxWireVersion;
- }
- if ('lastHello' in topologyOrServer && typeof topologyOrServer.lastHello === 'function') {
- const lastHello = topologyOrServer.lastHello();
- if (lastHello) {
- return lastHello.maxWireVersion;
- }
- }
- if (topologyOrServer.description &&
- 'maxWireVersion' in topologyOrServer.description &&
- topologyOrServer.description.maxWireVersion != null) {
- return topologyOrServer.description.maxWireVersion;
- }
- }
- return 0;
- }
- exports.maxWireVersion = maxWireVersion;
- /**
- * Applies the function `eachFn` to each item in `arr`, in parallel.
- * @internal
- *
- * @param arr - An array of items to asynchronously iterate over
- * @param eachFn - A function to call on each item of the array. The callback signature is `(item, callback)`, where the callback indicates iteration is complete.
- * @param callback - The callback called after every item has been iterated
- */
- function eachAsync(arr, eachFn, callback) {
- arr = arr || [];
- let idx = 0;
- let awaiting = 0;
- for (idx = 0; idx < arr.length; ++idx) {
- awaiting++;
- eachFn(arr[idx], eachCallback);
- }
- if (awaiting === 0) {
- callback();
- return;
- }
- function eachCallback(err) {
- awaiting--;
- if (err) {
- callback(err);
- return;
- }
- if (idx === arr.length && awaiting <= 0) {
- callback();
- }
- }
- }
- exports.eachAsync = eachAsync;
- /** @internal */
- function arrayStrictEqual(arr, arr2) {
- if (!Array.isArray(arr) || !Array.isArray(arr2)) {
- return false;
- }
- return arr.length === arr2.length && arr.every((elt, idx) => elt === arr2[idx]);
- }
- exports.arrayStrictEqual = arrayStrictEqual;
- /** @internal */
- function errorStrictEqual(lhs, rhs) {
- if (lhs === rhs) {
- return true;
- }
- if (!lhs || !rhs) {
- return lhs === rhs;
- }
- if ((lhs == null && rhs != null) || (lhs != null && rhs == null)) {
- return false;
- }
- if (lhs.constructor.name !== rhs.constructor.name) {
- return false;
- }
- if (lhs.message !== rhs.message) {
- return false;
- }
- return true;
- }
- exports.errorStrictEqual = errorStrictEqual;
- /** @internal */
- function makeStateMachine(stateTable) {
- return function stateTransition(target, newState) {
- const legalStates = stateTable[target.s.state];
- if (legalStates && legalStates.indexOf(newState) < 0) {
- throw new error_1.MongoRuntimeError(`illegal state transition from [${target.s.state}] => [${newState}], allowed: [${legalStates}]`);
- }
- target.emit('stateChanged', target.s.state, newState);
- target.s.state = newState;
- };
- }
- exports.makeStateMachine = makeStateMachine;
- /** @internal */
- function now() {
- const hrtime = process.hrtime();
- return Math.floor(hrtime[0] * 1000 + hrtime[1] / 1000000);
- }
- exports.now = now;
- /** @internal */
- function calculateDurationInMs(started) {
- if (typeof started !== 'number') {
- throw new error_1.MongoInvalidArgumentError('Numeric value required to calculate duration');
- }
- const elapsed = now() - started;
- return elapsed < 0 ? 0 : elapsed;
- }
- exports.calculateDurationInMs = calculateDurationInMs;
- /** @internal */
- function hasAtomicOperators(doc) {
- if (Array.isArray(doc)) {
- for (const document of doc) {
- if (hasAtomicOperators(document)) {
- return true;
- }
- }
- return false;
- }
- const keys = Object.keys(doc);
- return keys.length > 0 && keys[0][0] === '$';
- }
- exports.hasAtomicOperators = hasAtomicOperators;
- /**
- * Merge inherited properties from parent into options, prioritizing values from options,
- * then values from parent.
- * @internal
- */
- function resolveOptions(parent, options) {
- const result = Object.assign({}, options, (0, bson_1.resolveBSONOptions)(options, parent));
- // Users cannot pass a readConcern/writeConcern to operations in a transaction
- const session = options?.session;
- if (!session?.inTransaction()) {
- const readConcern = read_concern_1.ReadConcern.fromOptions(options) ?? parent?.readConcern;
- if (readConcern) {
- result.readConcern = readConcern;
- }
- const writeConcern = write_concern_1.WriteConcern.fromOptions(options) ?? parent?.writeConcern;
- if (writeConcern) {
- result.writeConcern = writeConcern;
- }
- }
- const readPreference = read_preference_1.ReadPreference.fromOptions(options) ?? parent?.readPreference;
- if (readPreference) {
- result.readPreference = readPreference;
- }
- return result;
- }
- exports.resolveOptions = resolveOptions;
- function isSuperset(set, subset) {
- set = Array.isArray(set) ? new Set(set) : set;
- subset = Array.isArray(subset) ? new Set(subset) : subset;
- for (const elem of subset) {
- if (!set.has(elem)) {
- return false;
- }
- }
- return true;
- }
- exports.isSuperset = isSuperset;
- /**
- * Checks if the document is a Hello request
- * @internal
- */
- function isHello(doc) {
- return doc[constants_2.LEGACY_HELLO_COMMAND] || doc.hello ? true : false;
- }
- exports.isHello = isHello;
- /** Returns the items that are uniquely in setA */
- function setDifference(setA, setB) {
- const difference = new Set(setA);
- for (const elem of setB) {
- difference.delete(elem);
- }
- return difference;
- }
- exports.setDifference = setDifference;
- const HAS_OWN = (object, prop) => Object.prototype.hasOwnProperty.call(object, prop);
- function isRecord(value, requiredKeys = undefined) {
- if (!isObject(value)) {
- return false;
- }
- const ctor = value.constructor;
- if (ctor && ctor.prototype) {
- if (!isObject(ctor.prototype)) {
- return false;
- }
- // Check to see if some method exists from the Object exists
- if (!HAS_OWN(ctor.prototype, 'isPrototypeOf')) {
- return false;
- }
- }
- if (requiredKeys) {
- const keys = Object.keys(value);
- return isSuperset(keys, requiredKeys);
- }
- return true;
- }
- exports.isRecord = isRecord;
- /**
- * Make a deep copy of an object
- *
- * NOTE: This is not meant to be the perfect implementation of a deep copy,
- * but instead something that is good enough for the purposes of
- * command monitoring.
- */
- function deepCopy(value) {
- if (value == null) {
- return value;
- }
- else if (Array.isArray(value)) {
- return value.map(item => deepCopy(item));
- }
- else if (isRecord(value)) {
- const res = {};
- for (const key in value) {
- res[key] = deepCopy(value[key]);
- }
- return res;
- }
- const ctor = value.constructor;
- if (ctor) {
- switch (ctor.name.toLowerCase()) {
- case 'date':
- return new ctor(Number(value));
- case 'map':
- return new Map(value);
- case 'set':
- return new Set(value);
- case 'buffer':
- return Buffer.from(value);
- }
- }
- return value;
- }
- exports.deepCopy = deepCopy;
- /**
- * A sequential list of items in a circularly linked list
- * @remarks
- * The head node is special, it is always defined and has a value of null.
- * It is never "included" in the list, in that, it is not returned by pop/shift or yielded by the iterator.
- * The circular linkage and always defined head node are to reduce checks for null next/prev references to zero.
- * New nodes are declared as object literals with keys always in the same order: next, prev, value.
- * @internal
- */
- class List {
- get length() {
- return this.count;
- }
- get [Symbol.toStringTag]() {
- return 'List';
- }
- constructor() {
- this.count = 0;
- // this is carefully crafted:
- // declaring a complete and consistently key ordered
- // object is beneficial to the runtime optimizations
- this.head = {
- next: null,
- prev: null,
- value: null
- };
- this.head.next = this.head;
- this.head.prev = this.head;
- }
- toArray() {
- return Array.from(this);
- }
- toString() {
- return `head <=> ${this.toArray().join(' <=> ')} <=> head`;
- }
- *[Symbol.iterator]() {
- for (const node of this.nodes()) {
- yield node.value;
- }
- }
- *nodes() {
- let ptr = this.head.next;
- while (ptr !== this.head) {
- // Save next before yielding so that we make removing within iteration safe
- const { next } = ptr;
- yield ptr;
- ptr = next;
- }
- }
- /** Insert at end of list */
- push(value) {
- this.count += 1;
- const newNode = {
- next: this.head,
- prev: this.head.prev,
- value
- };
- this.head.prev.next = newNode;
- this.head.prev = newNode;
- }
- /** Inserts every item inside an iterable instead of the iterable itself */
- pushMany(iterable) {
- for (const value of iterable) {
- this.push(value);
- }
- }
- /** Insert at front of list */
- unshift(value) {
- this.count += 1;
- const newNode = {
- next: this.head.next,
- prev: this.head,
- value
- };
- this.head.next.prev = newNode;
- this.head.next = newNode;
- }
- remove(node) {
- if (node === this.head || this.length === 0) {
- return null;
- }
- this.count -= 1;
- const prevNode = node.prev;
- const nextNode = node.next;
- prevNode.next = nextNode;
- nextNode.prev = prevNode;
- return node.value;
- }
- /** Removes the first node at the front of the list */
- shift() {
- return this.remove(this.head.next);
- }
- /** Removes the last node at the end of the list */
- pop() {
- return this.remove(this.head.prev);
- }
- /** Iterates through the list and removes nodes where filter returns true */
- prune(filter) {
- for (const node of this.nodes()) {
- if (filter(node.value)) {
- this.remove(node);
- }
- }
- }
- clear() {
- this.count = 0;
- this.head.next = this.head;
- this.head.prev = this.head;
- }
- /** Returns the first item in the list, does not remove */
- first() {
- // If the list is empty, value will be the head's null
- return this.head.next.value;
- }
- /** Returns the last item in the list, does not remove */
- last() {
- // If the list is empty, value will be the head's null
- return this.head.prev.value;
- }
- }
- exports.List = List;
- /**
- * A pool of Buffers which allow you to read them as if they were one
- * @internal
- */
- class BufferPool {
- constructor() {
- this.buffers = new List();
- this.totalByteLength = 0;
- }
- get length() {
- return this.totalByteLength;
- }
- /** Adds a buffer to the internal buffer pool list */
- append(buffer) {
- this.buffers.push(buffer);
- this.totalByteLength += buffer.length;
- }
- /**
- * If BufferPool contains 4 bytes or more construct an int32 from the leading bytes,
- * otherwise return null. Size can be negative, caller should error check.
- */
- getInt32() {
- if (this.totalByteLength < 4) {
- return null;
- }
- const firstBuffer = this.buffers.first();
- if (firstBuffer != null && firstBuffer.byteLength >= 4) {
- return firstBuffer.readInt32LE(0);
- }
- // Unlikely case: an int32 is split across buffers.
- // Use read and put the returned buffer back on top
- const top4Bytes = this.read(4);
- const value = top4Bytes.readInt32LE(0);
- // Put it back.
- this.totalByteLength += 4;
- this.buffers.unshift(top4Bytes);
- return value;
- }
- /** Reads the requested number of bytes, optionally consuming them */
- read(size) {
- if (typeof size !== 'number' || size < 0) {
- throw new error_1.MongoInvalidArgumentError('Argument "size" must be a non-negative number');
- }
- // oversized request returns empty buffer
- if (size > this.totalByteLength) {
- return Buffer.alloc(0);
- }
- // We know we have enough, we just don't know how it is spread across chunks
- // TODO(NODE-4732): alloc API should change based on raw option
- const result = Buffer.allocUnsafe(size);
- for (let bytesRead = 0; bytesRead < size;) {
- const buffer = this.buffers.shift();
- if (buffer == null) {
- break;
- }
- const bytesRemaining = size - bytesRead;
- const bytesReadable = Math.min(bytesRemaining, buffer.byteLength);
- const bytes = buffer.subarray(0, bytesReadable);
- result.set(bytes, bytesRead);
- bytesRead += bytesReadable;
- this.totalByteLength -= bytesReadable;
- if (bytesReadable < buffer.byteLength) {
- this.buffers.unshift(buffer.subarray(bytesReadable));
- }
- }
- return result;
- }
- }
- exports.BufferPool = BufferPool;
- /** @public */
- class HostAddress {
- constructor(hostString) {
- this.host = undefined;
- this.port = undefined;
- this.socketPath = undefined;
- this.isIPv6 = false;
- const escapedHost = hostString.split(' ').join('%20'); // escape spaces, for socket path hosts
- if (escapedHost.endsWith('.sock')) {
- // heuristically determine if we're working with a domain socket
- this.socketPath = decodeURIComponent(escapedHost);
- return;
- }
- const urlString = `iLoveJS://${escapedHost}`;
- let url;
- try {
- url = new url_1.URL(urlString);
- }
- catch (urlError) {
- const runtimeError = new error_1.MongoRuntimeError(`Unable to parse ${escapedHost} with URL`);
- runtimeError.cause = urlError;
- throw runtimeError;
- }
- const hostname = url.hostname;
- const port = url.port;
- let normalized = decodeURIComponent(hostname).toLowerCase();
- if (normalized.startsWith('[') && normalized.endsWith(']')) {
- this.isIPv6 = true;
- normalized = normalized.substring(1, hostname.length - 1);
- }
- this.host = normalized.toLowerCase();
- if (typeof port === 'number') {
- this.port = port;
- }
- else if (typeof port === 'string' && port !== '') {
- this.port = Number.parseInt(port, 10);
- }
- else {
- this.port = 27017;
- }
- if (this.port === 0) {
- throw new error_1.MongoParseError('Invalid port (zero) with hostname');
- }
- Object.freeze(this);
- }
- [Symbol.for('nodejs.util.inspect.custom')]() {
- return this.inspect();
- }
- inspect() {
- return `new HostAddress('${this.toString()}')`;
- }
- toString() {
- if (typeof this.host === 'string') {
- if (this.isIPv6) {
- return `[${this.host}]:${this.port}`;
- }
- return `${this.host}:${this.port}`;
- }
- return `${this.socketPath}`;
- }
- static fromString(s) {
- return new HostAddress(s);
- }
- static fromHostPort(host, port) {
- if (host.includes(':')) {
- host = `[${host}]`; // IPv6 address
- }
- return HostAddress.fromString(`${host}:${port}`);
- }
- static fromSrvRecord({ name, port }) {
- return HostAddress.fromHostPort(name, port);
- }
- toHostPort() {
- if (this.socketPath) {
- return { host: this.socketPath, port: 0 };
- }
- const host = this.host ?? '';
- const port = this.port ?? 0;
- return { host, port };
- }
- }
- exports.HostAddress = HostAddress;
- exports.DEFAULT_PK_FACTORY = {
- // We prefer not to rely on ObjectId having a createPk method
- createPk() {
- return new bson_1.ObjectId();
- }
- };
- /**
- * When the driver used emitWarning the code will be equal to this.
- * @public
- *
- * @example
- * ```ts
- * process.on('warning', (warning) => {
- * if (warning.code === MONGODB_WARNING_CODE) console.error('Ah an important warning! :)')
- * })
- * ```
- */
- exports.MONGODB_WARNING_CODE = 'MONGODB DRIVER';
- /** @internal */
- function emitWarning(message) {
- return process.emitWarning(message, { code: exports.MONGODB_WARNING_CODE });
- }
- exports.emitWarning = emitWarning;
- const emittedWarnings = new Set();
- /**
- * Will emit a warning once for the duration of the application.
- * Uses the message to identify if it has already been emitted
- * so using string interpolation can cause multiple emits
- * @internal
- */
- function emitWarningOnce(message) {
- if (!emittedWarnings.has(message)) {
- emittedWarnings.add(message);
- return emitWarning(message);
- }
- }
- exports.emitWarningOnce = emitWarningOnce;
- /**
- * Takes a JS object and joins the values into a string separated by ', '
- */
- function enumToString(en) {
- return Object.values(en).join(', ');
- }
- exports.enumToString = enumToString;
- /**
- * Determine if a server supports retryable writes.
- *
- * @internal
- */
- function supportsRetryableWrites(server) {
- if (!server) {
- return false;
- }
- if (server.loadBalanced) {
- // Loadbalanced topologies will always support retry writes
- return true;
- }
- if (server.description.logicalSessionTimeoutMinutes != null) {
- // that supports sessions
- if (server.description.type !== common_1.ServerType.Standalone) {
- // and that is not a standalone
- return true;
- }
- }
- return false;
- }
- exports.supportsRetryableWrites = supportsRetryableWrites;
- /**
- * Fisher–Yates Shuffle
- *
- * Reference: https://bost.ocks.org/mike/shuffle/
- * @param sequence - items to be shuffled
- * @param limit - Defaults to `0`. If nonzero shuffle will slice the randomized array e.g, `.slice(0, limit)` otherwise will return the entire randomized array.
- */
- function shuffle(sequence, limit = 0) {
- const items = Array.from(sequence); // shallow copy in order to never shuffle the input
- if (limit > items.length) {
- throw new error_1.MongoRuntimeError('Limit must be less than the number of items');
- }
- let remainingItemsToShuffle = items.length;
- const lowerBound = limit % items.length === 0 ? 1 : items.length - limit;
- while (remainingItemsToShuffle > lowerBound) {
- // Pick a remaining element
- const randomIndex = Math.floor(Math.random() * remainingItemsToShuffle);
- remainingItemsToShuffle -= 1;
- // And swap it with the current element
- const swapHold = items[remainingItemsToShuffle];
- items[remainingItemsToShuffle] = items[randomIndex];
- items[randomIndex] = swapHold;
- }
- return limit % items.length === 0 ? items : items.slice(lowerBound);
- }
- exports.shuffle = shuffle;
- // TODO(NODE-4936): read concern eligibility for commands should be codified in command construction
- // @see https://github.com/mongodb/specifications/blob/master/source/read-write-concern/read-write-concern.rst#read-concern
- function commandSupportsReadConcern(command) {
- if (command.aggregate || command.count || command.distinct || command.find || command.geoNear) {
- return true;
- }
- return false;
- }
- exports.commandSupportsReadConcern = commandSupportsReadConcern;
- /**
- * Compare objectIds. `null` is always less
- * - `+1 = oid1 is greater than oid2`
- * - `-1 = oid1 is less than oid2`
- * - `+0 = oid1 is equal oid2`
- */
- function compareObjectId(oid1, oid2) {
- if (oid1 == null && oid2 == null) {
- return 0;
- }
- if (oid1 == null) {
- return -1;
- }
- if (oid2 == null) {
- return 1;
- }
- return exports.ByteUtils.compare(oid1.id, oid2.id);
- }
- exports.compareObjectId = compareObjectId;
- function parseInteger(value) {
- if (typeof value === 'number')
- return Math.trunc(value);
- const parsedValue = Number.parseInt(String(value), 10);
- return Number.isNaN(parsedValue) ? null : parsedValue;
- }
- exports.parseInteger = parseInteger;
- function parseUnsignedInteger(value) {
- const parsedInt = parseInteger(value);
- return parsedInt != null && parsedInt >= 0 ? parsedInt : null;
- }
- exports.parseUnsignedInteger = parseUnsignedInteger;
- /**
- * Determines whether a provided address matches the provided parent domain.
- *
- * If a DNS server were to become compromised SRV records would still need to
- * advertise addresses that are under the same domain as the srvHost.
- *
- * @param address - The address to check against a domain
- * @param srvHost - The domain to check the provided address against
- * @returns Whether the provided address matches the parent domain
- */
- function matchesParentDomain(address, srvHost) {
- // Remove trailing dot if exists on either the resolved address or the srv hostname
- const normalizedAddress = address.endsWith('.') ? address.slice(0, address.length - 1) : address;
- const normalizedSrvHost = srvHost.endsWith('.') ? srvHost.slice(0, srvHost.length - 1) : srvHost;
- const allCharacterBeforeFirstDot = /^.*?\./;
- // Remove all characters before first dot
- // Add leading dot back to string so
- // an srvHostDomain = '.trusted.site'
- // will not satisfy an addressDomain that endsWith '.fake-trusted.site'
- const addressDomain = `.${normalizedAddress.replace(allCharacterBeforeFirstDot, '')}`;
- const srvHostDomain = `.${normalizedSrvHost.replace(allCharacterBeforeFirstDot, '')}`;
- return addressDomain.endsWith(srvHostDomain);
- }
- exports.matchesParentDomain = matchesParentDomain;
- async function request(uri, options = {}) {
- return new Promise((resolve, reject) => {
- const requestOptions = {
- method: 'GET',
- timeout: 10000,
- json: true,
- ...url.parse(uri),
- ...options
- };
- const req = http.request(requestOptions, res => {
- res.setEncoding('utf8');
- let data = '';
- res.on('data', d => {
- data += d;
- });
- res.once('end', () => {
- if (options.json === false) {
- resolve(data);
- return;
- }
- try {
- const parsed = JSON.parse(data);
- resolve(parsed);
- }
- catch {
- // TODO(NODE-3483)
- reject(new error_1.MongoRuntimeError(`Invalid JSON response: "${data}"`));
- }
- });
- });
- req.once('timeout', () => req.destroy(new error_1.MongoNetworkTimeoutError(`Network request to ${uri} timed out after ${options.timeout} ms`)));
- req.once('error', error => reject(error));
- req.end();
- });
- }
- exports.request = request;
- /**
- * A custom AbortController that aborts after a specified timeout.
- *
- * If `timeout` is undefined or \<=0, the abort controller never aborts.
- *
- * This class provides two benefits over the built-in AbortSignal.timeout() method.
- * - This class provides a mechanism for cancelling the timeout
- * - This class supports infinite timeouts by interpreting a timeout of 0 as infinite. This is
- * consistent with existing timeout options in the Node driver (serverSelectionTimeoutMS, for example).
- * @internal
- */
- class TimeoutController extends AbortController {
- constructor(timeout = 0, timeoutId = timeout > 0 ? (0, timers_1.setTimeout)(() => this.abort(), timeout) : null) {
- super();
- this.timeoutId = timeoutId;
- }
- clear() {
- if (this.timeoutId != null) {
- (0, timers_1.clearTimeout)(this.timeoutId);
- }
- this.timeoutId = null;
- }
- }
- exports.TimeoutController = TimeoutController;
- //# sourceMappingURL=utils.js.map
|