@skybloxsystems/ticket-bot
Version:
110 lines • 4.74 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.ListCollectionsCursor = exports.ListCollectionsOperation = void 0;
const command_1 = require("./command");
const operation_1 = require("./operation");
const utils_1 = require("../utils");
const CONSTANTS = require("../constants");
const abstract_cursor_1 = require("../cursor/abstract_cursor");
const execute_operation_1 = require("./execute_operation");
const LIST_COLLECTIONS_WIRE_VERSION = 3;
/** @internal */
class ListCollectionsOperation extends command_1.CommandOperation {
constructor(db, filter, options) {
super(db, options);
this.options = options !== null && options !== void 0 ? options : {};
this.db = db;
this.filter = filter;
this.nameOnly = !!this.options.nameOnly;
this.authorizedCollections = !!this.options.authorizedCollections;
if (typeof this.options.batchSize === 'number') {
this.batchSize = this.options.batchSize;
}
}
execute(server, session, callback) {
if ((0, utils_1.maxWireVersion)(server) < LIST_COLLECTIONS_WIRE_VERSION) {
let filter = this.filter;
const databaseName = this.db.s.namespace.db;
// If we have legacy mode and have not provided a full db name filter it
if (typeof filter.name === 'string' && !new RegExp(`^${databaseName}\\.`).test(filter.name)) {
filter = Object.assign({}, filter);
filter.name = this.db.s.namespace.withCollection(filter.name).toString();
}
// No filter, filter by current database
if (filter == null) {
filter = { name: `/${databaseName}/` };
}
// Rewrite the filter to use $and to filter out indexes
if (filter.name) {
filter = { $and: [{ name: filter.name }, { name: /^((?!\$).)*$/ }] };
}
else {
filter = { name: /^((?!\$).)*$/ };
}
const documentTransform = (doc) => {
const matching = `${databaseName}.`;
const index = doc.name.indexOf(matching);
// Remove database name if available
if (doc.name && index === 0) {
doc.name = doc.name.substr(index + matching.length);
}
return doc;
};
server.query(new utils_1.MongoDBNamespace(databaseName, CONSTANTS.SYSTEM_NAMESPACE_COLLECTION), { query: filter }, { batchSize: this.batchSize || 1000, readPreference: this.readPreference }, (err, result) => {
if (result && result.documents && Array.isArray(result.documents)) {
result.documents = result.documents.map(documentTransform);
}
callback(err, result);
});
return;
}
return super.executeCommand(server, session, this.generateCommand(), callback);
}
/* This is here for the purpose of unit testing the final command that gets sent. */
generateCommand() {
return {
listCollections: 1,
filter: this.filter,
cursor: this.batchSize ? { batchSize: this.batchSize } : {},
nameOnly: this.nameOnly,
authorizedCollections: this.authorizedCollections
};
}
}
exports.ListCollectionsOperation = ListCollectionsOperation;
/** @public */
class ListCollectionsCursor extends abstract_cursor_1.AbstractCursor {
constructor(db, filter, options) {
super((0, utils_1.getTopology)(db), db.s.namespace, options);
this.parent = db;
this.filter = filter;
this.options = options;
}
clone() {
return new ListCollectionsCursor(this.parent, this.filter, {
...this.options,
...this.cursorOptions
});
}
/** @internal */
_initialize(session, callback) {
const operation = new ListCollectionsOperation(this.parent, this.filter, {
...this.cursorOptions,
...this.options,
session
});
(0, execute_operation_1.executeOperation)((0, utils_1.getTopology)(this.parent), operation, (err, response) => {
if (err || response == null)
return callback(err);
// TODO: NODE-2882
callback(undefined, { server: operation.server, session, response });
});
}
}
exports.ListCollectionsCursor = ListCollectionsCursor;
(0, operation_1.defineAspects)(ListCollectionsOperation, [
operation_1.Aspect.READ_OPERATION,
operation_1.Aspect.RETRYABLE,
operation_1.Aspect.CURSOR_CREATING
]);
//# sourceMappingURL=list_collections.js.map