Memory RxStorage: Add categorizedByWriteInput property that can be …

…used by other plugins
pubkey · Apr 25, 2024 · 69da30e · 69da30e
1 parent e30da3d
commit 69da30e
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,7 +2,7 @@
 # RxDB Changelog
 
 <!-- CHANGELOG NEWEST -->
-
+- Memory RxStorage: Add `categorizedByWriteInput` property that can be used by other plugins.
 <!-- ADD new changes here! -->
 
 <!-- /CHANGELOG NEWEST -->

diff --git a/src/plugins/storage-memory/rx-storage-instance-memory.ts b/src/plugins/storage-memory/rx-storage-instance-memory.ts
@@ -12,6 +12,7 @@ import {
 } from '../../rx-storage-helper.ts';
 import type {
     BulkWriteRow,
+    CategorizeBulkWriteRowsOutput,
     EventBulk,
     PreparedQuery,
     QueryMatcher,
@@ -79,6 +80,14 @@ export class RxStorageInstanceMemory<RxDocType> implements RxStorageInstance<
     public readonly primaryPath: StringKeys<RxDocumentData<RxDocType>>;
     public closed = false;
 
+    /**
+     * Used by some plugins and storage wrappers
+     * to find out details about the internals of a write operation.
+     * For example if you want to know which documents really have been replaced
+     * or newly inserted.
+     */
+    public categorizedByWriteInput = new WeakMap<BulkWriteRow<RxDocType>[], CategorizeBulkWriteRowsOutput<RxDocType>>();
+
     constructor(
         public readonly storage: RxStorageMemory,
         public readonly databaseName: string,
@@ -124,6 +133,7 @@ export class RxStorageInstanceMemory<RxDocType> implements RxStorageInstance<
             success.push(doc);
         }
 
+        this.categorizedByWriteInput.set(documentWrites, categorized);
         this.internals.ensurePersistenceTask = categorized;
         if (!this.internals.ensurePersistenceIdlePromise) {
             this.internals.ensurePersistenceIdlePromise = requestIdlePromiseNoQueue().then(() => {

diff --git a/src/rx-storage-helper.ts b/src/rx-storage-helper.ts
@@ -111,7 +111,6 @@ export function observeSingle<RxDocType>(
     return ret;
 }
 
-
 /**
  * Checkpoints must be stackable over another.
  * This is required form some RxStorage implementations