94 lines
2.7 KiB
JavaScript
94 lines
2.7 KiB
JavaScript
"use strict";
|
|
|
|
Object.defineProperty(exports, "__esModule", {
|
|
value: true
|
|
});
|
|
|
|
var _utils = require("./utils/index.cjs");
|
|
var _ioredis = _interopRequireDefault(require("ioredis"));
|
|
function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
|
|
const DRIVER_NAME = "redis";
|
|
module.exports = (0, _utils.defineDriver)(opts => {
|
|
let redisClient;
|
|
const getRedisClient = () => {
|
|
if (redisClient) {
|
|
return redisClient;
|
|
}
|
|
if (opts.cluster) {
|
|
redisClient = new _ioredis.default.Cluster(opts.cluster, opts.clusterOptions);
|
|
} else if (opts.url) {
|
|
redisClient = new _ioredis.default(opts.url, opts);
|
|
} else {
|
|
redisClient = new _ioredis.default(opts);
|
|
}
|
|
return redisClient;
|
|
};
|
|
const base = (opts.base || "").replace(/:$/, "");
|
|
const p = (...keys) => (0, _utils.joinKeys)(base, ...keys);
|
|
const d = key => base ? key.replace(`${base}:`, "") : key;
|
|
if (opts.preConnect) {
|
|
try {
|
|
getRedisClient();
|
|
} catch (error) {
|
|
console.error(error);
|
|
}
|
|
}
|
|
const scan = async pattern => {
|
|
const client = getRedisClient();
|
|
const keys = [];
|
|
let cursor = "0";
|
|
do {
|
|
const [nextCursor, scanKeys] = opts.scanCount ? await client.scan(cursor, "MATCH", pattern, "COUNT", opts.scanCount) : await client.scan(cursor, "MATCH", pattern);
|
|
cursor = nextCursor;
|
|
keys.push(...scanKeys);
|
|
} while (cursor !== "0");
|
|
return keys;
|
|
};
|
|
return {
|
|
name: DRIVER_NAME,
|
|
options: opts,
|
|
getInstance: getRedisClient,
|
|
async hasItem(key) {
|
|
return Boolean(await getRedisClient().exists(p(key)));
|
|
},
|
|
async getItem(key) {
|
|
const value = await getRedisClient().get(p(key));
|
|
return value ?? null;
|
|
},
|
|
async getItems(items) {
|
|
const keys = items.map(item => p(item.key));
|
|
const data = await getRedisClient().mget(...keys);
|
|
return keys.map((key, index) => {
|
|
return {
|
|
key: d(key),
|
|
value: data[index] ?? null
|
|
};
|
|
});
|
|
},
|
|
async setItem(key, value, tOptions) {
|
|
const ttl = tOptions?.ttl ?? opts.ttl;
|
|
if (ttl) {
|
|
await getRedisClient().set(p(key), value, "EX", ttl);
|
|
} else {
|
|
await getRedisClient().set(p(key), value);
|
|
}
|
|
},
|
|
async removeItem(key) {
|
|
await getRedisClient().unlink(p(key));
|
|
},
|
|
async getKeys(base2) {
|
|
const keys = await scan(p(base2, "*"));
|
|
return keys.map(key => d(key));
|
|
},
|
|
async clear(base2) {
|
|
const keys = await scan(p(base2, "*"));
|
|
if (keys.length === 0) {
|
|
return;
|
|
}
|
|
await getRedisClient().unlink(keys);
|
|
},
|
|
dispose() {
|
|
return getRedisClient().disconnect();
|
|
}
|
|
};
|
|
}); |