99AI/dist/modules/midjourney/midjourney.module.js
2024-07-07 13:09:08 +08:00

48 lines
2.3 KiB
JavaScript

"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;
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.MidjourneyModule = void 0;
const bull_1 = require("@nestjs/bull");
const common_1 = require("@nestjs/common");
const typeorm_1 = require("@nestjs/typeorm");
const redisCache_service_1 = require("../redisCache/redisCache.service");
const user_entity_1 = require("../user/user.entity");
const midjourney_controller_1 = require("./midjourney.controller");
const midjourney_entity_1 = require("./midjourney.entity");
const midjourney_service_1 = require("./midjourney.service");
const prompt_entity_1 = require("./prompt.entity");
let MidjourneyModule = class MidjourneyModule {
};
MidjourneyModule = __decorate([
(0, common_1.Global)(),
(0, common_1.Module)({
imports: [
bull_1.BullModule.registerQueueAsync({
name: 'MJDRAW',
useFactory: () => {
const config = {
port: +process.env.REDIS_PORT,
host: process.env.REDIS_HOST,
db: Number(process.env.REDIS_DB || 0),
};
process.env.REDIS_PASSWORD &&
(config.password = process.env.REDIS_PASSWORD);
return {
redis: config,
};
},
}),
typeorm_1.TypeOrmModule.forFeature([midjourney_entity_1.MidjourneyEntity, user_entity_1.UserEntity, prompt_entity_1.mjPromptEntity]),
],
controllers: [midjourney_controller_1.MidjourneyController],
providers: [midjourney_service_1.MidjourneyService, redisCache_service_1.RedisCacheService],
exports: [midjourney_service_1.MidjourneyService],
})
], MidjourneyModule);
exports.MidjourneyModule = MidjourneyModule;