"use strict";
/**
 * redis模块
 * @date 2022-03
 * @author 何华东
 */
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());
    });
};
var __generator = (this && this.__generator) || function (thisArg, body) {
    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
    return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
    function verb(n) { return function (v) { return step([n, v]); }; }
    function step(op) {
        if (f) throw new TypeError("Generator is already executing.");
        while (_) try {
            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
            if (y = 0, t) op = [op[0] & 2, t.value];
            switch (op[0]) {
                case 0: case 1: t = op; break;
                case 4: _.label++; return { value: op[1], done: false };
                case 5: _.label++; y = op[1]; op = [0]; continue;
                case 7: op = _.ops.pop(); _.trys.pop(); continue;
                default:
                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
                    if (t[2]) _.ops.pop();
                    _.trys.pop(); continue;
            }
            op = body.call(thisArg, _);
        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
    }
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.RedisLocker = exports.redisClient = void 0;
var mslib_1 = require("mslib");
var Config_1 = require("./Config");
var ConfigCenter_1 = require("./ConfigCenter");
var Contants_1 = require("./Contants");
var Throttle_1 = require("./Throttle");
var Global_1 = require("./Global");
var Utils_1 = require("./Utils");
var IORedis = require("ioredis");
var Date_1 = require("./Date");
/**
 * redis客户端实例
 *
 * 1.`Ioc`注册`ioredis`配置将会自动创建该实例; @see {@link Config}
 *
 * 2.如已引用配置依赖[NsConfig](https://me.keytop.cn/service/config)将会自动创建该实例;
 *
 * @see [ioredis](https://www.npmjs.com/package/ioredis)
 */
exports.redisClient = null;
init();
/**
 * 场端从sqlite获取redis密码
 */
function getRedisPwdFromSqlite() {
    var _a;
    return __awaiter(this, void 0, void 0, function () {
        var pwd, knexInstance, rows;
        return __generator(this, function (_b) {
            switch (_b.label) {
                case 0:
                    knexInstance = require('knex')({
                        client: 'sqlite3',
                        connection: {
                            filename: "/home/keytop/data/lotconfig.db"
                        }
                    });
                    return [4 /*yield*/, knexInstance('t_config').where({ session: 'redis', name: 'password' })];
                case 1:
                    rows = _b.sent();
                    if (rows && rows.length > 0 && rows[0].value)
                        pwd = (_a = rows[0]) === null || _a === void 0 ? void 0 : _a.value;
                    return [2 /*return*/, pwd];
            }
        });
    });
}
function init() {
    return __awaiter(this, void 0, void 0, function () {
        var config, _a, _b, error_1, logFunc_1, error_2;
        return __generator(this, function (_c) {
            switch (_c.label) {
                case 0:
                    _c.trys.push([0, 7, , 8]);
                    _a = Config_1.IocConfigProvider.getIORedisConfig();
                    if (_a) return [3 /*break*/, 2];
                    return [4 /*yield*/, ConfigCenter_1.ConfigCenterConfigProvider.getRedisConfig()];
                case 1:
                    _a = (_c.sent());
                    _c.label = 2;
                case 2:
                    config = _a;
                    if (!config)
                        return [2 /*return*/];
                    if (!(Contants_1.IS_PARK && !config.password)) return [3 /*break*/, 6];
                    _c.label = 3;
                case 3:
                    _c.trys.push([3, 5, , 6]);
                    _b = config;
                    return [4 /*yield*/, getRedisPwdFromSqlite()];
                case 4:
                    _b.password = _c.sent();
                    return [3 /*break*/, 6];
                case 5:
                    error_1 = _c.sent();
                    mslib_1.logger.error('从sqlite获取redis密码失败', error_1);
                    return [3 /*break*/, 6];
                case 6:
                    logFunc_1 = Throttle_1.throttle1(function (error) { mslib_1.logger.error(error); });
                    //默认懒连接
                    exports.redisClient = new IORedis(Object.assign({ lazyConnect: true }, config));
                    exports.redisClient.on("error", function (error) {
                        if (['ETIMEDOUT', 'ECONNREFUSED'].includes(error.code)) {
                            logFunc_1(error);
                        }
                        else {
                            mslib_1.logger.error(error);
                        }
                    });
                    mslib_1.logger.info("redis\u5BA2\u6237\u7AEF\u5B9E\u4F8B\u521B\u5EFA\u6210\u529F,host:" + (config.host || 'localhost'));
                    //初始化i18n
                    Global_1.init();
                    return [3 /*break*/, 8];
                case 7:
                    error_2 = _c.sent();
                    mslib_1.logger.error('redis客户端实例创建失败', error_2);
                    return [3 /*break*/, 8];
                case 8: return [2 /*return*/];
            }
        });
    });
}
/**
 * redis分布式锁装饰器
 *
 * **注：如果加锁失败则不会执行被装饰的函数,而是执行lockFailedHandler**
 *
 * ```ts
 * import { RedisLocker } from "refs/nslib";
 *
 * //去掉注释
 * //@RedisLocker(args => { return `${args.parkId}.TASK`; }, args => { return '任务已在执行,请稍后再试'; })
 * async function exec(){
 *
 * }
 * ```
 * @param key redis锁的key
 * @param lockFailedHandler 加锁失败处理
 */
function RedisLocker(key, lockFailedHandler, opt) {
    return function (target, propertyKey, descriptor) {
        var method = descriptor.value;
        descriptor.value = function () {
            var args = [];
            for (var _i = 0; _i < arguments.length; _i++) {
                args[_i] = arguments[_i];
            }
            return __awaiter(this, void 0, void 0, function () {
                var _key, value, _timer, res, res_1;
                var _this = this;
                return __generator(this, function (_a) {
                    switch (_a.label) {
                        case 0:
                            _key = typeof key === 'function' ? key(args) : key;
                            value = Date_1.default.getUnixTimestamp().toString();
                            _timer = null;
                            res = null;
                            if (!(opt === null || opt === void 0 ? void 0 : opt.ex)) return [3 /*break*/, 2];
                            return [4 /*yield*/, exports.redisClient.set(_key, Utils_1.uuid(), 'NX', 'EX', opt.ex)];
                        case 1:
                            res = _a.sent();
                            return [3 /*break*/, 4];
                        case 2: return [4 /*yield*/, exports.redisClient.set(_key, value, 'NX')];
                        case 3:
                            res = _a.sent();
                            _a.label = 4;
                        case 4:
                            //加锁失败
                            if (res !== 'OK') {
                                mslib_1.logger.warn("redis \u52A0\u9501\u5931\u8D25,key:" + _key + ",value:" + value);
                                return [2 /*return*/, lockFailedHandler(args)];
                            }
                            else {
                                mslib_1.logger.info("redis \u52A0\u9501\u6210\u529F,key:" + _key);
                            }
                            //自动续约
                            if (opt === null || opt === void 0 ? void 0 : opt.autoRenew) {
                                _timer = setInterval(function () { return __awaiter(_this, void 0, void 0, function () {
                                    var res_2, error_3;
                                    return __generator(this, function (_a) {
                                        switch (_a.label) {
                                            case 0:
                                                _a.trys.push([0, 2, , 3]);
                                                return [4 /*yield*/, exports.redisClient.expire(_key, opt.ex)];
                                            case 1:
                                                res_2 = _a.sent();
                                                if (res_2 !== 1) {
                                                    mslib_1.logger.warn("redis \u9501\u81EA\u52A8\u7EED\u7EA6\u5931\u8D25,key:" + _key);
                                                    clearInterval(_timer);
                                                }
                                                console.log("redis \u9501\u81EA\u52A8\u7EED\u7EA6\u6210\u529F,key:" + _key);
                                                return [3 /*break*/, 3];
                                            case 2:
                                                error_3 = _a.sent();
                                                mslib_1.logger.error("redis \u9501\u81EA\u52A8\u7EED\u7EA6\u5931\u8D25,key:" + _key, error_3);
                                                return [3 /*break*/, 3];
                                            case 3: return [2 /*return*/];
                                        }
                                    });
                                }); }, opt.ex / 2 * 1000);
                            }
                            _a.label = 5;
                        case 5:
                            _a.trys.push([5, , 7, 9]);
                            return [4 /*yield*/, method.apply(this, args)];
                        case 6: return [2 /*return*/, _a.sent()];
                        case 7:
                            if (_timer)
                                clearInterval(_timer);
                            return [4 /*yield*/, exports.redisClient.del(_key)];
                        case 8:
                            res_1 = _a.sent();
                            mslib_1.logger.info("redis \u91CA\u653E\u9501,key:" + _key + ",res:" + res_1);
                            return [7 /*endfinally*/];
                        case 9: return [2 /*return*/];
                    }
                });
            });
        };
    };
}
exports.RedisLocker = RedisLocker;
//# sourceMappingURL=Redis.js.map