@cyber-rom/nestjs-scylladb
Version:
Based on https://www.npmjs.com/package/@ouato/nestjs-express-cassandra With support columns name mapping
145 lines • 6.76 kB
JavaScript
;
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 __rest = (this && this.__rest) || function (s, e) {
var t = {};
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
t[p] = s[p];
if (s != null && typeof Object.getOwnPropertySymbols === "function")
for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
t[p[i]] = s[p[i]];
}
return t;
};
var ScyllaCoreModule_1;
Object.defineProperty(exports, "__esModule", { value: true });
exports.ScyllaCoreModule = exports.SCYLLA_MODULE_ID = exports.SCYLLA_MODULE_OPTIONS = void 0;
const operators_1 = require("rxjs/operators");
const core_1 = require("@nestjs/core");
const rxjs_1 = require("rxjs");
const common_1 = require("@nestjs/common");
const orm_utils_1 = require("./utils/orm.utils");
const orm_1 = require("./orm");
exports.SCYLLA_MODULE_OPTIONS = 'ScyllaModuleOptions';
exports.SCYLLA_MODULE_ID = 'ScyllaModuleId';
let ScyllaCoreModule = ScyllaCoreModule_1 = class ScyllaCoreModule {
constructor(options, moduleRef) {
this.options = options;
this.moduleRef = moduleRef;
}
static forRoot(options = {}) {
const expressModuleOptions = {
provide: exports.SCYLLA_MODULE_OPTIONS,
useValue: options,
};
const connectionProvider = {
provide: (0, orm_utils_1.getConnectionToken)(options),
useFactory: () => __awaiter(this, void 0, void 0, function* () { return yield this.createConnectionFactory(options); }),
};
return {
providers: [expressModuleOptions, connectionProvider],
exports: [connectionProvider],
module: ScyllaCoreModule_1,
};
}
static forRootAsync(options) {
const connectionProvider = {
provide: (0, orm_utils_1.getConnectionToken)(options),
useFactory: (typeormOptions) => __awaiter(this, void 0, void 0, function* () {
if (options.name) {
return yield this.createConnectionFactory(Object.assign(Object.assign({}, typeormOptions), { name: options.name }));
}
return yield this.createConnectionFactory(typeormOptions);
}),
inject: [exports.SCYLLA_MODULE_OPTIONS],
};
const asyncProviders = this.createAsyncProviders(options);
return {
module: ScyllaCoreModule_1,
imports: options.imports,
providers: [
...asyncProviders,
connectionProvider,
{
provide: exports.SCYLLA_MODULE_ID,
useValue: (0, orm_utils_1.generateString)(),
},
],
exports: [connectionProvider],
};
}
onModuleDestroy() {
return __awaiter(this, void 0, void 0, function* () {
if (this.options.keepConnectionAlive) {
return;
}
common_1.Logger.log('Closing connection', 'ScyllaModule');
const connection = this.moduleRef.get((0, orm_utils_1.getConnectionToken)(this.options));
if (!connection) {
return;
}
yield connection.closeAsync();
});
}
static createAsyncProviders(options) {
if (options.useExisting || options.useFactory) {
return [this.createAsyncOptionsProvider(options)];
}
return [
this.createAsyncOptionsProvider(options),
{
provide: options.useClass,
useClass: options.useClass,
},
];
}
static createAsyncOptionsProvider(options) {
if (options.useFactory) {
return {
useFactory: options.useFactory,
provide: exports.SCYLLA_MODULE_OPTIONS,
inject: options.inject || [],
};
}
return {
provide: exports.SCYLLA_MODULE_OPTIONS,
useFactory: (optionsFactory) => __awaiter(this, void 0, void 0, function* () { return yield optionsFactory.createScyllaOptions(); }),
inject: [options.useClass || options.useExisting],
};
}
static createConnectionFactory(_a) {
var { retryDelay, retryAttempts } = _a, cassandraOptions = __rest(_a, ["retryDelay", "retryAttempts"]);
return __awaiter(this, void 0, void 0, function* () {
const connection = new orm_1.Connection(cassandraOptions);
return yield (0, rxjs_1.lastValueFrom)((0, rxjs_1.defer)(() => connection.initAsync()).pipe((0, orm_utils_1.handleRetry)(retryAttempts, retryDelay), (0, operators_1.map)(() => connection)));
});
}
};
ScyllaCoreModule = ScyllaCoreModule_1 = __decorate([
(0, common_1.Global)(),
(0, common_1.Module)({}),
__param(0, (0, common_1.Inject)(exports.SCYLLA_MODULE_OPTIONS)),
__metadata("design:paramtypes", [Object, core_1.ModuleRef])
], ScyllaCoreModule);
exports.ScyllaCoreModule = ScyllaCoreModule;
//# sourceMappingURL=scylla-core.module.js.map