41 lines
1.1 KiB
TypeScript
41 lines
1.1 KiB
TypeScript
|
import type { Repository } from 'typeorm';
|
|||
|
|
|
|||
|
|
import { UserEntity } from '@/models/entity/user.entity';
|
|||
|
|
import { Injectable, type OnModuleInit } from '@nestjs/common';
|
|||
|
|
import { InjectRepository } from '@nestjs/typeorm';
|
|||
|
|
|
|||
|
|
import { UsersService } from '../users/users.service';
|
|||
|
|
|
|||
|
|
@Injectable()
|
|||
|
|
export class DatabaseService implements OnModuleInit {
|
|||
|
|
constructor(
|
|||
|
|
@InjectRepository(UserEntity)
|
|||
|
|
private usersRepository: Repository<UserEntity>,
|
|||
|
|
private userService: UsersService,
|
|||
|
|
) {}
|
|||
|
|
async onModuleInit() {
|
|||
|
|
// data/db.sqlite会被git忽略，方式数据库文件被提交到git
|
|||
|
|
// 清空表，并初始化两条数据
|
|||
|
|
await this.usersRepository.clear();
|
|||
|
|
|
|||
|
|
await this.userService.create({
|
|||
|
|
id: 0,
|
|||
|
|
password: '123456',
|
|||
|
|
realName: 'Administrator',
|
|||
|
|
roles: ['admin'],
|
|||
|
|
username: 'vben',
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
await this.userService.create({
|
|||
|
|
id: 1,
|
|||
|
|
password: '123456',
|
|||
|
|
realName: 'Jack',
|
|||
|
|
roles: ['user'],
|
|||
|
|
username: 'jack',
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
const count = await this.usersRepository.count();
|
|||
|
|
console.log('Database has been initialized with seed data, count:', count);
|
|||
|
|
}
|
|||
|
|
}
|