-
Notifications
You must be signed in to change notification settings - Fork 1
/
memory.ts
68 lines (54 loc) · 1.69 KB
/
memory.ts
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
import { Snowflake } from "../../types";
import { CacheStore, NonParentCacheStore } from ".";
export const create = <T>(): CacheStore<T> => {
const map = new Map<string, T>();
const parentMap = new Map<Snowflake, Map<string, T>>();
return {
size: () => Promise.resolve(map.size),
sizeForParent: (parentId) =>
Promise.resolve(parentMap.get(parentId)?.size ?? 0),
get: (resourceId) => Promise.resolve(map.get(resourceId)),
getForParent: (parentId) => {
const map = parentMap.get(parentId) || new Map<string, T>();
return Promise.resolve(map);
},
set: (parentId, resourceId, resource) => {
map.set(resourceId, resource);
if (!parentMap.has(parentId)) {
parentMap.set(parentId, new Map());
}
parentMap.get(parentId)!.set(resourceId, resource);
return Promise.resolve();
},
delete: (parentId, resourceId) => {
map.delete(resourceId);
parentMap.get(parentId)?.delete(resourceId);
return Promise.resolve();
},
parentDelete: (parentId) => {
const ids = parentMap.get(parentId)?.keys();
if (ids) {
for (const id in ids) {
map.delete(id);
}
}
parentMap.delete(parentId);
return Promise.resolve();
},
};
};
export const createNonParent = <T>(): NonParentCacheStore<T> => {
const map = new Map<string, T>();
return {
size: () => Promise.resolve(map.size),
get: (resourceId) => Promise.resolve(map.get(resourceId)),
set: (resourceId, resource) => {
map.set(resourceId, resource);
return Promise.resolve();
},
delete: (resourceId) => {
map.delete(resourceId);
return Promise.resolve();
},
};
};