如何在NestJS启动时默认向MongoDB添加管理员用户？

方案一：利用NestJS生命周期钩子（OnModuleInit） #

这是最贴合你原有Express逻辑的方式，在模块初始化完成后自动执行检查和创建操作。

1. 创建一个初始化服务，实现OnModuleInit接口：

import { Injectable, OnModuleInit } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm';
import { User } from './user.entity'; // 你的User实体类
import { Roles } from './roles.enum'; // 你的角色枚举
import { PasswordService } from './password.service'; // 你的密码哈希服务

@Injectable()
export class InitService implements OnModuleInit {
  constructor(
    @InjectRepository(User)
    private readonly userRepository: Repository<User>,
    private readonly passwordService: PasswordService,
  ) {}

  async onModuleInit() {
    await this.addDefaultAdmin();
  }

  private async addDefaultAdmin() {
    const existingAdmin = await this.userRepository.findOne({
      where: { role: Roles.admin },
    });

    if (!existingAdmin) {
      try {
        const defaultAdmin = this.userRepository.create({
          // 填写默认管理员的信息
          username: 'admin',
          email: 'admin@example.com',
          role: Roles.admin,
        });
        // 哈希密码
        defaultAdmin.password = await this.passwordService.hashPassword('your-default-password');
        await this.userRepository.save(defaultAdmin);
        console.log('默认管理员已成功创建');
      } catch (error) {
        console.error('创建默认管理员失败:', error);
      }
    }
  }
}

2. 将这个服务添加到根模块（比如AppModule）的providers数组中，确保Nest启动时能加载执行。

方案二：使用TypeORM种子数据（Seeding） #

如果需要更灵活的初始化数据管理（比如支持多环境、可手动触发执行），可以用TypeORM的种子机制。

1. 安装依赖：

npm install typeorm-extension --save-dev

2. 创建种子类：

import { DataSource } from 'typeorm';
import { Seeder } from 'typeorm-extension';
import { User } from './user.entity';
import { Roles } from './roles.enum';
import { PasswordService } from './password.service';

export class AdminSeeder implements Seeder {
  async run(dataSource: DataSource): Promise<void> {
    const userRepository = dataSource.getRepository(User);
    const passwordService = new PasswordService(); // 或根据项目结构通过DI注入

    const existingAdmin = await userRepository.findOne({
      where: { role: Roles.admin },
    });

    if (!existingAdmin) {
      const defaultAdmin = userRepository.create({
        username: 'admin',
        email: 'admin@example.com',
        role: Roles.admin,
        password: await passwordService.hashPassword('your-default-password'),
      });
      await userRepository.save(defaultAdmin);
      console.log('默认管理员种子数据已插入');
    }
  }
}

3. 在TypeORM配置文件中添加种子配置：

import { DataSource } from 'typeorm';
import { AdminSeeder } from './seeders/admin.seeder';

export const dataSource = new DataSource({
  // 你的数据库连接配置
  type: 'mysql',
  host: 'localhost',
  port: 3306,
  username: 'root',
  password: 'password',
  database: 'your-db',
  entities: [User],
  seeds: [AdminSeeder],
});

4. 执行种子命令：

npx typeorm-extension seed:run

注意事项 #

• 密码哈希逻辑复用项目中已有的加密方法，避免重复实现。
• 生产环境不要硬编码默认密码，建议通过环境变量（如process.env.DEFAULT_ADMIN_PASSWORD）获取。
• 使用生命周期钩子时，确保服务仅在启动时执行一次，避免重复创建管理员。

内容的提问来源于stack exchange，提问作者Abolfazl Almas