mirror of https://gitgud.io/fatchan/jschan.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
154 lines
3.3 KiB
154 lines
3.3 KiB
'use strict';
|
|
|
|
const Redis = require('ioredis')
|
|
, secrets = require(__dirname+'/configs/secrets.js')
|
|
, sharedClient = new Redis(secrets.redis)
|
|
, subscriber = new Redis(secrets.redis)
|
|
, publisher = new Redis(secrets.redis)
|
|
, messageCallbacks = {
|
|
'config': [],
|
|
'roles': [],
|
|
};
|
|
|
|
module.exports = {
|
|
|
|
redisClient: sharedClient,
|
|
redisSubscriber: subscriber,
|
|
redisPublisher: publisher,
|
|
|
|
close: () => {
|
|
sharedClient.quit();
|
|
publisher.quit();
|
|
subscriber.quit();
|
|
},
|
|
|
|
addCallback: (channel, cb) => {
|
|
if (messageCallbacks[channel].length === 0) {
|
|
subscriber.subscribe('config', (err, count) => {
|
|
if (err) {
|
|
return console.error(err);
|
|
}
|
|
});
|
|
subscriber.subscribe('roles', (err, count) => {
|
|
if (err) {
|
|
return console.error(err);
|
|
}
|
|
});
|
|
subscriber.on("message", (channel, message) => {
|
|
secrets.debugLogs && console.log(`Subscriber message from channel ${channel}`);
|
|
let data;
|
|
if (message) {
|
|
data = JSON.parse(message);
|
|
}
|
|
messageCallbacks[channel].forEach(cb => {
|
|
cb(data);
|
|
});
|
|
});
|
|
}
|
|
messageCallbacks[channel].push(cb);
|
|
},
|
|
|
|
//get a value with key
|
|
get: (key) => {
|
|
return sharedClient.get(key).then(res => { return JSON.parse(res) });
|
|
},
|
|
|
|
//set a value on key
|
|
set: (key, value, ttl) => {
|
|
if (ttl) {
|
|
return sharedClient.set(key, JSON.stringify(value), 'EX', ttl);
|
|
} else {
|
|
return sharedClient.set(key, JSON.stringify(value));
|
|
}
|
|
},
|
|
|
|
//set a value on key if not exist
|
|
setnx: (key, value) => {
|
|
return sharedClient.setnx(key, JSON.stringify(value));
|
|
},
|
|
|
|
//add items to a set
|
|
sadd: (key, value) => {
|
|
return sharedClient.sadd(key, value);
|
|
},
|
|
|
|
//get all members of a set
|
|
sgetall: (key) => {
|
|
return sharedClient.smembers(key);
|
|
},
|
|
|
|
//remove an item from a set
|
|
srem: (key, value) => {
|
|
return sharedClient.srem(key, value);
|
|
},
|
|
|
|
//get random item from set
|
|
srand: (key) => {
|
|
return sharedClient.srandmember(key);
|
|
},
|
|
|
|
//delete value with key
|
|
del: (keyOrKeys) => {
|
|
if (Array.isArray(keyOrKeys)) {
|
|
return sharedClient.del(...keyOrKeys);
|
|
} else {
|
|
return sharedClient.del(keyOrKeys);
|
|
}
|
|
},
|
|
|
|
getPattern: (pattern) => {
|
|
return new Promise((resolve, reject) => {
|
|
const stream = sharedClient.scanStream({
|
|
match: pattern
|
|
});
|
|
let allKeys = [];
|
|
stream.on('data', (keys) => {
|
|
allKeys = allKeys.concat(keys);
|
|
});
|
|
stream.on('end', async () => {
|
|
const pipeline = sharedClient.pipeline();
|
|
for (let i = 0; i < allKeys.length; i++) {
|
|
pipeline.get(allKeys[i]);
|
|
}
|
|
let results;
|
|
try {
|
|
results = await pipeline.exec();
|
|
} catch(e) {
|
|
return reject(e);
|
|
}
|
|
const data = {};
|
|
for (let i = 0; i < results.length; i++) {
|
|
data[allKeys[i]] = JSON.parse(results[i][1]);
|
|
}
|
|
resolve(data);
|
|
});
|
|
stream.on('error', (err) => {
|
|
reject(err);
|
|
});
|
|
});
|
|
},
|
|
|
|
deletePattern: (pattern) => {
|
|
return new Promise((resolve, reject) => {
|
|
const stream = sharedClient.scanStream({
|
|
match: pattern
|
|
});
|
|
stream.on('data', (keys) => {
|
|
if (keys.length > 0) {
|
|
const pipeline = sharedClient.pipeline();
|
|
for (let i = 0; i < keys.length; i++) {
|
|
pipeline.del(keys[i]);
|
|
}
|
|
pipeline.exec();
|
|
}
|
|
});
|
|
stream.on('end', () => {
|
|
resolve();
|
|
});
|
|
stream.on('error', (err) => {
|
|
reject(err);
|
|
});
|
|
});
|
|
},
|
|
|
|
}
|
|
|