misskey/packages/backend/src/GlobalModule.ts

67 lines
1.7 KiB
TypeScript
Raw Normal View History

2022-09-17 20:27:08 +02:00
import { Global, Inject, Module } from '@nestjs/common';
2022-09-20 21:54:21 +02:00
import Redis from 'ioredis';
2022-09-17 20:27:08 +02:00
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 './models/RepositoryModule.js';
2022-09-17 20:27:08 +02:00
import type { Provider, OnApplicationShutdown } from '@nestjs/common';
const config = loadConfig();
const $config: Provider = {
provide: DI.config,
useValue: config,
};
const $db: Provider = {
provide: DI.db,
2022-09-21 22:11:26 +02:00
useFactory: async (config) => {
const db = createPostgreDataSource(config);
2022-09-17 20:27:08 +02:00
return await db.initialize();
},
2022-09-21 22:11:26 +02:00
inject: [DI.config],
2022-09-17 20:27:08 +02:00
};
const $redis: Provider = {
provide: DI.redis,
2022-09-21 22:11:26 +02:00
useFactory: (config) => {
const redisClient = createRedisConnection(config);
2022-09-17 20:27:08 +02:00
return redisClient;
},
2022-09-21 22:11:26 +02:00
inject: [DI.config],
2022-09-17 20:27:08 +02:00
};
const $redisSubscriber: Provider = {
provide: DI.redisSubscriber,
2022-09-21 22:11:26 +02:00
useFactory: (config) => {
const redisSubscriber = createRedisConnection(config);
2022-09-17 20:27:08 +02:00
redisSubscriber.subscribe(config.host);
return redisSubscriber;
},
2022-09-21 22:11:26 +02:00
inject: [DI.config],
2022-09-17 20:27:08 +02:00
};
@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,
2022-09-20 21:54:21 +02:00
@Inject(DI.redis) private redisClient: Redis.Redis,
@Inject(DI.redisSubscriber) private redisSubscriber: Redis.Redis,
2022-09-17 20:27:08 +02:00
) {}
async onApplicationShutdown(signal: string): Promise<void> {
await Promise.all([
this.db.destroy(),
this.redisClient.disconnect(),
this.redisSubscriber.disconnect(),
]);
}
}