@nestjs/mongoose
Version:
Nest - modern, fast, powerful node.js web framework (@mongoose)
31 lines (30 loc) • 1.33 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.handleRetry = exports.getConnectionToken = exports.getModelToken = void 0;
const common_1 = require("@nestjs/common");
const operators_1 = require("rxjs/operators");
const mongoose_constants_1 = require("../mongoose.constants");
function getModelToken(model, connectionName) {
if (connectionName === undefined) {
return `${model}Model`;
}
return `${getConnectionToken(connectionName)}/${model}Model`;
}
exports.getModelToken = getModelToken;
function getConnectionToken(name) {
return name && name !== mongoose_constants_1.DEFAULT_DB_CONNECTION
? `${name}Connection`
: mongoose_constants_1.DEFAULT_DB_CONNECTION;
}
exports.getConnectionToken = getConnectionToken;
function handleRetry(retryAttempts = 9, retryDelay = 3000) {
const logger = new common_1.Logger('MongooseModule');
return (source) => source.pipe((0, operators_1.retryWhen)((e) => e.pipe((0, operators_1.scan)((errorCount, error) => {
logger.error(`Unable to connect to the database. Retrying (${errorCount + 1})...`, '');
if (errorCount + 1 >= retryAttempts) {
throw error;
}
return errorCount + 1;
}, 0), (0, operators_1.delay)(retryDelay))));
}
exports.handleRetry = handleRetry;