summary refs log tree commit diff
path: root/src/util/cache/LocalCache.ts
blob: 9c4f23b397327104ec6c292f4363007cfa290faf (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
/*
	Fosscord: A FOSS re-implementation and extension of the Discord.com backend.
	Copyright (C) 2023 Fosscord and Fosscord Contributors
	
	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU Affero General Public License as published
	by the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.
	
	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU Affero General Public License for more details.
	
	You should have received a copy of the GNU Affero General Public License
	along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/
/* eslint-disable @typescript-eslint/ban-ts-comment */
import { BaseEntityWithId, Cache } from "./Cache";

export const cacheTimeout = 1000 * 60 * 5;

export class LocalCache extends Map<string, BaseEntityWithId> implements Cache {
	last_access = new Map<string, number>();

	constructor() {
		super();

		setInterval(() => {
			const now = Date.now();
			for (const [key, value] of this.last_access) {
				if (now - value > cacheTimeout) {
					this.delete(key);
					this.last_access.delete(key);
				}
			}
		}, cacheTimeout);
	}

	set(key: string, value: BaseEntityWithId): this {
		if (this.has(key)) {
			this.update(key, value);
			return this;
		}
		this.last_access.set(key, Date.now());
		return super.set(key, value as never);
	}

	get(key: string) {
		const value = super.get(key);
		if (value) this.last_access.set(key, Date.now());
		return value;
	}

	update(id: string, entity: BaseEntityWithId) {
		const oldEntity = this.get(id);
		if (!oldEntity) return;
		for (const key in entity) {
			// @ts-ignore
			if (entity[key] === undefined) continue; // @ts-ignore
			oldEntity[key] = entity[key];
		}
	}

	find(options: Record<string, never>): BaseEntityWithId | undefined {
		if (options.id && Object.keys(options).length === 1) {
			return this.get(options.id);
		}
		for (const entity of this.values()) {
			if (objectFulfillsQuery(entity, options)) return entity;
		}
	}

	filter(options: Record<string, never>): BaseEntityWithId[] {
		const result = [];
		for (const entity of this.values()) {
			if (objectFulfillsQuery(entity, options)) {
				result.push(entity);
			}
		}
		return result;
	}
}

function objectFulfillsQuery(
	entity: BaseEntityWithId,
	options: Record<string, never>,
) {
	for (const key in options) {
		// @ts-ignore
		if (entity[key] !== options[key]) return false;
	}
	return true;
}