Method upsertMany

packages/core/src/EntityManager.ts:1391–1655

* Creates or updates the entity, based on whether it is already present in the database. * This method performs an `insert on conflict merge` query ensuring the database is in sync, returning a managed * entity instance. The method accepts either `entityName` together with the entity `data`, o

(
    entityNameOrEntity: EntityName<Entity> | Entity[],
    data?: (EntityData<Entity> | NoInfer<Entity>)[],
    options: UpsertManyOptions<Entity, Fields> = {},
  )

   * If the entity is already present in current context, there won't be any queries - instead, the entity data will be assigned and an explicit `flush` will be required for those changes to be persisted.
   */
  async upsertMany<Entity extends object, Fields extends string = any>(
    entityNameOrEntity: EntityName<Entity> | Entity[],
    data?: (EntityData<Entity> | NoInfer<Entity>)[],
    options: UpsertManyOptions<Entity, Fields> = {},
  ): Promise<Entity[]> {
    if (options.disableIdentityMap ?? this.config.get('disableIdentityMap')) {
      const em = this.getContext(false);
      const fork = em.fork({ keepTransactionContext: true });
      const ret = await fork.upsertMany(entityNameOrEntity, data, { ...options, disableIdentityMap: false });
      fork.clear();

      return ret;
    }

    const em = this.getContext(false);
    em.prepareOptions(options);

    let entityName: EntityName<Entity>;
    let propIndex: number | false;

    if (data === undefined) {
      entityName = (entityNameOrEntity as Entity[])[0].constructor;
      data = entityNameOrEntity as Entity[];
    } else {
      entityName = entityNameOrEntity as EntityName<Entity>;
    }

    const batchSize = options.batchSize ?? this.config.get('batchSize');

    if (data.length > batchSize) {
      const ret: Entity[] = [];

      for (let i = 0; i < data.length; i += batchSize) {
        const chunk = data.slice(i, i + batchSize);
        ret.push(...(await this.upsertMany(entityName, chunk, options)));
      }

      return ret;
    }

    const meta = this.metadata.get<Entity>(entityName);
    const convertCustomTypes = !Utils.isEntity(data[0]);
    const allData: EntityData<Entity>[] = [];
    const allWhere: FilterQuery<Entity>[] = [];
    const entities = new Map<Entity, EntityData<Entity>>();
    const entitiesByData = new Map<EntityData<Entity>, Entity>();
    const entitiesByAllDataIdx = new Map<number, Entity>();

    for (let i = 0; i < data.length; i++) {
      let row = data[i];
      let where: FilterQuery<Entity>;

      if (Utils.isEntity(row)) {
        const entity = row as Entity;

        if (helper(entity).__managed && helper(entity).__em === em && !this.config.get('upsertManaged')) {
          em.#entityFactory.mergeData(meta, entity, row, { initialized: true });
          entities.set(entity, row);