mirror of https://github.com/fantasticit/think.git
124 lines
3.6 KiB
TypeScript
124 lines
3.6 KiB
TypeScript
import { forwardRef, Inject, Module } from '@nestjs/common';
|
||
import { ConfigModule, ConfigService } from '@nestjs/config';
|
||
import { Cron, ScheduleModule } from '@nestjs/schedule';
|
||
import { TypeOrmModule, TypeOrmModuleOptions } from '@nestjs/typeorm';
|
||
import { LoggerModule } from 'nestjs-pino';
|
||
|
||
import { getConfig } from '@think/config';
|
||
|
||
import { AuthEntity } from '@entities/auth.entity';
|
||
import { CommentEntity } from '@entities/comment.entity';
|
||
import { DocumentEntity } from '@entities/document.entity';
|
||
import { MessageEntity } from '@entities/message.entity';
|
||
import { OrganizationEntity } from '@entities/organization.entity';
|
||
import { StarEntity } from '@entities/star.entity';
|
||
import { SystemEntity } from '@entities/system.entity';
|
||
import { TemplateEntity } from '@entities/template.entity';
|
||
import { UserEntity } from '@entities/user.entity';
|
||
import { WikiEntity } from '@entities/wiki.entity';
|
||
import { IS_PRODUCTION } from '@helpers/env.helper';
|
||
import { getLogFileName, ONE_DAY } from '@helpers/log.helper';
|
||
import { AuthModule } from '@modules/auth.module';
|
||
import { CommentModule } from '@modules/comment.module';
|
||
import { DocumentModule } from '@modules/document.module';
|
||
import { FileModule } from '@modules/file.module';
|
||
import { MessageModule } from '@modules/message.module';
|
||
import { OrganizationModule } from '@modules/organization.module';
|
||
import { StarModule } from '@modules/star.module';
|
||
import { SystemModule } from '@modules/system.module';
|
||
import { TemplateModule } from '@modules/template.module';
|
||
import { UserModule } from '@modules/user.module';
|
||
import { VerifyModule } from '@modules/verify.module';
|
||
import { ViewModule } from '@modules/view.module';
|
||
import { WikiModule } from '@modules/wiki.module';
|
||
import * as fs from 'fs-extra';
|
||
import * as path from 'path';
|
||
import pino from 'pino';
|
||
|
||
const ENTITIES = [
|
||
UserEntity,
|
||
AuthEntity,
|
||
OrganizationEntity,
|
||
WikiEntity,
|
||
DocumentEntity,
|
||
StarEntity,
|
||
CommentEntity,
|
||
MessageEntity,
|
||
TemplateEntity,
|
||
SystemEntity,
|
||
];
|
||
|
||
const MODULES = [
|
||
UserModule,
|
||
AuthModule,
|
||
OrganizationModule,
|
||
WikiModule,
|
||
DocumentModule,
|
||
StarModule,
|
||
FileModule,
|
||
CommentModule,
|
||
MessageModule,
|
||
TemplateModule,
|
||
ViewModule,
|
||
VerifyModule,
|
||
SystemModule,
|
||
];
|
||
|
||
@Module({
|
||
imports: [
|
||
ConfigModule.forRoot({
|
||
cache: true,
|
||
load: [getConfig],
|
||
isGlobal: true,
|
||
}),
|
||
ScheduleModule.forRoot(),
|
||
IS_PRODUCTION &&
|
||
LoggerModule.forRoot({
|
||
pinoHttp: {
|
||
stream: pino.destination({
|
||
dest: `./logs/${getLogFileName(new Date())}`,
|
||
minLength: 4096,
|
||
mkdir: true,
|
||
sync: false,
|
||
}),
|
||
},
|
||
}),
|
||
TypeOrmModule.forRootAsync({
|
||
imports: [ConfigModule],
|
||
inject: [ConfigService],
|
||
useFactory: (config: ConfigService) => {
|
||
return {
|
||
type: 'mysql',
|
||
entities: ENTITIES,
|
||
keepConnectionAlive: true,
|
||
...config.get('db.mysql'),
|
||
} as TypeOrmModuleOptions;
|
||
},
|
||
}),
|
||
...MODULES,
|
||
].filter(Boolean),
|
||
controllers: [],
|
||
providers: [],
|
||
})
|
||
export class AppModule {
|
||
constructor(
|
||
@Inject(forwardRef(() => ConfigService))
|
||
private readonly configService: ConfigService
|
||
) {}
|
||
|
||
/**
|
||
* 每天早上9点，清理日志
|
||
*/
|
||
@Cron('0 0 9 * * *')
|
||
deleteLog() {
|
||
let retainDays = this.configService.get('server.logRetainDays');
|
||
const startDate = new Date(new Date().valueOf() - retainDays * ONE_DAY).valueOf();
|
||
|
||
do {
|
||
const filepath = path.join(__dirname, '../logs', getLogFileName(startDate, retainDays));
|
||
fs.removeSync(filepath);
|
||
retainDays -= 1;
|
||
} while (retainDays > 0);
|
||
}
|
||
}
|