access-db/lib/redis/transaction.js

[统一数据库的连接]Unified access to all kinds of databases

"use strict";
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 fetchTransaction() {
    return new Promise((resolve_all, reject_all) => __awaiter(this, void 0, void 0, function* () {
        try {
            // 开始事务
            let begin = (callback) => {
                return new Promise((resolve, reject) => __awaiter(this, void 0, void 0, function* () {
                    yield dbRedis_1.redisClient.sendCommand(['MULTI']);
                    yield callback();
                    resolve(true);
                }));
            };
            // 触发事务
            let commit = () => {
                return new Promise((resolve, reject) => __awaiter(this, void 0, void 0, function* () {
                    const res = yield dbRedis_1.redisClient.sendCommand(['EXEC']);
                    resolve(res);
                }));
            };
            // 放弃事务
            let discard = () => {
                return new Promise((resolve, reject) => __awaiter(this, void 0, void 0, function* () {
                    yield dbRedis_1.redisClient.sendCommand(['DISCARD']);
                    resolve(true);
                }));
            };
            //check-and-set实现乐观锁 
            let watch = (table, id) => {
                return new Promise((resolve, reject) => __awaiter(this, void 0, void 0, function* () {
                    let hkey = [];
                    if (id) {
                        if ((0, utils_1.isArray)(id)) {
                            let tempIds = id;
                            for (let i = 0; i < tempIds.length; i++) {
                                hkey.push((0, dbRedis_1.reTable)(table) + ':' + tempIds[i]);
                            }
                        }
                        else {
                            hkey = [(0, dbRedis_1.reTable)(table) + ':' + id];
                        }
                    }
                    else {
                        hkey = [(0, dbRedis_1.reTable)(table)];
                    }
                    yield dbRedis_1.redisClient.sendCommand(['WATCH', ...hkey]);
                    resolve(true);
                }));
            };
            resolve_all({
                begin,
                discard,
                commit,
                watch,
            });
        }
        catch (err) {
            reject_all(err);
        }
    }));
}
exports.default = fetchTransaction;