@naturalcycles/firestore-lib 1.1.12 → 1.2.0

package/dist/firestore.db.d.ts

@@ -7,7 +7,7 @@ export interface FirestoreDBCfg {
 }
 export interface FirestoreDBOptions extends CommonDBOptions {
 }
-export interface FirestoreDBSaveOptions<ROW extends ObjectWithId = AnyObjectWithId> extends CommonDBSaveOptions<ROW> {
+export interface FirestoreDBSaveOptions<ROW extends Partial<ObjectWithId> = AnyObjectWithId> extends CommonDBSaveOptions<ROW> {
 }
 export declare class FirestoreDB extends BaseCommonDB implements CommonDB {
     cfg: FirestoreDBCfg;
@@ -18,7 +18,7 @@ export declare class FirestoreDB extends BaseCommonDB implements CommonDB {
     runFirestoreQuery<ROW extends ObjectWithId>(q: Query, _opt?: FirestoreDBOptions): Promise<ROW[]>;
     runQueryCount<ROW extends ObjectWithId>(q: DBQuery<ROW>, opt?: FirestoreDBOptions): Promise<number>;
     streamQuery<ROW extends ObjectWithId>(q: DBQuery<ROW>, _opt?: CommonDBStreamOptions): ReadableTyped<ROW>;
-    saveBatch<ROW extends ObjectWithId>(table: string, rows: ROW[], _opt?: FirestoreDBSaveOptions<ROW>): Promise<void>;
+    saveBatch<ROW extends Partial<ObjectWithId>>(table: string, rows: ROW[], opt?: FirestoreDBSaveOptions<ROW>): Promise<void>;
     deleteByQuery<ROW extends ObjectWithId>(q: DBQuery<ROW>, opt?: FirestoreDBOptions): Promise<number>;
     deleteByIds<ROW extends ObjectWithId>(table: string, ids: ROW['id'][], _opt?: FirestoreDBOptions): Promise<number>;
     private querySnapshotToArray;

package/dist/firestore.db.js

@@ -6,6 +6,11 @@ const js_lib_1 = require("@naturalcycles/js-lib");
 const nodejs_lib_1 = require("@naturalcycles/nodejs-lib");
 const firestore_util_1 = require("./firestore.util");
 const query_util_1 = require("./query.util");
+const methodMap = {
+    insert: 'create',
+    update: 'update',
+    upsert: 'set',
+};
 class FirestoreDB extends db_lib_1.BaseCommonDB {
     constructor(cfg) {
         super();
@@ -54,12 +59,14 @@ class FirestoreDB extends db_lib_1.BaseCommonDB {
         }));
     }
     // SAVE
-    async saveBatch(table, rows, _opt) {
+    async saveBatch(table, rows, opt = {}) {
+        const method = methodMap[opt.saveMethod] || 'set';
         // Firestore allows max 500 items in one batch
         await (0, js_lib_1.pMap)((0, js_lib_1._chunk)(rows, 500), async (chunk) => {
             const batch = this.cfg.firestore.batch();
             chunk.forEach(row => {
-                batch.set(this.cfg.firestore.collection(table).doc((0, firestore_util_1.escapeDocId)(row.id)), (0, js_lib_1._filterUndefinedValues)(row));
+                (0, js_lib_1._assert)(row.id, `firestore-db doesn't support id auto-generation, but empty id was provided in saveBatch`);
+                batch[method](this.cfg.firestore.collection(table).doc((0, firestore_util_1.escapeDocId)(row.id)), (0, js_lib_1._filterUndefinedValues)(row));
             });
             await batch.commit();
         }, { concurrency: 1 });

package/package.json

@@ -37,7 +37,7 @@
   "engines": {
     "node": ">=14.15.0"
   },
-  "version": "1.1.12",
+  "version": "1.2.0",
   "description": "Firestore implementation of CommonDB interface",
   "author": "Natural Cycles Team",
   "license": "MIT"

package/src/firestore.db.ts

@@ -3,6 +3,7 @@ import {
   BaseCommonDB,
   CommonDB,
   CommonDBOptions,
+  CommonDBSaveMethod,
   CommonDBSaveOptions,
   CommonDBStreamOptions,
   DBQuery,
@@ -17,6 +18,7 @@ import {
   _filterUndefinedValues,
   ObjectWithId,
   AnyObjectWithId,
+  _assert,
 } from '@naturalcycles/js-lib'
 import { ReadableTyped, transformMapSimple } from '@naturalcycles/nodejs-lib'
 import { escapeDocId, unescapeDocId } from './firestore.util'
@@ -27,9 +29,15 @@ export interface FirestoreDBCfg {
 }
 
 export interface FirestoreDBOptions extends CommonDBOptions {}
-export interface FirestoreDBSaveOptions<ROW extends ObjectWithId = AnyObjectWithId>
+export interface FirestoreDBSaveOptions<ROW extends Partial<ObjectWithId> = AnyObjectWithId>
   extends CommonDBSaveOptions<ROW> {}
 
+const methodMap: Record<CommonDBSaveMethod, string> = {
+  insert: 'create',
+  update: 'update',
+  upsert: 'set',
+}
+
 export class FirestoreDB extends BaseCommonDB implements CommonDB {
   constructor(public cfg: FirestoreDBCfg) {
     super()
@@ -116,11 +124,13 @@ export class FirestoreDB extends BaseCommonDB implements CommonDB {
   }
 
   // SAVE
-  override async saveBatch<ROW extends ObjectWithId>(
+  override async saveBatch<ROW extends Partial<ObjectWithId>>(
     table: string,
     rows: ROW[],
-    _opt?: FirestoreDBSaveOptions<ROW>,
+    opt: FirestoreDBSaveOptions<ROW> = {},
   ): Promise<void> {
+    const method = methodMap[opt.saveMethod!] || 'set'
+
     // Firestore allows max 500 items in one batch
     await pMap(
       _chunk(rows, 500),
@@ -128,7 +138,12 @@ export class FirestoreDB extends BaseCommonDB implements CommonDB {
         const batch = this.cfg.firestore.batch()
 
         chunk.forEach(row => {
-          batch.set(
+          _assert(
+            row.id,
+            `firestore-db doesn't support id auto-generation, but empty id was provided in saveBatch`,
+          )
+
+          batch[method](
             this.cfg.firestore.collection(table).doc(escapeDocId(row.id)),
             _filterUndefinedValues(row),
           )