This repository has been archived by the owner on Jan 31, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 7
/
memory_adapter.js
83 lines (62 loc) · 1.6 KB
/
memory_adapter.js
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
const LRU = require('lru-cache');
const epochTime = require('../helpers/epoch_time');
const storage = new LRU({});
function grantKeyFor(id) {
return `grant:${id}`;
}
function userCodeKeyFor(userCode) {
return `userCode:${userCode}`;
}
class MemoryAdapter {
constructor(name) {
this.name = name;
}
key(id) {
return `${this.name}:${id}`;
}
destroy(id) {
const key = this.key(id);
const found = storage.get(key);
const grantId = found && found.grantId;
storage.del(key);
if (grantId) {
const grantKey = grantKeyFor(grantId);
storage.get(grantKey).forEach(token => storage.del(token));
storage.del(grantKey);
}
return Promise.resolve();
}
consume(id) {
storage.get(this.key(id)).consumed = epochTime();
return Promise.resolve();
}
find(id) {
return Promise.resolve(storage.get(this.key(id)));
}
findByUserCode(userCode) {
const id = storage.get(userCodeKeyFor(userCode));
return this.find(id);
}
upsert(id, payload, expiresIn) {
const key = this.key(id);
const { grantId, userCode } = payload;
if (grantId) {
const grantKey = grantKeyFor(grantId);
const grant = storage.get(grantKey);
if (!grant) {
storage.set(grantKey, [key]);
} else {
grant.push(key);
}
}
if (userCode) {
storage.set(userCodeKeyFor(userCode), id, expiresIn * 1000);
}
storage.set(key, payload, expiresIn * 1000);
return Promise.resolve();
}
static connect(provider) { // eslint-disable-line no-unused-vars
// noop
}
}
module.exports = MemoryAdapter;