access-db
Version:
[统一数据库的连接]Unified access to all kinds of databases
69 lines (68 loc) • 3.1 kB
JavaScript
;
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
const dbRedis_1 = require("../utils/dbRedis");
const utils_1 = require("../utils/utils");
function fetchQueue(queueName, method, params, unique) {
return new Promise((resolve, reject) => __awaiter(this, void 0, void 0, function* () {
try {
const queue = (0, dbRedis_1.reTable)(queueName);
const _queue = (0, dbRedis_1.reTable)(queueName + '_queue_backups');
let func = [], tempParams = [];
if ((0, utils_1.isArray)(params)) {
tempParams = params;
}
else if (typeof (params) === 'string') {
tempParams = [params];
}
switch (method) {
case 'push':
if (tempParams.length === 0)
break;
if (unique) {
for (let j = 0; j < tempParams.length; j++) {
func.push(dbRedis_1.redisClient.LREM(queue, 0, tempParams[j]));
}
}
func.push(dbRedis_1.redisClient.LPUSH(queue, tempParams.reverse()));
break;
case 'pop':
for (let i = 0; i < params; i++) {
func.unshift(dbRedis_1.redisClient.RPOPLPUSH(queue, _queue));
}
break;
case 'remove':
if (tempParams.length === 0)
break;
for (let j = 0; j < tempParams.length; j++) {
func.push(dbRedis_1.redisClient.LREM(_queue, unique ? 0 : 1, tempParams[j]));
}
break;
case 'get':
break;
default:
break;
}
func.push(dbRedis_1.redisClient.LRANGE(queue, 0, -1));
func.push(dbRedis_1.redisClient.LRANGE(_queue, 0, -1));
const rData = yield Promise.all(func);
resolve({ data: {
queue: rData[rData.length - 2],
pop: method === 'pop' ? rData.slice(0, rData.length - 2) : [],
_queue: rData[rData.length - 1],
} });
}
catch (err) {
reject(err);
}
}));
}
exports.default = fetchQueue;