UNPKG

2.85 kBJavaScriptView Raw
1"use strict";
2var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3 function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4 return new (P || (P = Promise))(function (resolve, reject) {
5 function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6 function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7 function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8 step((generator = generator.apply(thisArg, _arguments || [])).next());
9 });
10};
11Object.defineProperty(exports, "__esModule", { value: true });
12exports.createMongooseAsyncProviders = exports.createMongooseProviders = void 0;
13const mongoose_utils_1 = require("./common/mongoose.utils");
14function createMongooseProviders(connectionName, options = []) {
15 return options.reduce((providers, option) => [
16 ...providers,
17 ...(option.discriminators || []).map((d) => ({
18 provide: (0, mongoose_utils_1.getModelToken)(d.name),
19 useFactory: (model) => model.discriminator(d.name, d.schema),
20 inject: [(0, mongoose_utils_1.getModelToken)(option.name)],
21 })),
22 {
23 provide: (0, mongoose_utils_1.getModelToken)(option.name),
24 useFactory: (connection) => {
25 const model = connection.model(option.name, option.schema, option.collection);
26 return model;
27 },
28 inject: [(0, mongoose_utils_1.getConnectionToken)(connectionName)],
29 },
30 ], []);
31}
32exports.createMongooseProviders = createMongooseProviders;
33function createMongooseAsyncProviders(connectionName, modelFactories = []) {
34 return modelFactories.reduce((providers, option) => {
35 return [
36 ...providers,
37 {
38 provide: (0, mongoose_utils_1.getModelToken)(option.name),
39 useFactory: (connection, ...args) => __awaiter(this, void 0, void 0, function* () {
40 const schema = yield option.useFactory(...args);
41 const model = connection.model(option.name, schema, option.collection);
42 return model;
43 }),
44 inject: [(0, mongoose_utils_1.getConnectionToken)(connectionName), ...(option.inject || [])],
45 },
46 ...(option.discriminators || []).map((d) => ({
47 provide: (0, mongoose_utils_1.getModelToken)(d.name),
48 useFactory: (model) => model.discriminator(d.name, d.schema),
49 inject: [(0, mongoose_utils_1.getModelToken)(option.name)],
50 })),
51 ];
52 }, []);
53}
54exports.createMongooseAsyncProviders = createMongooseAsyncProviders;