Files
bidding_watcher/src/database/database.module.ts

32 lines
1.2 KiB
TypeScript
Raw Normal View History

2026-01-09 23:18:52 +08:00
import { Module } from '@nestjs/common';
import { TypeOrmModule } from '@nestjs/typeorm';
import { ConfigModule, ConfigService } from '@nestjs/config';
import { BidItem } from '../bids/entities/bid-item.entity';
import { Keyword } from '../keywords/keyword.entity';
import { AiRecommendation } from '../ai/entities/ai-recommendation.entity';
import { CrawlInfoAdd } from '../crawler/entities/crawl-info-add.entity';
2026-01-09 23:18:52 +08:00
@Module({
imports: [
TypeOrmModule.forRootAsync({
imports: [ConfigModule],
inject: [ConfigService],
useFactory: (configService: ConfigService) => ({
type:
(configService.get<string>('DATABASE_TYPE', 'mariadb') as
| 'mariadb'
| 'mysql'
| 'postgres') || 'mariadb',
2026-01-09 23:18:52 +08:00
host: configService.get<string>('DATABASE_HOST', 'localhost'),
port: configService.get<number>('DATABASE_PORT', 3306),
username: configService.get<string>('DATABASE_USERNAME', 'root'),
password: configService.get<string>('DATABASE_PASSWORD', 'root'),
database: configService.get<string>('DATABASE_NAME', 'bidding'),
entities: [BidItem, Keyword, AiRecommendation, CrawlInfoAdd],
synchronize: false,
2026-01-09 23:18:52 +08:00
}),
}),
],
})
export class DatabaseModule {}