create_collection.js 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. exports.CreateCollectionOperation = void 0;
  4. exports.createCollections = createCollections;
  5. const constants_1 = require("../cmap/wire_protocol/constants");
  6. const responses_1 = require("../cmap/wire_protocol/responses");
  7. const collection_1 = require("../collection");
  8. const error_1 = require("../error");
  9. const timeout_1 = require("../timeout");
  10. const utils_1 = require("../utils");
  11. const command_1 = require("./command");
  12. const execute_operation_1 = require("./execute_operation");
  13. const indexes_1 = require("./indexes");
  14. const operation_1 = require("./operation");
  15. const ILLEGAL_COMMAND_FIELDS = new Set([
  16. 'w',
  17. 'wtimeout',
  18. 'timeoutMS',
  19. 'j',
  20. 'fsync',
  21. 'pkFactory',
  22. 'raw',
  23. 'readPreference',
  24. 'session',
  25. 'readConcern',
  26. 'writeConcern',
  27. 'raw',
  28. 'fieldsAsRaw',
  29. 'useBigInt64',
  30. 'promoteLongs',
  31. 'promoteValues',
  32. 'promoteBuffers',
  33. 'bsonRegExp',
  34. 'serializeFunctions',
  35. 'ignoreUndefined',
  36. 'enableUtf8Validation'
  37. ]);
  38. /* @internal */
  39. const INVALID_QE_VERSION = 'Driver support of Queryable Encryption is incompatible with server. Upgrade server to use Queryable Encryption.';
  40. /** @internal */
  41. class CreateCollectionOperation extends command_1.CommandOperation {
  42. constructor(db, name, options = {}) {
  43. super(db, options);
  44. this.SERVER_COMMAND_RESPONSE_TYPE = responses_1.MongoDBResponse;
  45. this.options = options;
  46. this.db = db;
  47. this.name = name;
  48. }
  49. get commandName() {
  50. return 'create';
  51. }
  52. buildCommandDocument(_connection, _session) {
  53. const isOptionValid = ([k, v]) => v != null && typeof v !== 'function' && !ILLEGAL_COMMAND_FIELDS.has(k);
  54. return {
  55. create: this.name,
  56. ...Object.fromEntries(Object.entries(this.options).filter(isOptionValid))
  57. };
  58. }
  59. handleOk(_response) {
  60. return new collection_1.Collection(this.db, this.name, this.options);
  61. }
  62. }
  63. exports.CreateCollectionOperation = CreateCollectionOperation;
  64. async function createCollections(db, name, options) {
  65. const timeoutContext = timeout_1.TimeoutContext.create({
  66. session: options.session,
  67. serverSelectionTimeoutMS: db.client.s.options.serverSelectionTimeoutMS,
  68. waitQueueTimeoutMS: db.client.s.options.waitQueueTimeoutMS,
  69. timeoutMS: options.timeoutMS
  70. });
  71. const encryptedFields = options.encryptedFields ??
  72. db.client.s.options.autoEncryption?.encryptedFieldsMap?.[`${db.databaseName}.${name}`];
  73. if (encryptedFields) {
  74. class CreateSupportingFLEv2CollectionOperation extends CreateCollectionOperation {
  75. buildCommandDocument(connection, session) {
  76. if (!connection.description.loadBalanced &&
  77. (0, utils_1.maxWireVersion)(connection) < constants_1.MIN_SUPPORTED_QE_WIRE_VERSION) {
  78. throw new error_1.MongoCompatibilityError(`${INVALID_QE_VERSION} The minimum server version required is ${constants_1.MIN_SUPPORTED_QE_SERVER_VERSION}`);
  79. }
  80. return super.buildCommandDocument(connection, session);
  81. }
  82. }
  83. // Create auxilliary collections for queryable encryption support.
  84. const escCollection = encryptedFields.escCollection ?? `enxcol_.${name}.esc`;
  85. const ecocCollection = encryptedFields.ecocCollection ?? `enxcol_.${name}.ecoc`;
  86. for (const collectionName of [escCollection, ecocCollection]) {
  87. const createOp = new CreateSupportingFLEv2CollectionOperation(db, collectionName, {
  88. clusteredIndex: {
  89. key: { _id: 1 },
  90. unique: true
  91. },
  92. session: options.session
  93. });
  94. await (0, execute_operation_1.executeOperation)(db.client, createOp, timeoutContext);
  95. }
  96. if (!options.encryptedFields) {
  97. options = { ...options, encryptedFields };
  98. }
  99. }
  100. const coll = await (0, execute_operation_1.executeOperation)(db.client, new CreateCollectionOperation(db, name, options), timeoutContext);
  101. if (encryptedFields) {
  102. // Create the required index for queryable encryption support.
  103. const createIndexOp = indexes_1.CreateIndexesOperation.fromIndexSpecification(db, name, { __safeContent__: 1 }, { session: options.session });
  104. await (0, execute_operation_1.executeOperation)(db.client, createIndexOp, timeoutContext);
  105. }
  106. return coll;
  107. }
  108. (0, operation_1.defineAspects)(CreateCollectionOperation, [operation_1.Aspect.WRITE_OPERATION]);
  109. //# sourceMappingURL=create_collection.js.map