初始化项目

15815213711

2024-08-26 67b8b6731811983447e053d4396b3708c14dfe3c

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
95
96
97
98
99
100
101
102
103
'use strict';
 
const EventEmitter = require('events');
const JSONB = require('json-buffer');
 
const loadStore = opts => {
    const adapters = {
        redis: '@keyv/redis',
        mongodb: '@keyv/mongo',
        mongo: '@keyv/mongo',
        sqlite: '@keyv/sqlite',
        postgresql: '@keyv/postgres',
        postgres: '@keyv/postgres',
        mysql: '@keyv/mysql'
    };
    if (opts.adapter || opts.uri) {
        const adapter = opts.adapter || /^[^:]*/.exec(opts.uri)[0];
        return new (require(adapters[adapter]))(opts);
    }
    return new Map();
};
 
class Keyv extends EventEmitter {
    constructor(uri, opts) {
        super();
        this.opts = Object.assign(
            {
                namespace: 'keyv',
                serialize: JSONB.stringify,
                deserialize: JSONB.parse
            },
            (typeof uri === 'string') ? { uri } : uri,
            opts
        );
 
        if (!this.opts.store) {
            const adapterOpts = Object.assign({}, this.opts);
            this.opts.store = loadStore(adapterOpts);
        }
 
        if (typeof this.opts.store.on === 'function') {
            this.opts.store.on('error', err => this.emit('error', err));
        }
 
        this.opts.store.namespace = this.opts.namespace;
    }
 
    _getKeyPrefix(key) {
        return `${this.opts.namespace}:${key}`;
    }
 
    get(key) {
        key = this._getKeyPrefix(key);
        const store = this.opts.store;
        return Promise.resolve()
            .then(() => store.get(key))
            .then(data => {
                data = (typeof data === 'string') ? this.opts.deserialize(data) : data;
                if (data === undefined) {
                    return undefined;
                }
                if (typeof data.expires === 'number' && Date.now() > data.expires) {
                    this.delete(key);
                    return undefined;
                }
                return data.value;
            });
    }
 
    set(key, value, ttl) {
        key = this._getKeyPrefix(key);
        if (typeof ttl === 'undefined') {
            ttl = this.opts.ttl;
        }
        if (ttl === 0) {
            ttl = undefined;
        }
        const store = this.opts.store;
 
        return Promise.resolve()
            .then(() => {
                const expires = (typeof ttl === 'number') ? (Date.now() + ttl) : null;
                value = { value, expires };
                return store.set(key, this.opts.serialize(value), ttl);
            })
            .then(() => true);
    }
 
    delete(key) {
        key = this._getKeyPrefix(key);
        const store = this.opts.store;
        return Promise.resolve()
            .then(() => store.delete(key));
    }
 
    clear() {
        const store = this.opts.store;
        return Promise.resolve()
            .then(() => store.clear());
    }
}
 
module.exports = Keyv;