summaryrefslogtreecommitdiff
path: root/packages/backend/src/misc/collapsed-queue.ts
diff options
context:
space:
mode:
authorKOBA789 <kobahide789@gmail.com>2024-09-26 10:25:20 +0900
committerGitHub <noreply@github.com>2024-09-26 10:25:20 +0900
commit7134d24c1f25859e7e092f757ecd327469d75a8f (patch)
tree672ebd9c7dbb888a1f7e485fc779f9d61ac3bc8f /packages/backend/src/misc/collapsed-queue.ts
parentUpdate CHANGELOG.md (diff)
downloadsharkey-7134d24c1f25859e7e092f757ecd327469d75a8f.tar.gz
sharkey-7134d24c1f25859e7e092f757ecd327469d75a8f.tar.bz2
sharkey-7134d24c1f25859e7e092f757ecd327469d75a8f.zip
perf(backend): Defer instance metadata update (#14558)
* Defer instance metadata update * Fix last new line * Fix typo * Add license notice * Fix syntax * Perform deferred jobs on shutdown * Fix missing async/await * Fix typo :) * Update collapsed-queue.ts --------- Co-authored-by: syuilo <4439005+syuilo@users.noreply.github.com>
Diffstat (limited to 'packages/backend/src/misc/collapsed-queue.ts')
-rw-r--r--packages/backend/src/misc/collapsed-queue.ts44
1 files changed, 44 insertions, 0 deletions
diff --git a/packages/backend/src/misc/collapsed-queue.ts b/packages/backend/src/misc/collapsed-queue.ts
new file mode 100644
index 0000000000..5bc20a78ae
--- /dev/null
+++ b/packages/backend/src/misc/collapsed-queue.ts
@@ -0,0 +1,44 @@
+/*
+ * SPDX-FileCopyrightText: syuilo and misskey-project
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+type Job<V> = {
+ value: V;
+ timer: NodeJS.Timeout;
+};
+
+// TODO: redis使えるようにする
+export class CollapsedQueue<K, V> {
+ private jobs: Map<K, Job<V>> = new Map();
+
+ constructor(
+ private timeout: number,
+ private collapse: (oldValue: V, newValue: V) => V,
+ private perform: (key: K, value: V) => Promise<void>,
+ ) {}
+
+ enqueue(key: K, value: V) {
+ if (this.jobs.has(key)) {
+ const old = this.jobs.get(key)!;
+ const merged = this.collapse(old.value, value);
+ this.jobs.set(key, { ...old, value: merged });
+ } else {
+ const timer = setTimeout(() => {
+ const job = this.jobs.get(key)!;
+ this.jobs.delete(key);
+ this.perform(key, job.value);
+ }, this.timeout);
+ this.jobs.set(key, { value, timer });
+ }
+ }
+
+ async performAllNow() {
+ const entries = [...this.jobs.entries()];
+ this.jobs.clear();
+ for (const [_key, job] of entries) {
+ clearTimeout(job.timer);
+ }
+ await Promise.allSettled(entries.map(([key, job]) => this.perform(key, job.value)));
+ }
+}