@mikro-orm/core 7.0.2 → 7.0.3-dev.0

package/unit-of-work/ChangeSetPersister.js

@@ -8,468 +8,447 @@ import { OptimisticLockError, ValidationError } from '../errors.js';
 import { ReferenceKind } from '../enums.js';
 /** @internal Executes change sets against the database, handling inserts, updates, and deletes. */
 export class ChangeSetPersister {
-  #platform;
-  #comparator;
-  #usesReturningStatement;
-  #driver;
-  #metadata;
-  #hydrator;
-  #factory;
-  #config;
-  #em;
-  constructor(em) {
-    this.#em = em;
-    this.#driver = this.#em.getDriver();
-    this.#config = this.#em.config;
-    this.#metadata = this.#em.getMetadata();
-    this.#factory = this.#em.getEntityFactory();
-    this.#platform = this.#driver.getPlatform();
-    this.#hydrator = this.#config.getHydrator(this.#metadata);
-    this.#comparator = this.#config.getComparator(this.#metadata);
-    this.#usesReturningStatement = this.#platform.usesReturningStatement() || this.#platform.usesOutputStatement();
-  }
-  /** Executes all pending INSERT change sets, using batch inserts when possible. */
-  async executeInserts(changeSets, options, withSchema) {
-    if (!withSchema) {
-      return this.runForEachSchema(changeSets, 'executeInserts', options);
-    }
-    const meta = changeSets[0].meta;
-    changeSets.forEach(changeSet => this.processProperties(changeSet));
-    if (changeSets.length > 1 && this.#config.get('useBatchInserts', this.#platform.usesBatchInserts())) {
-      return this.persistNewEntities(meta, changeSets, options);
-    }
-    for (const changeSet of changeSets) {
-      await this.persistNewEntity(meta, changeSet, options);
-    }
-  }
-  /** Executes all pending UPDATE change sets, using batch updates when possible. */
-  async executeUpdates(changeSets, batched, options, withSchema) {
-    if (!withSchema) {
-      return this.runForEachSchema(changeSets, 'executeUpdates', options, batched);
-    }
-    const meta = changeSets[0].meta;
-    changeSets.forEach(changeSet => this.processProperties(changeSet));
-    // For STI with conflicting fieldNames (renamedFrom properties), we can't batch mixed child types
-    const hasSTIConflicts = meta.root.props.some(p => p.renamedFrom);
-    const hasMixedTypes = hasSTIConflicts && changeSets.some(cs => cs.meta.class !== meta.class);
-    if (
-      batched &&
-      changeSets.length > 1 &&
-      !hasMixedTypes &&
-      this.#config.get('useBatchUpdates', this.#platform.usesBatchUpdates())
-    ) {
-      return this.persistManagedEntities(meta, changeSets, options);
-    }
-    for (const changeSet of changeSets) {
-      await this.persistManagedEntity(changeSet, options);
-    }
-  }
-  /** Executes all pending DELETE change sets in batches. */
-  async executeDeletes(changeSets, options, withSchema) {
-    if (!withSchema) {
-      return this.runForEachSchema(changeSets, 'executeDeletes', options);
-    }
-    const size = this.#config.get('batchSize');
-    const meta = changeSets[0].meta;
-    const pk = Utils.getPrimaryKeyHash(meta.primaryKeys);
-    for (let i = 0; i < changeSets.length; i += size) {
-      const chunk = changeSets.slice(i, i + size);
-      const pks = chunk.map(cs => cs.getPrimaryKey());
-      options = this.prepareOptions(meta, options);
-      await this.#driver.nativeDelete(meta.root.class, { [pk]: { $in: pks } }, options);
-    }
-  }
-  async runForEachSchema(changeSets, method, options, ...args) {
-    const groups = new Map();
-    changeSets.forEach(cs => {
-      const group = groups.get(cs.schema) ?? [];
-      group.push(cs);
-      groups.set(cs.schema, group);
-    });
-    for (const [key, group] of groups.entries()) {
-      options = { ...options, schema: key };
-      // @ts-ignore
-      await this[method](group, ...args, options, true);
-    }
-  }
-  validateRequired(entity) {
-    const wrapped = helper(entity);
-    for (const prop of wrapped.__meta.props) {
-      if (
-        !prop.nullable &&
-        !prop.autoincrement &&
-        !prop.default &&
-        !prop.defaultRaw &&
-        !prop.onCreate &&
-        !prop.generated &&
-        !prop.embedded &&
-        ![ReferenceKind.ONE_TO_MANY, ReferenceKind.MANY_TO_MANY].includes(prop.kind) &&
-        prop.name !== wrapped.__meta.root.discriminatorColumn &&
-        prop.type !== 'ObjectId' &&
-        prop.persist !== false &&
-        entity[prop.name] == null
-      ) {
-        throw ValidationError.propertyRequired(entity, prop);
-      }
-    }
-  }
-  processProperties(changeSet) {
-    const meta = changeSet.meta;
-    for (const prop of meta.relations) {
-      this.processProperty(changeSet, prop);
-    }
-    if (changeSet.type === ChangeSetType.CREATE && this.#config.get('validateRequired')) {
-      this.validateRequired(changeSet.entity);
-    }
-  }
-  async persistNewEntity(meta, changeSet, options) {
-    const wrapped = helper(changeSet.entity);
-    options = this.prepareOptions(meta, options, {
-      convertCustomTypes: false,
-    });
-    this.resolveTPTIdentifiers(changeSet);
-    // Use changeSet's own meta for STI entities to get correct field mappings
-    const res = await this.#driver.nativeInsertMany(changeSet.meta.class, [changeSet.payload], options);
-    if (!wrapped.hasPrimaryKey()) {
-      this.mapPrimaryKey(meta, res.insertId ?? res.row?.[meta.primaryKeys[0]], changeSet);
-    }
-    this.mapReturnedValues(changeSet.entity, changeSet.payload, res.row, meta);
-    this.markAsPopulated(changeSet, meta);
-    wrapped.__initialized = true;
-    wrapped.__managed = true;
-    if (!this.#usesReturningStatement) {
-      await this.reloadVersionValues(meta, [changeSet], options);
-    }
-    changeSet.persisted = true;
-  }
-  async persistNewEntities(meta, changeSets, options) {
-    const size = this.#config.get('batchSize');
-    for (let i = 0; i < changeSets.length; i += size) {
-      const chunk = changeSets.slice(i, i + size);
-      await this.persistNewEntitiesBatch(meta, chunk, options);
-      if (!this.#usesReturningStatement) {
-        await this.reloadVersionValues(meta, chunk, options);
-      }
-    }
-  }
-  prepareOptions(meta, options, additionalOptions) {
-    const loggerContext = Utils.merge(
-      { id: this.#em._id },
-      this.#em.getLoggerContext({ disableContextResolution: true }),
-    );
-    return {
-      ...options,
-      ...additionalOptions,
-      schema: options?.schema ?? meta.schema,
-      loggerContext,
-    };
-  }
-  async persistNewEntitiesBatch(meta, changeSets, options) {
-    options = this.prepareOptions(meta, options, {
-      convertCustomTypes: false,
-      processCollections: false,
-    });
-    for (const changeSet of changeSets) {
-      this.resolveTPTIdentifiers(changeSet);
-    }
-    const res = await this.#driver.nativeInsertMany(
-      meta.class,
-      changeSets.map(cs => cs.payload),
-      options,
-    );
-    for (let i = 0; i < changeSets.length; i++) {
-      const changeSet = changeSets[i];
-      const wrapped = helper(changeSet.entity);
-      if (!wrapped.hasPrimaryKey()) {
-        const field = meta.getPrimaryProps()[0].fieldNames[0];
-        this.mapPrimaryKey(meta, res.rows[i][field], changeSet);
-      }
-      if (res.rows) {
-        this.mapReturnedValues(changeSet.entity, changeSet.payload, res.rows[i], meta);
-      }
-      this.markAsPopulated(changeSet, meta);
-      wrapped.__initialized = true;
-      wrapped.__managed = true;
-      changeSet.persisted = true;
-    }
-  }
-  async persistManagedEntity(changeSet, options) {
-    const meta = changeSet.meta;
-    const res = await this.updateEntity(meta, changeSet, options);
-    this.checkOptimisticLock(meta, changeSet, res);
-    this.mapReturnedValues(changeSet.entity, changeSet.payload, res.row, meta);
-    await this.reloadVersionValues(meta, [changeSet], options);
-    changeSet.persisted = true;
-  }
-  async persistManagedEntities(meta, changeSets, options) {
-    const size = this.#config.get('batchSize');
-    for (let i = 0; i < changeSets.length; i += size) {
-      const chunk = changeSets.slice(i, i + size);
-      await this.persistManagedEntitiesBatch(meta, chunk, options);
-      await this.reloadVersionValues(meta, chunk, options);
-    }
-  }
-  checkConcurrencyKeys(meta, changeSet, cond) {
-    const tmp = [];
-    for (const key of meta.concurrencyCheckKeys) {
-      cond[key] = changeSet.originalEntity[key];
-      if (changeSet.payload[key]) {
-        tmp.push(key);
-      }
-    }
-    if (tmp.length === 0 && meta.concurrencyCheckKeys.size > 0) {
-      throw OptimisticLockError.lockFailed(changeSet.entity);
-    }
-  }
-  async persistManagedEntitiesBatch(meta, changeSets, options) {
-    await this.checkOptimisticLocks(meta, changeSets, options);
-    options = this.prepareOptions(meta, options, {
-      convertCustomTypes: false,
-      processCollections: false,
-    });
-    const cond = [];
-    const payload = [];
-    for (const changeSet of changeSets) {
-      const where = changeSet.getPrimaryKey(true);
-      this.checkConcurrencyKeys(meta, changeSet, where);
-      cond.push(where);
-      payload.push(changeSet.payload);
-    }
-    const res = await this.#driver.nativeUpdateMany(meta.class, cond, payload, options);
-    const map = new Map();
-    res.rows?.forEach(item => map.set(Utils.getCompositeKeyHash(item, meta, true, this.#platform, true), item));
-    for (const changeSet of changeSets) {
-      if (res.rows) {
-        const row = map.get(helper(changeSet.entity).getSerializedPrimaryKey());
-        this.mapReturnedValues(changeSet.entity, changeSet.payload, row, meta);
-      }
-      changeSet.persisted = true;
-    }
-  }
-  mapPrimaryKey(meta, value, changeSet) {
-    const prop = meta.properties[meta.primaryKeys[0]];
-    const insertId = prop.customType ? prop.customType.convertToJSValue(value, this.#platform) : value;
-    const wrapped = helper(changeSet.entity);
-    if (!wrapped.hasPrimaryKey()) {
-      wrapped.setPrimaryKey(insertId);
-    }
-    // some drivers might be returning bigint PKs as numbers when the number is small enough,
-    // but we need to have it as string so comparison works in change set tracking, so instead
-    // of using the raw value from db, we convert it back to the db value explicitly
-    value = prop.customType
-      ? prop.customType.convertToDatabaseValue(insertId, this.#platform, { mode: 'serialization' })
-      : value;
-    changeSet.payload[wrapped.__meta.primaryKeys[0]] = value;
-    if (wrapped.__identifier && !Array.isArray(wrapped.__identifier)) {
-      wrapped.__identifier.setValue(value);
-    }
-  }
-  /**
-   * Sets populate flag to new entities so they are serialized like if they were loaded from the db
-   */
-  markAsPopulated(changeSet, meta) {
-    helper(changeSet.entity).__schema = this.#driver.getSchemaName(meta, changeSet);
-    if (!this.#config.get('populateAfterFlush')) {
-      return;
+    #platform;
+    #comparator;
+    #usesReturningStatement;
+    #driver;
+    #metadata;
+    #hydrator;
+    #factory;
+    #config;
+    #em;
+    constructor(em) {
+        this.#em = em;
+        this.#driver = this.#em.getDriver();
+        this.#config = this.#em.config;
+        this.#metadata = this.#em.getMetadata();
+        this.#factory = this.#em.getEntityFactory();
+        this.#platform = this.#driver.getPlatform();
+        this.#hydrator = this.#config.getHydrator(this.#metadata);
+        this.#comparator = this.#config.getComparator(this.#metadata);
+        this.#usesReturningStatement = this.#platform.usesReturningStatement() || this.#platform.usesOutputStatement();
+    }
+    /** Executes all pending INSERT change sets, using batch inserts when possible. */
+    async executeInserts(changeSets, options, withSchema) {
+        if (!withSchema) {
+            return this.runForEachSchema(changeSets, 'executeInserts', options);
+        }
+        const meta = changeSets[0].meta;
+        changeSets.forEach(changeSet => this.processProperties(changeSet));
+        if (changeSets.length > 1 && this.#config.get('useBatchInserts', this.#platform.usesBatchInserts())) {
+            return this.persistNewEntities(meta, changeSets, options);
+        }
+        for (const changeSet of changeSets) {
+            await this.persistNewEntity(meta, changeSet, options);
+        }
     }
-    helper(changeSet.entity).populated();
-    meta.relations.forEach(prop => {
-      const value = changeSet.entity[prop.name];
-      if (Utils.isEntity(value, true)) {
-        value.__helper.populated();
-      } else if (Utils.isCollection(value)) {
-        value.populated();
-      }
-    });
-  }
-  async updateEntity(meta, changeSet, options) {
-    const cond = changeSet.getPrimaryKey(true);
-    options = this.prepareOptions(meta, options, {
-      convertCustomTypes: false,
-    });
-    if (
-      meta.concurrencyCheckKeys.size === 0 &&
-      (!meta.versionProperty || changeSet.entity[meta.versionProperty] == null)
-    ) {
-      return this.#driver.nativeUpdate(changeSet.meta.class, cond, changeSet.payload, options);
+    /** Executes all pending UPDATE change sets, using batch updates when possible. */
+    async executeUpdates(changeSets, batched, options, withSchema) {
+        if (!withSchema) {
+            return this.runForEachSchema(changeSets, 'executeUpdates', options, batched);
+        }
+        const meta = changeSets[0].meta;
+        changeSets.forEach(changeSet => this.processProperties(changeSet));
+        // For STI with conflicting fieldNames (renamedFrom properties), we can't batch mixed child types
+        const hasSTIConflicts = meta.root.props.some(p => p.renamedFrom);
+        const hasMixedTypes = hasSTIConflicts && changeSets.some(cs => cs.meta.class !== meta.class);
+        if (batched &&
+            changeSets.length > 1 &&
+            !hasMixedTypes &&
+            this.#config.get('useBatchUpdates', this.#platform.usesBatchUpdates())) {
+            return this.persistManagedEntities(meta, changeSets, options);
+        }
+        for (const changeSet of changeSets) {
+            await this.persistManagedEntity(changeSet, options);
+        }
     }
-    if (meta.versionProperty) {
-      cond[meta.versionProperty] = this.#platform.convertVersionValue(
-        changeSet.entity[meta.versionProperty],
-        meta.properties[meta.versionProperty],
-      );
+    /** Executes all pending DELETE change sets in batches. */
+    async executeDeletes(changeSets, options, withSchema) {
+        if (!withSchema) {
+            return this.runForEachSchema(changeSets, 'executeDeletes', options);
+        }
+        const size = this.#config.get('batchSize');
+        const meta = changeSets[0].meta;
+        const pk = Utils.getPrimaryKeyHash(meta.primaryKeys);
+        for (let i = 0; i < changeSets.length; i += size) {
+            const chunk = changeSets.slice(i, i + size);
+            const pks = chunk.map(cs => cs.getPrimaryKey());
+            options = this.prepareOptions(meta, options);
+            await this.#driver.nativeDelete(meta.root.class, { [pk]: { $in: pks } }, options);
+        }
     }
-    this.checkConcurrencyKeys(meta, changeSet, cond);
-    return this.#driver.nativeUpdate(changeSet.meta.class, cond, changeSet.payload, options);
-  }
-  async checkOptimisticLocks(meta, changeSets, options) {
-    if (
-      meta.concurrencyCheckKeys.size === 0 &&
-      (!meta.versionProperty || changeSets.every(cs => cs.entity[meta.versionProperty] == null))
-    ) {
-      return;
+    async runForEachSchema(changeSets, method, options, ...args) {
+        const groups = new Map();
+        changeSets.forEach(cs => {
+            const group = groups.get(cs.schema) ?? [];
+            group.push(cs);
+            groups.set(cs.schema, group);
+        });
+        for (const [key, group] of groups.entries()) {
+            options = { ...options, schema: key };
+            // @ts-ignore
+            await this[method](group, ...args, options, true);
+        }
     }
-    // skip entity references as they don't have version values loaded
-    changeSets = changeSets.filter(cs => helper(cs.entity).__initialized);
-    const $or = changeSets.map(cs => {
-      const cond = Utils.getPrimaryKeyCond(cs.originalEntity, meta.primaryKeys.concat(...meta.concurrencyCheckKeys));
-      if (meta.versionProperty) {
-        // @ts-ignore
-        cond[meta.versionProperty] = this.#platform.convertVersionValue(
-          cs.entity[meta.versionProperty],
-          meta.properties[meta.versionProperty],
-        );
-      }
-      return cond;
-    });
-    const primaryKeys = meta.primaryKeys.concat(...meta.concurrencyCheckKeys);
-    options = this.prepareOptions(meta, options, {
-      fields: primaryKeys,
-    });
-    const res = await this.#driver.find(meta.root.class, { $or }, options);
-    if (res.length !== changeSets.length) {
-      const compare = (a, b, keys) => keys.every(k => a[k] === b[k]);
-      const entity = changeSets.find(cs => {
-        return !res.some(row => compare(Utils.getPrimaryKeyCond(cs.entity, primaryKeys), row, primaryKeys));
-      }).entity;
-      throw OptimisticLockError.lockFailed(entity);
+    validateRequired(entity) {
+        const wrapped = helper(entity);
+        for (const prop of wrapped.__meta.props) {
+            if (!prop.nullable &&
+                !prop.autoincrement &&
+                !prop.default &&
+                !prop.defaultRaw &&
+                !prop.onCreate &&
+                !prop.generated &&
+                !prop.embedded &&
+                ![ReferenceKind.ONE_TO_MANY, ReferenceKind.MANY_TO_MANY].includes(prop.kind) &&
+                prop.name !== wrapped.__meta.root.discriminatorColumn &&
+                prop.type !== 'ObjectId' &&
+                prop.persist !== false &&
+                entity[prop.name] == null) {
+                throw ValidationError.propertyRequired(entity, prop);
+            }
+        }
     }
-  }
-  checkOptimisticLock(meta, changeSet, res) {
-    if ((meta.versionProperty || meta.concurrencyCheckKeys.size > 0) && res && !res.affectedRows) {
-      throw OptimisticLockError.lockFailed(changeSet.entity);
+    processProperties(changeSet) {
+        const meta = changeSet.meta;
+        for (const prop of meta.relations) {
+            this.processProperty(changeSet, prop);
+        }
+        if (changeSet.type === ChangeSetType.CREATE && this.#config.get('validateRequired')) {
+            this.validateRequired(changeSet.entity);
+        }
     }
-  }
-  /**
-   * This method also handles reloading of database default values for inserts and raw property updates,
-   * so we use a single query in case of both versioning and default values is used.
-   */
-  async reloadVersionValues(meta, changeSets, options) {
-    const reloadProps =
-      meta.versionProperty && !this.#usesReturningStatement ? [meta.properties[meta.versionProperty]] : [];
-    if (changeSets[0].type === ChangeSetType.CREATE) {
-      for (const prop of meta.props) {
-        if (prop.persist === false) {
-          continue;
-        }
-        if (isRaw(changeSets[0].entity[prop.name])) {
-          reloadProps.push(prop);
-          continue;
-        }
-        // do not reload things that already had a runtime value
-        if (changeSets[0].entity[prop.name] != null || prop.defaultRaw === 'null') {
-          continue;
-        }
-        if (prop.autoincrement || prop.generated || prop.defaultRaw) {
-          reloadProps.push(prop);
-        }
-      }
+    async persistNewEntity(meta, changeSet, options) {
+        const wrapped = helper(changeSet.entity);
+        options = this.prepareOptions(meta, options, {
+            convertCustomTypes: false,
+        });
+        this.resolveTPTIdentifiers(changeSet);
+        // Use changeSet's own meta for STI entities to get correct field mappings
+        const res = await this.#driver.nativeInsertMany(changeSet.meta.class, [changeSet.payload], options);
+        if (!wrapped.hasPrimaryKey()) {
+            this.mapPrimaryKey(meta, res.insertId ?? res.row?.[meta.primaryKeys[0]], changeSet);
+        }
+        this.mapReturnedValues(changeSet.entity, changeSet.payload, res.row, meta);
+        this.markAsPopulated(changeSet, meta);
+        wrapped.__initialized = true;
+        wrapped.__managed = true;
+        if (!this.#usesReturningStatement) {
+            await this.reloadVersionValues(meta, [changeSet], options);
+        }
+        changeSet.persisted = true;
+    }
+    async persistNewEntities(meta, changeSets, options) {
+        const size = this.#config.get('batchSize');
+        for (let i = 0; i < changeSets.length; i += size) {
+            const chunk = changeSets.slice(i, i + size);
+            await this.persistNewEntitiesBatch(meta, chunk, options);
+            if (!this.#usesReturningStatement) {
+                await this.reloadVersionValues(meta, chunk, options);
+            }
+        }
     }
-    if (changeSets[0].type === ChangeSetType.UPDATE) {
-      const returning = new Set();
-      changeSets.forEach(cs => {
-        Utils.keys(cs.payload).forEach(k => {
-          if (isRaw(cs.payload[k]) && isRaw(cs.entity[k])) {
-            returning.add(meta.properties[k]);
-          }
+    prepareOptions(meta, options, additionalOptions) {
+        const loggerContext = Utils.merge({ id: this.#em._id }, this.#em.getLoggerContext({ disableContextResolution: true }));
+        return {
+            ...options,
+            ...additionalOptions,
+            schema: options?.schema ?? meta.schema,
+            loggerContext,
+        };
+    }
+    async persistNewEntitiesBatch(meta, changeSets, options) {
+        options = this.prepareOptions(meta, options, {
+            convertCustomTypes: false,
+            processCollections: false,
         });
-      });
-      // reload generated columns
-      if (!this.#usesReturningStatement) {
-        meta.props.filter(prop => prop.generated && !prop.primary).forEach(prop => reloadProps.push(prop));
-        reloadProps.push(...returning);
-      }
+        for (const changeSet of changeSets) {
+            this.resolveTPTIdentifiers(changeSet);
+        }
+        const res = await this.#driver.nativeInsertMany(meta.class, changeSets.map(cs => cs.payload), options);
+        for (let i = 0; i < changeSets.length; i++) {
+            const changeSet = changeSets[i];
+            const wrapped = helper(changeSet.entity);
+            if (!wrapped.hasPrimaryKey()) {
+                const field = meta.getPrimaryProps()[0].fieldNames[0];
+                this.mapPrimaryKey(meta, res.rows[i][field], changeSet);
+            }
+            if (res.rows) {
+                this.mapReturnedValues(changeSet.entity, changeSet.payload, res.rows[i], meta);
+            }
+            this.markAsPopulated(changeSet, meta);
+            wrapped.__initialized = true;
+            wrapped.__managed = true;
+            changeSet.persisted = true;
+        }
     }
-    if (reloadProps.length === 0) {
-      return;
+    async persistManagedEntity(changeSet, options) {
+        const meta = changeSet.meta;
+        const res = await this.updateEntity(meta, changeSet, options);
+        this.checkOptimisticLock(meta, changeSet, res);
+        this.mapReturnedValues(changeSet.entity, changeSet.payload, res.row, meta);
+        await this.reloadVersionValues(meta, [changeSet], options);
+        changeSet.persisted = true;
+    }
+    async persistManagedEntities(meta, changeSets, options) {
+        const size = this.#config.get('batchSize');
+        for (let i = 0; i < changeSets.length; i += size) {
+            const chunk = changeSets.slice(i, i + size);
+            await this.persistManagedEntitiesBatch(meta, chunk, options);
+            await this.reloadVersionValues(meta, chunk, options);
+        }
     }
-    reloadProps.unshift(...meta.getPrimaryProps());
-    const pk = Utils.getPrimaryKeyHash(meta.primaryKeys);
-    const pks = changeSets.map(cs => {
-      const val = helper(cs.entity).getPrimaryKey(true);
-      if (Utils.isPlainObject(val)) {
-        return Utils.getCompositeKeyValue(val, meta, false, this.#platform);
-      }
-      return val;
-    });
-    options = this.prepareOptions(meta, options, {
-      fields: Utils.unique(reloadProps.map(prop => prop.name)),
-    });
-    const data = await this.#driver.find(meta.class, { [pk]: { $in: pks } }, options);
-    const map = new Map();
-    data.forEach(item => map.set(Utils.getCompositeKeyHash(item, meta, false, this.#platform, true), item));
-    for (const changeSet of changeSets) {
-      const data = map.get(helper(changeSet.entity).getSerializedPrimaryKey());
-      this.#hydrator.hydrate(changeSet.entity, meta, data, this.#factory, 'full', false, true);
-      Object.assign(changeSet.payload, data); // merge to the changeset payload, so it gets saved to the entity snapshot
+    checkConcurrencyKeys(meta, changeSet, cond) {
+        const tmp = [];
+        for (const key of meta.concurrencyCheckKeys) {
+            cond[key] = changeSet.originalEntity[key];
+            if (changeSet.payload[key]) {
+                tmp.push(key);
+            }
+        }
+        if (tmp.length === 0 && meta.concurrencyCheckKeys.size > 0) {
+            throw OptimisticLockError.lockFailed(changeSet.entity);
+        }
     }
-  }
-  /**
-   * For TPT child tables, resolve EntityIdentifier values in PK fields.
-   * The parent table insert assigns the actual PK value, which the child table references.
-   */
-  resolveTPTIdentifiers(changeSet) {
-    if (changeSet.meta.inheritanceType !== 'tpt' || !changeSet.meta.tptParent) {
-      return;
+    async persistManagedEntitiesBatch(meta, changeSets, options) {
+        await this.checkOptimisticLocks(meta, changeSets, options);
+        options = this.prepareOptions(meta, options, {
+            convertCustomTypes: false,
+            processCollections: false,
+        });
+        const cond = [];
+        const payload = [];
+        for (const changeSet of changeSets) {
+            const where = changeSet.getPrimaryKey(true);
+            this.checkConcurrencyKeys(meta, changeSet, where);
+            cond.push(where);
+            payload.push(changeSet.payload);
+        }
+        const res = await this.#driver.nativeUpdateMany(meta.class, cond, payload, options);
+        const map = new Map();
+        res.rows?.forEach(item => map.set(Utils.getCompositeKeyHash(item, meta, true, this.#platform, true), item));
+        for (const changeSet of changeSets) {
+            if (res.rows) {
+                const row = map.get(helper(changeSet.entity).getSerializedPrimaryKey());
+                this.mapReturnedValues(changeSet.entity, changeSet.payload, row, meta);
+            }
+            changeSet.persisted = true;
+        }
     }
-    for (const pk of changeSet.meta.primaryKeys) {
-      const value = changeSet.payload[pk];
-      if (value instanceof EntityIdentifier) {
-        changeSet.payload[pk] = value.getValue();
-      }
+    mapPrimaryKey(meta, value, changeSet) {
+        const prop = meta.properties[meta.primaryKeys[0]];
+        const insertId = prop.customType ? prop.customType.convertToJSValue(value, this.#platform) : value;
+        const wrapped = helper(changeSet.entity);
+        if (!wrapped.hasPrimaryKey()) {
+            wrapped.setPrimaryKey(insertId);
+        }
+        // some drivers might be returning bigint PKs as numbers when the number is small enough,
+        // but we need to have it as string so comparison works in change set tracking, so instead
+        // of using the raw value from db, we convert it back to the db value explicitly
+        value = prop.customType
+            ? prop.customType.convertToDatabaseValue(insertId, this.#platform, { mode: 'serialization' })
+            : value;
+        changeSet.payload[wrapped.__meta.primaryKeys[0]] = value;
+        if (wrapped.__identifier && !Array.isArray(wrapped.__identifier)) {
+            wrapped.__identifier.setValue(value);
+        }
     }
-  }
-  processProperty(changeSet, prop) {
-    const meta = changeSet.meta;
-    const value = changeSet.payload[prop.name]; // for inline embeddables
-    if (value instanceof EntityIdentifier) {
-      changeSet.payload[prop.name] = value.getValue();
-      return;
+    /**
+     * Sets populate flag to new entities so they are serialized like if they were loaded from the db
+     */
+    markAsPopulated(changeSet, meta) {
+        helper(changeSet.entity).__schema = this.#driver.getSchemaName(meta, changeSet);
+        if (!this.#config.get('populateAfterFlush')) {
+            return;
+        }
+        helper(changeSet.entity).populated();
+        meta.relations.forEach(prop => {
+            const value = changeSet.entity[prop.name];
+            if (Utils.isEntity(value, true)) {
+                value.__helper.populated();
+            }
+            else if (Utils.isCollection(value)) {
+                value.populated();
+            }
+        });
     }
-    if (value instanceof PolymorphicRef) {
-      if (value.id instanceof EntityIdentifier) {
-        value.id = value.id.getValue();
-      }
-      return;
+    async updateEntity(meta, changeSet, options) {
+        const cond = changeSet.getPrimaryKey(true);
+        options = this.prepareOptions(meta, options, {
+            convertCustomTypes: false,
+        });
+        if (meta.concurrencyCheckKeys.size === 0 &&
+            (!meta.versionProperty || changeSet.entity[meta.versionProperty] == null)) {
+            return this.#driver.nativeUpdate(changeSet.meta.class, cond, changeSet.payload, options);
+        }
+        if (meta.versionProperty) {
+            cond[meta.versionProperty] = this.#platform.convertVersionValue(changeSet.entity[meta.versionProperty], meta.properties[meta.versionProperty]);
+        }
+        this.checkConcurrencyKeys(meta, changeSet, cond);
+        return this.#driver.nativeUpdate(changeSet.meta.class, cond, changeSet.payload, options);
     }
-    if (Array.isArray(value) && value.every(item => item instanceof EntityIdentifier)) {
-      changeSet.payload[prop.name] = value.map(item => item.getValue());
-      return;
+    async checkOptimisticLocks(meta, changeSets, options) {
+        if (meta.concurrencyCheckKeys.size === 0 &&
+            (!meta.versionProperty || changeSets.every(cs => cs.entity[meta.versionProperty] == null))) {
+            return;
+        }
+        // skip entity references as they don't have version values loaded
+        changeSets = changeSets.filter(cs => helper(cs.entity).__initialized);
+        const $or = changeSets.map(cs => {
+            const cond = Utils.getPrimaryKeyCond(cs.originalEntity, meta.primaryKeys.concat(...meta.concurrencyCheckKeys));
+            if (meta.versionProperty) {
+                // @ts-ignore
+                cond[meta.versionProperty] = this.#platform.convertVersionValue(cs.entity[meta.versionProperty], meta.properties[meta.versionProperty]);
+            }
+            return cond;
+        });
+        const primaryKeys = meta.primaryKeys.concat(...meta.concurrencyCheckKeys);
+        options = this.prepareOptions(meta, options, {
+            fields: primaryKeys,
+        });
+        const res = await this.#driver.find(meta.root.class, { $or }, options);
+        if (res.length !== changeSets.length) {
+            const compare = (a, b, keys) => keys.every(k => a[k] === b[k]);
+            const entity = changeSets.find(cs => {
+                return !res.some(row => compare(Utils.getPrimaryKeyCond(cs.entity, primaryKeys), row, primaryKeys));
+            }).entity;
+            throw OptimisticLockError.lockFailed(entity);
+        }
     }
-    if (prop.kind === ReferenceKind.MANY_TO_MANY && Array.isArray(value)) {
-      changeSet.payload[prop.name] = value.map(val => (val instanceof EntityIdentifier ? val.getValue() : val));
-      return;
+    checkOptimisticLock(meta, changeSet, res) {
+        if ((meta.versionProperty || meta.concurrencyCheckKeys.size > 0) && res && !res.affectedRows) {
+            throw OptimisticLockError.lockFailed(changeSet.entity);
+        }
     }
-    if (prop.name in changeSet.payload) {
-      return;
+    /**
+     * This method also handles reloading of database default values for inserts and raw property updates,
+     * so we use a single query in case of both versioning and default values is used.
+     */
+    async reloadVersionValues(meta, changeSets, options) {
+        const reloadProps = meta.versionProperty && !this.#usesReturningStatement ? [meta.properties[meta.versionProperty]] : [];
+        if (changeSets[0].type === ChangeSetType.CREATE) {
+            for (const prop of meta.props) {
+                if (prop.persist === false) {
+                    continue;
+                }
+                if (isRaw(changeSets[0].entity[prop.name])) {
+                    reloadProps.push(prop);
+                    continue;
+                }
+                // do not reload things that already had a runtime value
+                if (changeSets[0].entity[prop.name] != null || prop.defaultRaw === 'null') {
+                    continue;
+                }
+                if (prop.autoincrement || prop.generated || prop.defaultRaw) {
+                    reloadProps.push(prop);
+                }
+            }
+        }
+        if (changeSets[0].type === ChangeSetType.UPDATE) {
+            const returning = new Set();
+            changeSets.forEach(cs => {
+                Utils.keys(cs.payload).forEach(k => {
+                    if (isRaw(cs.payload[k]) && isRaw(cs.entity[k])) {
+                        returning.add(meta.properties[k]);
+                    }
+                });
+            });
+            // reload generated columns
+            if (!this.#usesReturningStatement) {
+                meta.props.filter(prop => prop.generated && !prop.primary).forEach(prop => reloadProps.push(prop));
+                reloadProps.push(...returning);
+            }
+        }
+        if (reloadProps.length === 0) {
+            return;
+        }
+        reloadProps.unshift(...meta.getPrimaryProps());
+        const pk = Utils.getPrimaryKeyHash(meta.primaryKeys);
+        const pks = changeSets.map(cs => {
+            const val = helper(cs.entity).getPrimaryKey(true);
+            if (Utils.isPlainObject(val)) {
+                return Utils.getCompositeKeyValue(val, meta, false, this.#platform);
+            }
+            return val;
+        });
+        options = this.prepareOptions(meta, options, {
+            fields: Utils.unique(reloadProps.map(prop => prop.name)),
+        });
+        const data = await this.#driver.find(meta.class, { [pk]: { $in: pks } }, options);
+        const map = new Map();
+        data.forEach(item => map.set(Utils.getCompositeKeyHash(item, meta, false, this.#platform, true), item));
+        for (const changeSet of changeSets) {
+            const data = map.get(helper(changeSet.entity).getSerializedPrimaryKey());
+            this.#hydrator.hydrate(changeSet.entity, meta, data, this.#factory, 'full', false, true);
+            Object.assign(changeSet.payload, data); // merge to the changeset payload, so it gets saved to the entity snapshot
+        }
     }
-    const values = Utils.unwrapProperty(changeSet.payload, meta, prop, true); // for object embeddables
-    values.forEach(([value, indexes]) => {
-      if (value instanceof EntityIdentifier) {
-        Utils.setPayloadProperty(changeSet.payload, meta, prop, value.getValue(), indexes);
-      }
-    });
-  }
-  /**
-   * Maps values returned via `returning` statement (postgres) or the inserted id (other sql drivers).
-   * No need to handle composite keys here as they need to be set upfront.
-   * We do need to map to the change set payload too, as it will be used in the originalEntityData for new entities.
-   */
-  mapReturnedValues(entity, payload, row, meta, upsert = false) {
-    if ((!this.#usesReturningStatement && !upsert) || !row || !Utils.hasObjectKeys(row)) {
-      return;
+    /**
+     * For TPT child tables, resolve EntityIdentifier values in PK fields.
+     * The parent table insert assigns the actual PK value, which the child table references.
+     */
+    resolveTPTIdentifiers(changeSet) {
+        if (changeSet.meta.inheritanceType !== 'tpt' || !changeSet.meta.tptParent) {
+            return;
+        }
+        for (const pk of changeSet.meta.primaryKeys) {
+            const value = changeSet.payload[pk];
+            if (value instanceof EntityIdentifier) {
+                changeSet.payload[pk] = value.getValue();
+            }
+        }
     }
-    const mapped = this.#comparator.mapResult(meta, row);
-    if (entity) {
-      this.#hydrator.hydrate(entity, meta, mapped, this.#factory, 'full', false, true);
+    processProperty(changeSet, prop) {
+        const meta = changeSet.meta;
+        const value = changeSet.payload[prop.name]; // for inline embeddables
+        if (value instanceof EntityIdentifier) {
+            changeSet.payload[prop.name] = value.getValue();
+            return;
+        }
+        if (value instanceof PolymorphicRef) {
+            if (value.id instanceof EntityIdentifier) {
+                value.id = value.id.getValue();
+            }
+            return;
+        }
+        if (Array.isArray(value) && value.every(item => item instanceof EntityIdentifier)) {
+            changeSet.payload[prop.name] = value.map(item => item.getValue());
+            return;
+        }
+        if (prop.kind === ReferenceKind.MANY_TO_MANY && Array.isArray(value)) {
+            changeSet.payload[prop.name] = value.map(val => (val instanceof EntityIdentifier ? val.getValue() : val));
+            return;
+        }
+        if (prop.name in changeSet.payload) {
+            return;
+        }
+        const values = Utils.unwrapProperty(changeSet.payload, meta, prop, true); // for object embeddables
+        values.forEach(([value, indexes]) => {
+            if (value instanceof EntityIdentifier) {
+                Utils.setPayloadProperty(changeSet.payload, meta, prop, value.getValue(), indexes);
+            }
+        });
     }
-    if (upsert) {
-      for (const prop of meta.props) {
-        if (prop.customType && prop.name in mapped) {
-          mapped[prop.name] = prop.customType.convertToJSValue(mapped[prop.name], this.#platform);
+    /**
+     * Maps values returned via `returning` statement (postgres) or the inserted id (other sql drivers).
+     * No need to handle composite keys here as they need to be set upfront.
+     * We do need to map to the change set payload too, as it will be used in the originalEntityData for new entities.
+     */
+    mapReturnedValues(entity, payload, row, meta, upsert = false) {
+        if ((!this.#usesReturningStatement && !upsert) || !row || !Utils.hasObjectKeys(row)) {
+            return;
+        }
+        const mapped = this.#comparator.mapResult(meta, row);
+        if (entity) {
+            this.#hydrator.hydrate(entity, meta, mapped, this.#factory, 'full', false, true);
+        }
+        if (upsert) {
+            for (const prop of meta.props) {
+                if (prop.customType && prop.name in mapped) {
+                    mapped[prop.name] = prop.customType.convertToJSValue(mapped[prop.name], this.#platform);
+                }
+            }
         }
-      }
+        Object.assign(payload, mapped);
     }
-    Object.assign(payload, mapped);
-  }
 }