UNPKG

@atmus/nestjs-cache

Version:
90 lines (89 loc) 4.22 kB
"use strict"; var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) { var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d; if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc); else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r; return c > 3 && r && Object.defineProperty(target, key, r), r; }; var __metadata = (this && this.__metadata) || function (k, v) { if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v); }; var __param = (this && this.__param) || function (paramIndex, decorator) { return function (target, key) { decorator(target, key, paramIndex); } }; 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 CacheService_1; Object.defineProperty(exports, "__esModule", { value: true }); exports.CacheService = void 0; const common_1 = require("@nestjs/common"); const crypto = require("crypto"); const ioredis_1 = require("ioredis"); const util = require("util"); const constants_1 = require("./constants"); let CacheService = CacheService_1 = class CacheService { constructor(config) { this.logger = new common_1.Logger(CacheService_1.name); this.redis = new ioredis_1.Redis(config); this.redis.on('connect', () => this.logger.debug('Redis is connected')); this.redis.on('error', this.logger.error); } set(key, value, option, optionValue) { const setPromisefy = util.promisify(this.redis.set).bind(this.redis); if (option !== undefined && optionValue !== undefined) { return setPromisefy(key, value, option, optionValue); } return setPromisefy(key, value); } setFromParams(key, params, value, option, optionValue) { const finalKey = `${key}:${this.encryptParams(params)}`; if (option !== undefined && optionValue !== undefined) { return this.set(finalKey, value, option, optionValue); } return this.set(finalKey, value); } get(key) { const getPromisefy = util.promisify(this.redis.get).bind(this.redis); return getPromisefy(key); } getFromParams(key, params) { const finalKey = `${key}:${this.encryptParams(params)}`; return this.get(finalKey); } getKeys(pattern) { const getKeysPromisefy = util.promisify(this.redis.keys).bind(this.redis); return getKeysPromisefy(pattern); } del(key) { const delPromisefy = util.promisify(this.redis.del).bind(this.redis); return delPromisefy(key); } delFromPattern(pattern) { return __awaiter(this, void 0, void 0, function* () { const all = yield this.getKeys(pattern); for (const item of all) this.del(item); }); } delFromParams(key, params) { const finalKey = `${key}:${this.encryptParams(params)}`; return this.del(finalKey); } encryptParams(params) { const str = JSON.stringify(params); return crypto.createHash('sha256').update(str).digest('base64'); } }; exports.CacheService = CacheService; exports.CacheService = CacheService = CacheService_1 = __decorate([ (0, common_1.Injectable)(), __param(0, (0, common_1.Inject)(constants_1.CACHE_CONFIG_OPTIONS)), __metadata("design:paramtypes", [Object]) ], CacheService);