@naturalcycles/mysql-lib 3.12.4 → 3.13.0

package/dist/mysql.db.d.ts

@@ -1,3 +1,4 @@
+import { DBPatch } from '@naturalcycles/db-lib';
 import { BaseCommonDB, CommonDB, CommonDBCreateOptions, CommonDBOptions, CommonDBSaveOptions, DBQuery, RunQueryResult } from '@naturalcycles/db-lib';
 import { AnyObjectWithId, CommonLogger, JsonSchemaObject, JsonSchemaRootObject, ObjectWithId } from '@naturalcycles/js-lib';
 import { ReadableTyped } from '@naturalcycles/nodejs-lib';
@@ -66,4 +67,5 @@ export declare class MysqlDB extends BaseCommonDB implements CommonDB {
     createTable<ROW extends ObjectWithId>(table: string, schema: JsonSchemaObject<ROW>, opt?: CommonDBCreateOptions): Promise<void>;
     getTables(): Promise<string[]>;
     getTableSchema<ROW extends ObjectWithId>(table: string): Promise<JsonSchemaRootObject<ROW>>;
+    updateByQuery<ROW extends ObjectWithId>(q: DBQuery<ROW>, patch: DBPatch<ROW>): Promise<number>;
 }

package/dist/mysql.db.js

@@ -191,7 +191,7 @@ class MysqlDB extends db_lib_1.BaseCommonDB {
             return v;
         }));
         const verb = opt.saveMethod === 'insert' ? 'INSERT' : 'REPLACE';
-        if (opt.assignGeneratedIds) {
+        if (opt.assignGeneratedIds && opt.saveMethod !== 'update') {
             // Insert rows one-by-one, to get their auto-generated id
             let i = -1;
             for await (const row of rows) {
@@ -204,9 +204,18 @@ class MysqlDB extends db_lib_1.BaseCommonDB {
             return;
         }
         // inserts are split into multiple sentenses to respect the max_packet_size (1Mb usually)
-        const sqls = (0, query_util_1.insertSQL)(table, rows, verb, this.cfg.logger);
-        for await (const sql of sqls) {
-            await this.runSQL({ sql });
+        if (opt.saveMethod === 'update') {
+            for await (const row of rows) {
+                (0, js_lib_1._assert)(row.id, 'Cannot update without providing an id');
+                const query = new db_lib_1.DBQuery(table).filterEq('id', row.id);
+                await this.updateByQuery(query, (0, js_lib_1._omit)(row, ['id']));
+            }
+        }
+        else {
+            const sqls = (0, query_util_1.insertSQL)(table, rows, verb, this.cfg.logger);
+            for await (const sql of sqls) {
+                await this.runSQL({ sql });
+            }
         }
     }
     // DELETE
@@ -255,6 +264,13 @@ class MysqlDB extends db_lib_1.BaseCommonDB {
         });
         return (0, mysql_schema_util_1.mysqlTableStatsToJsonSchemaField)(table, stats, this.cfg.logger);
     }
+    async updateByQuery(q, patch) {
+        const sql = (0, query_util_1.dbQueryToSQLUpdate)(q, patch);
+        if (!sql)
+            return 0;
+        const { affectedRows } = await this.runSQL({ sql });
+        return affectedRows;
+    }
 }
 tslib_1.__decorate([
     (0, js_lib_1._Memo)()

package/package.json

@@ -36,7 +36,7 @@
   "engines": {
     "node": ">=16.15.0"
   },
-  "version": "3.12.4",
+  "version": "3.13.0",
   "description": "MySQL client implementing CommonDB interface",
   "keywords": [
     "mysql",

package/src/mysql.db.ts

@@ -1,5 +1,6 @@
 import { Readable, Transform } from 'node:stream'
 import { promisify } from 'node:util'
+import { DBPatch } from '@naturalcycles/db-lib'
 import {
   BaseCommonDB,
   CommonDB,
@@ -15,6 +16,7 @@ import {
   _mapKeys,
   _mapValues,
   _Memo,
+  _omit,
   AnyObjectWithId,
   CommonLogger,
   commonLoggerPrefix,
@@ -34,7 +36,7 @@ import {
   TypeCast,
 } from 'mysql'
 import * as mysql from 'mysql'
-import { dbQueryToSQLDelete, dbQueryToSQLSelect, insertSQL } from './query.util'
+import { dbQueryToSQLDelete, dbQueryToSQLSelect, dbQueryToSQLUpdate, insertSQL } from './query.util'
 import {
   jsonSchemaToMySQLDDL,
   mapNameFromMySQL,
@@ -306,7 +308,7 @@ export class MysqlDB extends BaseCommonDB implements CommonDB {
 
     const verb = opt.saveMethod === 'insert' ? 'INSERT' : 'REPLACE'
 
-    if (opt.assignGeneratedIds) {
+    if (opt.assignGeneratedIds && opt.saveMethod !== 'update') {
       // Insert rows one-by-one, to get their auto-generated id
 
       let i = -1
@@ -323,10 +325,18 @@ export class MysqlDB extends BaseCommonDB implements CommonDB {
     }
 
     // inserts are split into multiple sentenses to respect the max_packet_size (1Mb usually)
-    const sqls = insertSQL(table, rows, verb, this.cfg.logger)
+    if (opt.saveMethod === 'update') {
+      for await (const row of rows) {
+        _assert(row.id, 'Cannot update without providing an id')
+        const query = new DBQuery(table).filterEq('id', row.id)
+        await this.updateByQuery(query, _omit(row, ['id']))
+      }
+    } else {
+      const sqls = insertSQL(table, rows, verb, this.cfg.logger)
 
-    for await (const sql of sqls) {
-      await this.runSQL({ sql })
+      for await (const sql of sqls) {
+        await this.runSQL({ sql })
+      }
     }
   }
 
@@ -393,4 +403,15 @@ export class MysqlDB extends BaseCommonDB implements CommonDB {
 
     return mysqlTableStatsToJsonSchemaField<ROW>(table, stats, this.cfg.logger)
   }
+
+  override async updateByQuery<ROW extends ObjectWithId>(
+    q: DBQuery<ROW>,
+    patch: DBPatch<ROW>,
+  ): Promise<number> {
+    const sql = dbQueryToSQLUpdate(q, patch)
+    if (!sql) return 0
+
+    const { affectedRows } = await this.runSQL<OkPacket>({ sql })
+    return affectedRows
+  }
 }