@mikro-orm/knex 6.6.2-dev.5 → 6.6.2-dev.6

package/AbstractSqlDriver.js

@@ -757,7 +757,7 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
             }
             const tableName = `${schema ?? '_'}.${pivotMeta.tableName}`;
             const persister = groups[tableName] ??= new PivotCollectionPersister_1.PivotCollectionPersister(pivotMeta, this, options?.ctx, schema, options?.loggerContext);
-            persister.enqueueUpdate(coll.property, insertDiff, deleteDiff, pks);
+            persister.enqueueUpdate(coll.property, insertDiff, deleteDiff, pks, coll.isInitialized());
         }
         for (const persister of core_1.Utils.values(groups)) {
             await this.rethrow(persister.execute());

package/PivotCollectionPersister.d.ts

@@ -6,14 +6,17 @@ export declare class PivotCollectionPersister<Entity extends object> {
     private readonly ctx?;
     private readonly schema?;
     private readonly loggerContext?;
-    private readonly platform;
     private readonly inserts;
+    private readonly upserts;
     private readonly deletes;
     private readonly batchSize;
     private order;
     constructor(meta: EntityMetadata<Entity>, driver: AbstractSqlDriver, ctx?: Transaction | undefined, schema?: string | undefined, loggerContext?: Dictionary | undefined);
-    enqueueUpdate(prop: EntityProperty<Entity>, insertDiff: Primary<Entity>[][], deleteDiff: Primary<Entity>[][] | boolean, pks: Primary<Entity>[]): void;
+    enqueueUpdate(prop: EntityProperty<Entity>, insertDiff: Primary<Entity>[][], deleteDiff: Primary<Entity>[][] | boolean, pks: Primary<Entity>[], isInitialized?: boolean): void;
     private enqueueInsert;
+    private enqueueUpsert;
+    private createInsertStatement;
     private enqueueDelete;
+    private collectStatements;
     execute(): Promise<void>;
 }

package/PivotCollectionPersister.js

@@ -1,7 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.PivotCollectionPersister = void 0;
-const core_1 = require("@mikro-orm/core");
 class InsertStatement {
     keys;
     data;
@@ -42,8 +41,8 @@ class PivotCollectionPersister {
     ctx;
     schema;
     loggerContext;
-    platform;
     inserts = new Map();
+    upserts = new Map();
     deletes = new Map();
     batchSize;
     order = 0;
@@ -53,12 +52,16 @@ class PivotCollectionPersister {
         this.ctx = ctx;
         this.schema = schema;
         this.loggerContext = loggerContext;
-        this.platform = this.driver.getPlatform();
         this.batchSize = this.driver.config.get('batchSize');
     }
-    enqueueUpdate(prop, insertDiff, deleteDiff, pks) {
+    enqueueUpdate(prop, insertDiff, deleteDiff, pks, isInitialized = true) {
         if (insertDiff.length) {
-            this.enqueueInsert(prop, insertDiff, pks);
+            if (isInitialized) {
+                this.enqueueInsert(prop, insertDiff, pks);
+            }
+            else {
+                this.enqueueUpsert(prop, insertDiff, pks);
+            }
         }
         if (deleteDiff === true || (Array.isArray(deleteDiff) && deleteDiff.length)) {
             this.enqueueDelete(prop, deleteDiff, pks);
@@ -66,17 +69,29 @@ class PivotCollectionPersister {
     }
     enqueueInsert(prop, insertDiff, pks) {
         for (const fks of insertDiff) {
-            const data = prop.owner ? [...fks, ...pks] : [...pks, ...fks];
-            const keys = prop.owner
-                ? [...prop.inverseJoinColumns, ...prop.joinColumns]
-                : [...prop.joinColumns, ...prop.inverseJoinColumns];
-            const statement = new InsertStatement(keys, data, this.order++);
+            const statement = this.createInsertStatement(prop, fks, pks);
             const hash = statement.getHash();
             if (prop.owner || !this.inserts.has(hash)) {
                 this.inserts.set(hash, statement);
             }
         }
     }
+    enqueueUpsert(prop, insertDiff, pks) {
+        for (const fks of insertDiff) {
+            const statement = this.createInsertStatement(prop, fks, pks);
+            const hash = statement.getHash();
+            if (prop.owner || !this.upserts.has(hash)) {
+                this.upserts.set(hash, statement);
+            }
+        }
+    }
+    createInsertStatement(prop, fks, pks) {
+        const data = prop.owner ? [...fks, ...pks] : [...pks, ...fks];
+        const keys = prop.owner
+            ? [...prop.inverseJoinColumns, ...prop.joinColumns]
+            : [...prop.joinColumns, ...prop.inverseJoinColumns];
+        return new InsertStatement(keys, data, this.order++);
+    }
     enqueueDelete(prop, deleteDiff, pks) {
         if (deleteDiff === true) {
             const statement = new DeleteStatement(prop.joinColumns, pks);
@@ -92,6 +107,13 @@ class PivotCollectionPersister {
             this.deletes.set(statement.getHash(), statement);
         }
     }
+    collectStatements(statements) {
+        const items = [];
+        for (const statement of statements.values()) {
+            items[statement.order] = statement.getData();
+        }
+        return items.filter(Boolean);
+    }
     async execute() {
         if (this.deletes.size > 0) {
             const deletes = [...this.deletes.values()];
@@ -108,18 +130,10 @@ class PivotCollectionPersister {
                 });
             }
         }
-        if (this.inserts.size === 0) {
-            return;
-        }
-        let items = [];
-        for (const insert of this.inserts.values()) {
-            items[insert.order] = insert.getData();
-        }
-        items = items.filter(i => i);
-        /* istanbul ignore else */
-        if (this.platform.allowsMultiInsert()) {
-            for (let i = 0; i < items.length; i += this.batchSize) {
-                const chunk = items.slice(i, i + this.batchSize);
+        if (this.inserts.size > 0) {
+            const filtered = this.collectStatements(this.inserts);
+            for (let i = 0; i < filtered.length; i += this.batchSize) {
+                const chunk = filtered.slice(i, i + this.batchSize);
                 await this.driver.nativeInsertMany(this.meta.className, chunk, {
                     ctx: this.ctx,
                     schema: this.schema,
@@ -129,13 +143,20 @@ class PivotCollectionPersister {
                 });
             }
         }
-        else {
-            await core_1.Utils.runSerial(items, item => {
-                return this.driver.createQueryBuilder(this.meta.className, this.ctx, 'write', false, this.loggerContext)
-                    .withSchema(this.schema)
-                    .insert(item)
-                    .execute('run', false);
-            });
+        if (this.upserts.size > 0) {
+            const filtered = this.collectStatements(this.upserts);
+            for (let i = 0; i < filtered.length; i += this.batchSize) {
+                const chunk = filtered.slice(i, i + this.batchSize);
+                await this.driver.nativeUpdateMany(this.meta.className, [], chunk, {
+                    ctx: this.ctx,
+                    schema: this.schema,
+                    convertCustomTypes: false,
+                    processCollections: false,
+                    upsert: true,
+                    onConflictAction: 'ignore',
+                    loggerContext: this.loggerContext,
+                });
+            }
         }
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mikro-orm/knex",
-  "version": "6.6.2-dev.5",
+  "version": "6.6.2-dev.6",
   "description": "TypeScript ORM for Node.js based on Data Mapper, Unit of Work and Identity Map patterns. Supports MongoDB, MySQL, PostgreSQL and SQLite databases as well as usage with vanilla JavaScript.",
   "main": "index.js",
   "module": "index.mjs",
@@ -66,7 +66,7 @@
     "@mikro-orm/core": "^6.6.1"
   },
   "peerDependencies": {
-    "@mikro-orm/core": "6.6.2-dev.5",
+    "@mikro-orm/core": "6.6.2-dev.6",
     "better-sqlite3": "*",
     "libsql": "*",
     "mariadb": "*"