import { Global, Inject, Module } from '@nestjs/common'; import Redis from 'ioredis'; import { DataSource } from 'typeorm'; import { createRedisConnection } from '@/redis.js'; import { DI } from './di-symbols.js'; import { loadConfig } from './config.js'; import { createPostgreDataSource } from './postgre.js'; import { RepositoryModule } from './RepositoryModule.js'; import type { Provider, OnApplicationShutdown } from '@nestjs/common'; const config = loadConfig(); const $config: Provider = { provide: DI.config, useValue: config, }; const $db: Provider = { provide: DI.db, useFactory: async (config) => { const db = createPostgreDataSource(config); return await db.initialize(); }, inject: [DI.config], }; const $redis: Provider = { provide: DI.redis, useFactory: (config) => { const redisClient = createRedisConnection(config); return redisClient; }, inject: [DI.config], }; const $redisSubscriber: Provider = { provide: DI.redisSubscriber, useFactory: (config) => { const redisSubscriber = createRedisConnection(config); redisSubscriber.subscribe(config.host); return redisSubscriber; }, inject: [DI.config], }; @Global() @Module({ imports: [RepositoryModule], providers: [$config, $db, $redis, $redisSubscriber], exports: [$config, $db, $redis, $redisSubscriber, RepositoryModule], }) export class GlobalModule implements OnApplicationShutdown { constructor( @Inject(DI.db) private db: DataSource, @Inject(DI.redis) private redisClient: Redis.Redis, @Inject(DI.redisSubscriber) private redisSubscriber: Redis.Redis, ) {} async onApplicationShutdown(signal: string): Promise { await Promise.all([ this.db.destroy(), this.redisClient.disconnect(), this.redisSubscriber.disconnect(), ]); } }