backend/libs/common/src/database/database.module.ts

import { Module } from '@nestjs/common';
import { ConfigModule, ConfigService } from '@nestjs/config';
import { TypeOrmModule } from '@nestjs/typeorm';
import { SnakeNamingStrategy } from './strategies';
import { UserEntity } from '../modules/user/entities/user.entity';
import { UserSessionEntity } from '../modules/session/entities/session.entity';
import { UserOtpEntity } from '../modules/user/entities';
import { ProductEntity } from '../modules/product/entities';
import { DeviceEntity } from '../modules/device/entities';
import { PermissionTypeEntity } from '../modules/permission/entities';
import {
  SpaceEntity,
  SpaceLinkEntity,
  SubspaceEntity,
} from '../modules/space/entities';
import { UserSpaceEntity } from '../modules/user/entities';
import { DeviceUserPermissionEntity } from '../modules/device/entities';
import { RoleTypeEntity } from '../modules/role-type/entities';
import { UserNotificationEntity } from '../modules/user/entities';
import { DeviceNotificationEntity } from '../modules/device/entities';
import { RegionEntity } from '../modules/region/entities';
import { TimeZoneEntity } from '../modules/timezone/entities';
import { VisitorPasswordEntity } from '../modules/visitor-password/entities';
import { CommunityEntity } from '../modules/community/entities';
import { DeviceStatusLogEntity } from '../modules/device-status-log/entities';
import { SceneEntity, SceneIconEntity } from '../modules/scene/entities';
import { SceneDeviceEntity } from '../modules/scene-device/entities';
import { ProjectEntity } from '../modules/project/entities';
import {
  SpaceModelEntity,
  SubspaceModelEntity,
  TagModel,
} from '../modules/space-model/entities';
import {
  InviteUserEntity,
  InviteUserSpaceEntity,
} from '../modules/Invite-user/entities';
@Module({
  imports: [
    TypeOrmModule.forRootAsync({
      imports: [ConfigModule],
      inject: [ConfigService],
      useFactory: (configService: ConfigService) => ({
        name: 'default',
        type: 'postgres',
        host: configService.get('DB_HOST'),
        port: configService.get('DB_PORT'),
        username: configService.get('DB_USER'),
        password: configService.get('DB_PASSWORD'),
        database: configService.get('DB_NAME'),
        entities: [
          ProjectEntity,
          UserEntity,
          UserSessionEntity,
          UserOtpEntity,
          ProductEntity,
          DeviceUserPermissionEntity,
          DeviceEntity,
          PermissionTypeEntity,
          CommunityEntity,
          SpaceEntity,
          SpaceLinkEntity,
          SubspaceEntity,
          UserSpaceEntity,
          DeviceUserPermissionEntity,
          RoleTypeEntity,
          UserNotificationEntity,
          DeviceNotificationEntity,
          RegionEntity,
          TimeZoneEntity,
          VisitorPasswordEntity,
          DeviceStatusLogEntity,
          SceneEntity,
          SceneIconEntity,
          SceneDeviceEntity,
          SpaceModelEntity,
          SubspaceModelEntity,
          TagModel,
          InviteUserEntity,
          InviteUserSpaceEntity,
        ],
        namingStrategy: new SnakeNamingStrategy(),
        synchronize: Boolean(JSON.parse(configService.get('DB_SYNC'))),
        logging: false,
        extra: {
          charset: 'utf8mb4',
          max: 20, // set pool max size
          idleTimeoutMillis: 5000, // close idle clients after 5 second
          connectionTimeoutMillis: 11_000, // return an error after 11 second if connection could not be established
          maxUses: 7500, // close (and replace) a connection after it has been used 7500 times (see below for discussion)
        },
        continuationLocalStorage: true,
        ssl: Boolean(JSON.parse(configService.get('DB_SSL'))),
      }),
    }),
  ],
})
export class DatabaseModule {}