UNPKG

nestjs-express-cassandra

Version:
135 lines (134 loc) 6.12 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 ExpressCassandraCoreModule_1; Object.defineProperty(exports, "__esModule", { value: true }); exports.ExpressCassandraCoreModule = void 0; const common_1 = require("@nestjs/common"); const core_1 = require("@nestjs/core"); const nanoid_1 = require("nanoid"); const constants_1 = require("./constants"); const express_cassandra_module_1 = require("./express-cassandra.module"); const connection_1 = require("./utils/connection"); const providers_1 = require("./utils/providers"); let ExpressCassandraCoreModule = ExpressCassandraCoreModule_1 = class ExpressCassandraCoreModule { constructor(options, moduleRef) { this.options = options; this.moduleRef = moduleRef; this.logger = new common_1.Logger(express_cassandra_module_1.ExpressCassandraModule.name); } onModuleDestroy() { return __awaiter(this, void 0, void 0, function* () { if (this.options.keepConnectionAlive) { return; } const connection = this.moduleRef.get((0, providers_1.getConnectionProviderName)(this.options)); if (connection) { yield connection.closeAsync(); this.logger.log(`[${connection.name}] Connection closed`); } }); } static forRoot(options) { const moduleOptions = { provide: constants_1.EXPRESS_CASSANDRA_MODULE_OPTIONS, useValue: options, }; const connectionProvider = { provide: (0, providers_1.getConnectionProviderName)(options), useFactory: () => __awaiter(this, void 0, void 0, function* () { return yield (0, connection_1.createNewConnection)(options); }), }; return { module: ExpressCassandraCoreModule_1, providers: [connectionProvider, moduleOptions], exports: [connectionProvider], }; } static forRootAsync(options) { const connectionProvider = { provide: (0, providers_1.getConnectionProviderName)(options), useFactory: (typeormOptions) => __awaiter(this, void 0, void 0, function* () { return yield (0, connection_1.createNewConnection)(typeormOptions); }), inject: [constants_1.EXPRESS_CASSANDRA_MODULE_OPTIONS], }; const asyncProviders = this.createAsyncProviders(options); return { module: ExpressCassandraCoreModule_1, imports: options.imports, providers: [ connectionProvider, ...asyncProviders, { provide: constants_1.EXPRESS_CASSANDRA_MODULE_ID, useValue: (0, nanoid_1.nanoid)(), }, ], exports: [connectionProvider], }; } static createAsyncProviders(options) { if (options.useExisting || options.useFactory) { return [this.createAsyncOptionsProvider(options)]; } const useClass = options.useClass; return useClass ? [ this.createAsyncOptionsProvider(options), { provide: useClass, useClass, }, ] : [this.createAsyncOptionsProvider(options)]; } static createAsyncOptionsProvider(options) { if (options.useFactory) { return { provide: constants_1.EXPRESS_CASSANDRA_MODULE_OPTIONS, useFactory: options.useFactory, inject: options.inject || [], }; } const inject = options.useClass ? [options.useClass] : options.useExisting ? [options.useExisting] : []; return { provide: constants_1.EXPRESS_CASSANDRA_MODULE_OPTIONS, useFactory: (optionsFactory) => __awaiter(this, void 0, void 0, function* () { return optionsFactory ? yield optionsFactory.createExpressCassandraOptions(options.name) : null; }), inject, }; } }; ExpressCassandraCoreModule = ExpressCassandraCoreModule_1 = __decorate([ (0, common_1.Global)(), (0, common_1.Module)({}), __param(0, (0, common_1.Inject)(constants_1.EXPRESS_CASSANDRA_MODULE_OPTIONS)), __metadata("design:paramtypes", [Object, core_1.ModuleRef]) ], ExpressCassandraCoreModule); exports.ExpressCassandraCoreModule = ExpressCassandraCoreModule;