rxdb
Version:
A local-first realtime NoSQL Database for JavaScript applications - https://rxdb.info/
108 lines (103 loc) • 3.9 kB
JavaScript
import { PROMISE_RESOLVE_TRUE, getFromMapOrCreate } from "../../plugins/utils/index.js";
import { REPLICATION_STATE_BY_COLLECTION } from "../replication/index.js";
import { DEFAULT_CLEANUP_POLICY } from "./cleanup-helper.js";
import { runAsyncPluginHooks } from "../../hooks.js";
import { firstValueFrom } from 'rxjs';
/**
* Even on multiple databases,
* the calls to RxStorage().cleanup()
* must never run in parallel.
* The cleanup is a background task which should
* not affect the performance of other, more important tasks.
*/
var RXSTORAGE_CLEANUP_QUEUE = PROMISE_RESOLVE_TRUE;
export async function startCleanupForRxCollection(rxCollection) {
var rxDatabase = rxCollection.database;
var cleanupPolicy = Object.assign({}, DEFAULT_CLEANUP_POLICY, rxDatabase.cleanupPolicy ? rxDatabase.cleanupPolicy : {});
await initialCleanupWait(rxCollection, cleanupPolicy);
if (rxCollection.closed) {
return;
}
// initially cleanup the collection
await cleanupRxCollection(rxCollection, cleanupPolicy);
/**
* Afterwards we listen to deletes
* and only re-run the cleanup after
* minimumDeletedTime is reached.
*/
await runCleanupAfterDelete(rxCollection, cleanupPolicy);
}
export async function initialCleanupWait(collection, cleanupPolicy) {
/**
* Wait until minimumDatabaseInstanceAge is reached
* or collection is closed.
*/
await collection.promiseWait(cleanupPolicy.minimumCollectionAge);
if (collection.closed) {
return;
}
if (collection.database.multiInstance && cleanupPolicy.waitForLeadership) {
await collection.database.waitForLeadership();
}
}
/**
* Runs the cleanup for a single RxCollection
*/
export async function cleanupRxCollection(rxCollection, cleanupPolicy) {
var rxDatabase = rxCollection.database;
var storageInstance = rxCollection.storageInstance;
// run cleanup() until it returns true
var isDone = false;
while (!isDone && !rxCollection.closed) {
if (cleanupPolicy.awaitReplicationsInSync) {
var replicationStates = getFromMapOrCreate(REPLICATION_STATE_BY_COLLECTION, rxCollection, () => []);
await Promise.all(replicationStates.map(replicationState => {
if (!replicationState.isStopped()) {
return replicationState.awaitInSync();
}
}));
}
if (rxCollection.closed) {
return;
}
RXSTORAGE_CLEANUP_QUEUE = RXSTORAGE_CLEANUP_QUEUE.then(async () => {
if (rxCollection.closed) {
return true;
}
await rxDatabase.requestIdlePromise();
var allDone = [];
allDone.push(storageInstance.cleanup(cleanupPolicy.minimumDeletedTime));
var replicationStates = getFromMapOrCreate(REPLICATION_STATE_BY_COLLECTION, rxCollection, () => []);
for (var replicationState of replicationStates) {
var meta = replicationState.metaInstance;
if (meta) {
allDone.push(meta.cleanup(cleanupPolicy.minimumDeletedTime));
}
}
var hasFalse = (await Promise.all(allDone)).find(v => !v);
return !hasFalse;
});
isDone = await RXSTORAGE_CLEANUP_QUEUE;
}
await runAsyncPluginHooks('postCleanup', {
collectionName: rxCollection.name,
databaseName: rxDatabase.name
});
}
export async function runCleanupAfterDelete(rxCollection, cleanupPolicy) {
while (!rxCollection.closed) {
/**
* In theory we should wait here until a document is deleted.
* But this would mean we have to search through all events ever processed.
* So instead we just wait for any write event and then we anyway throttle
* the calls with the promiseWait() below.
*/
await firstValueFrom(rxCollection.eventBulks$).catch(() => {});
await rxCollection.promiseWait(cleanupPolicy.runEach);
if (rxCollection.closed) {
return;
}
await cleanupRxCollection(rxCollection, cleanupPolicy);
}
}
//# sourceMappingURL=cleanup.js.map