Вставка одной записи в Nest
Приложение уже подключилось к базе, но нужная таблица не содержит ни одной строки. Ждать первого запроса и тогда вручную заводить опорную запись неудобно для локальной разработки.
Сервис темы при инициализации модуля спрашивает у репозитория, сколько строк уже лежит в таблице. Число ноль означает, что можно сохранить один объект с заготовленным заголовком. Повторный старт приложения не дублирует строку, потому что счётчик уже больше нуля.
Покажем на примере постов:
import { Column, Entity, PrimaryGeneratedColumn } from 'typeorm';
@Entity()
export class Post {
@PrimaryGeneratedColumn()
id: number;
@Column()
title: string;
}
Сервис проверит таблицу при старте и сохранит одну запись, если она пуста:
import { Injectable, OnModuleInit } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm';
import { Post } from '../post.entity.js';
@Injectable()
export class PostService implements OnModuleInit {
postRepository: Repository<Post>;
constructor(
@InjectRepository(Post) postRepository: Repository<Post>,
) {
this.postRepository = postRepository;
}
async onModuleInit() {
const total = await this.postRepository.count();
if (total === 0) {
await this.postRepository.save({
title: 'Starter note',
});
}
}
findAll() {
return this.postRepository.find();
}
}
Контроллер откроет список для проверки после старта:
import { Controller, Get } from '@nestjs/common';
import { PostService } from './post.service.js';
@Controller('posts')
export class PostController {
postService: PostService;
constructor(postService: PostService) {
this.postService = postService;
}
@Get()
findAll() {
return this.postService.findAll();
}
}
Модуль темы регистрирует сущность и классы страницы:
import { Module } from '@nestjs/common';
import { TypeOrmModule } from '@nestjs/typeorm';
import { Post } from '../post.entity.js';
import { PostController } from './post.controller.js';
import { PostService } from './post.service.js';
@Module({
imports: [TypeOrmModule.forFeature([Post])],
controllers: [PostController],
providers: [PostService],
})
export class PostsModule {}
Корень поднимает канал к базе и импортирует модуль темы:
import { Module } from '@nestjs/common';
import { TypeOrmModule } from '@nestjs/typeorm';
import { PostsModule } from './posts/posts.module.js';
@Module({
imports: [
TypeOrmModule.forRoot({
type: 'postgres',
host: 'localhost',
port: 5432,
username: 'postgres',
password: 'postgres',
database: 'site',
autoLoadEntities: true,
synchronize: true,
}),
PostsModule,
],
})
export class AppModule {}
При старте модуля добавьте
в WorkerService одну строку
работника, если таблица пуста.
Список откройте в
WorkerController и
соберите WorkersModule
с базой в корне.