create_collection.js 4.6 KB

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