@naturalcycles/mysql-lib 3.11.2 → 3.12.0

package/dist/mysql.db.d.ts

@@ -50,7 +50,7 @@ export declare class MysqlDB extends BaseCommonDB implements CommonDB {
     runSQLString(s: string): Promise<void>;
     runQueryCount<ROW extends ObjectWithId>(q: DBQuery<ROW>, _opt?: CommonDBOptions): Promise<number>;
     streamQuery<ROW extends ObjectWithId, OUT = ROW>(q: DBQuery<ROW>, _opt?: MysqlDBOptions): ReadableTyped<OUT>;
-    saveBatch<ROW extends Partial<ObjectWithId>>(table: string, _rows: ROW[], opt?: MysqlDBSaveOptions<ROW>): Promise<void>;
+    saveBatch<ROW extends Partial<ObjectWithId>>(table: string, rowsInput: ROW[], opt?: MysqlDBSaveOptions<ROW>): Promise<void>;
     /**
      * Limitation: always returns [], regardless of which rows are actually deleted
      */

package/dist/mysql.db.js

@@ -171,15 +171,28 @@ class MysqlDB extends db_lib_1.BaseCommonDB {
         }));
     }
     // SAVE
-    async saveBatch(table, _rows, opt = {}) {
-        if (!_rows.length)
+    async saveBatch(table, rowsInput, opt = {}) {
+        if (!rowsInput.length)
             return;
         // Stringify object values
-        const rows = _rows.map(row => (0, js_lib_1._mapValues)(row, (_k, v) => {
+        const rows = rowsInput.map(row => (0, js_lib_1._mapValues)(row, (_k, v) => {
             return v && typeof v === 'object' && !Buffer.isBuffer(v) ? JSON.stringify(v) : v;
         }));
+        const verb = opt.saveMethod === 'insert' ? 'INSERT' : 'REPLACE';
+        if (opt.assignGeneratedIds) {
+            // Insert rows one-by-one, to get their auto-generated id
+            let i = -1;
+            for await (const row of rows) {
+                i++;
+                const sql = (0, query_util_1.insertSQL)(table, [row], verb, this.cfg.logger)[0];
+                const { insertId } = await this.runSQL({ sql });
+                // Mutate the input row with insertIt
+                rowsInput[i].id = insertId;
+            }
+            return;
+        }
         // inserts are split into multiple sentenses to respect the max_packet_size (1Mb usually)
-        const sqls = (0, query_util_1.insertSQL)(table, rows, opt.saveMethod === 'insert' ? 'INSERT' : 'REPLACE', this.cfg.logger);
+        const sqls = (0, query_util_1.insertSQL)(table, rows, verb, this.cfg.logger);
         for await (const sql of sqls) {
             await this.runSQL({ sql });
         }

package/package.json

@@ -14,7 +14,7 @@
     "@naturalcycles/dev-lib": "^12.7.5",
     "@types/node": "^17.0.5",
     "dotenv": "^16.0.0",
-    "jest": "^27.2.4"
+    "jest": "^28.1.0"
   },
   "files": [
     "dist",
@@ -36,7 +36,7 @@
   "engines": {
     "node": ">=14.15.0"
   },
-  "version": "3.11.2",
+  "version": "3.12.0",
   "description": "MySQL client implementing CommonDB interface",
   "keywords": [
     "mysql",

package/src/mysql.db.ts

@@ -23,7 +23,15 @@ import {
 } from '@naturalcycles/js-lib'
 import { ReadableTyped } from '@naturalcycles/nodejs-lib'
 import { white } from '@naturalcycles/nodejs-lib/dist/colors'
-import { Connection, Pool, PoolConfig, PoolConnection, QueryOptions, TypeCast } from 'mysql'
+import {
+  Connection,
+  OkPacket,
+  Pool,
+  PoolConfig,
+  PoolConnection,
+  QueryOptions,
+  TypeCast,
+} from 'mysql'
 import * as mysql from 'mysql'
 import { dbQueryToSQLDelete, dbQueryToSQLSelect, insertSQL } from './query.util'
 import {
@@ -266,25 +274,38 @@ export class MysqlDB extends BaseCommonDB implements CommonDB {
   // SAVE
   override async saveBatch<ROW extends Partial<ObjectWithId>>(
     table: string,
-    _rows: ROW[],
+    rowsInput: ROW[],
     opt: MysqlDBSaveOptions<ROW> = {},
   ): Promise<void> {
-    if (!_rows.length) return
+    if (!rowsInput.length) return
 
     // Stringify object values
-    const rows = _rows.map(row =>
+    const rows = rowsInput.map(row =>
       _mapValues(row, (_k, v) => {
         return v && typeof v === 'object' && !Buffer.isBuffer(v) ? JSON.stringify(v) : v
       }),
     )
 
+    const verb = opt.saveMethod === 'insert' ? 'INSERT' : 'REPLACE'
+
+    if (opt.assignGeneratedIds) {
+      // Insert rows one-by-one, to get their auto-generated id
+
+      let i = -1
+      for await (const row of rows) {
+        i++
+        const sql = insertSQL(table, [row], verb, this.cfg.logger)[0]!
+        const { insertId } = await this.runSQL<OkPacket>({ sql })
+
+        // Mutate the input row with insertIt
+        rowsInput[i]!.id = insertId
+      }
+
+      return
+    }
+
     // inserts are split into multiple sentenses to respect the max_packet_size (1Mb usually)
-    const sqls = insertSQL(
-      table,
-      rows,
-      opt.saveMethod === 'insert' ? 'INSERT' : 'REPLACE',
-      this.cfg.logger,
-    )
+    const sqls = insertSQL(table, rows, verb, this.cfg.logger)
 
     for await (const sql of sqls) {
       await this.runSQL({ sql })