@mikro-orm/core 7.0.2-dev.9 → 7.0.2

package/unit-of-work/ChangeSetPersister.js

@@ -6,445 +6,470 @@ import { isRaw } from '../utils/RawQueryFragment.js';
 import { Utils } from '../utils/Utils.js';
 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();
-    }
-    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);
-        }
+  #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);
     }
-    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);
-        }
+    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);
     }
-    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);
-        }
+    for (const changeSet of changeSets) {
+      await this.persistNewEntity(meta, changeSet, 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);
-        }
+  }
+  /** 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);
     }
-    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);
-            }
-        }
+    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);
     }
-    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);
-        }
+    for (const changeSet of changeSets) {
+      await this.persistManagedEntity(changeSet, options);
     }
-    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);
-            }
-        }
+  }
+  /** Executes all pending DELETE change sets in batches. */
+  async executeDeletes(changeSets, options, withSchema) {
+    if (!withSchema) {
+      return this.runForEachSchema(changeSets, 'executeDeletes', 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;
-        }
+    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 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);
-        }
+  }
+  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);
     }
-    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);
-        }
+  }
+  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);
+      }
     }
-    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;
-        }
+  }
+  processProperties(changeSet) {
+    const meta = changeSet.meta;
+    for (const prop of meta.relations) {
+      this.processProperty(changeSet, prop);
     }
-    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);
-        }
+    if (changeSet.type === ChangeSetType.CREATE && this.#config.get('validateRequired')) {
+      this.validateRequired(changeSet.entity);
     }
-    /**
-     * 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();
-            }
-        });
+  }
+  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);
     }
-    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);
+    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);
     }
-    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);
-        }
+    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);
+      }
     }
-    checkOptimisticLock(meta, changeSet, res) {
-        if ((meta.versionProperty || meta.concurrencyCheckKeys.size > 0) && res && !res.affectedRows) {
-            throw OptimisticLockError.lockFailed(changeSet.entity);
-        }
+  }
+  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);
     }
-    /**
-     * 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 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;
     }
-    /**
-     * 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();
-            }
-        }
+  }
+  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);
     }
-    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);
-            }
+  }
+  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;
+    }
+    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);
+    }
+    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);
+  }
+  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);
+    }
+  }
+  checkOptimisticLock(meta, changeSet, res) {
+    if ((meta.versionProperty || meta.concurrencyCheckKeys.size > 0) && res && !res.affectedRows) {
+      throw OptimisticLockError.lockFailed(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);
+        }
+      }
+    }
+    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);
+      }
     }
-    /**
-     * 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);
-                }
-            }
+    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
+    }
+  }
+  /**
+   * 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();
+      }
+    }
+  }
+  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);
+      }
+    });
+  }
+  /**
+   * 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);
+  }
 }