@nest-omni/core 4.1.2 → 4.1.3-0

package/common/boilerplate.polyfill.js

@@ -174,10 +174,41 @@ const insertWithPk = function (entity) {
                 valuePlaceholders.push(`$${columnValues.length}`);
             }
         }
-        const sql = `INSERT INTO ${tableName} (${columnNames.join(', ')}) VALUES (${valuePlaceholders.join(', ')}) RETURNING *`;
+        const connection = this.manager.connection;
+        const databaseType = connection.options.type;
         try {
-            const result = yield this.query(sql, columnValues);
-            return result[0];
+            let result;
+            if (['postgres', 'sqlite'].includes(databaseType)) {
+                const sql = `INSERT INTO ${tableName} (${columnNames.join(', ')}) VALUES (${valuePlaceholders.join(', ')}) RETURNING *`;
+                const queryResult = yield this.query(sql, columnValues);
+                result = queryResult;
+            }
+            else if (['mysql', 'mariadb'].includes(databaseType)) {
+                const insertSql = `INSERT INTO ${tableName} (${columnNames.join(', ')}) VALUES (${valuePlaceholders.join(', ')})`;
+                yield this.query(insertSql, columnValues);
+                const pkFields = this.getPrimaryKeyFields();
+                const whereConditions = pkFields.map(pkField => {
+                    const column = columns.find(col => col.propertyName === pkField);
+                    const dbName = (column === null || column === void 0 ? void 0 : column.databaseName) || pkField;
+                    return `${dbName} = ?`;
+                });
+                const pkValues = pkFields.map(pkField => entity[pkField]);
+                const selectSql = `SELECT * FROM ${tableName} WHERE ${whereConditions.join(' AND ')}`;
+                const queryResult = yield this.query(selectSql, pkValues);
+                result = queryResult;
+            }
+            else {
+                const sql = `INSERT INTO ${tableName} (${columnNames.join(', ')}) VALUES (${valuePlaceholders.join(', ')})`;
+                yield this.query(sql, columnValues);
+                return entity;
+            }
+            if (result && result.length > 0) {
+                return result[0];
+            }
+            else {
+                console.warn(`Native SQL insert completed but no result returned for ${tableName}, falling back to regular save`);
+                return yield this.save(entity);
+            }
         }
         catch (error) {
             console.warn(`Native SQL insert failed for ${tableName}, falling back to regular save:`, error);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nest-omni/core",
-  "version": "4.1.2",
+  "version": "4.1.3-0",
   "description": "A comprehensive NestJS framework for building enterprise-grade applications with best practices",
   "main": "index.js",
   "types": "index.d.ts",