Skip to main content

SQL (TypeORM)

本章仅适用于 TypeScript。

warning

在本文中,您将学习如何使用自定义提供程序机制从头开始创建基于 TypeORM 包的 DatabaseModule。 因此,此解决方案包含许多可以使用现成的、可立即使用的专用 @nestjs/typeorm 包省略的开销。 要了解更多信息,请参见此处

TypeORM 明显是 Node.js 世界中最成熟的对象关系映射(ORM)库。 由于它是用 TypeScript 编写的,因此它与 Nest 框架非常契合。

入门

要开始使用此库,我们必须安装所有必需的依赖项:

npm install --save typeorm mysql2

我们需要做的第一步是使用从 typeorm 包导入的 new DataSource().initialize() 类 来建立与我们的数据库的连接。initialize() 函数返回一个 Promise, 因此我们必须创建一个异步提供程序

database.providers.ts
import { DataSource } from 'typeorm';

export const databaseProviders = [
{
provide: 'DATA_SOURCE',
useFactory: async () => {
const dataSource = new DataSource({
type: 'mysql',
host: 'localhost',
port: 3306,
username: 'root',
password: 'root',
database: 'test',
entities: [
__dirname + '/../**/*.entity{.ts,.js}',
],
synchronize: true,
});

return dataSource.initialize();
},
},
];
warning

在生产中不应使用 synchronize: true,否则可能会丢失生产数据。

note

按照最佳实践,我们在具有 *.providers.ts 后缀的独立文件中声明了自定义提供程序。

然后,我们需要导出这些提供程序,以便它们可以被应用程序的其他部分访问。

database.module.ts
import { Module } from '@nestjs/common';
import { databaseProviders } from './database.providers';

@Module({
providers: [...databaseProviders],
exports: [...databaseProviders],
})
export class DatabaseModule {}

现在我们可以使用 @Inject() 装饰器注入 DATA_SOURCE 对象。 每个依赖于 DATA_SOURCE 异步提供程序的类都将等到 Promise 解析后再进行实例化。

仓储模式

TypeORM 支持仓储设计模式,因此每个实体都有自己的仓储。可以从数据库连接中获取这些仓储。

但首先,我们至少需要一个实体。我们将重用官方文档中的 Photo 实体。

photo.entity.ts
import { Entity, Column, PrimaryGeneratedColumn } from 'typeorm';

@Entity()
export class Photo {
@PrimaryGeneratedColumn()
id: number;

@Column({ length: 500 })
name: string;

@Column('text')
description: string;

@Column()
filename: string;

@Column('int')
views: number;

@Column()
isPublished: boolean;
}

Photo 实体属于 photo 目录。此目录表示 PhotoModule。现在,让我们创建一个 Repository 提供程序:

photo.providers.ts
import { DataSource } from 'typeorm';
import { Photo } from './photo.entity';

export const photoProviders = [
{
provide: 'PHOTO_REPOSITORY',
useFactory: (dataSource: DataSource) => dataSource.getRepository(Photo),
inject: ['DATA_SOURCE'],
},
];
warning

在实际应用程序中,应避免使用神奇字符串。PHOTO_REPOSITORYDATA_SOURCE 应该保留在单独的 constants.ts 文件中。

现在我们可以使用 @Inject() 装饰器将 Repository<Photo> 注入到 PhotoService 中:

photo.service.ts
import { Injectable, Inject } from '@nestjs/common';
import { Repository } from 'typeorm';
import { Photo } from './photo.entity';

@Injectable()
export class PhotoService {
constructor(
@Inject('PHOTO_REPOSITORY')
private photoRepository: Repository<Photo>,
) {}

async findAll(): Promise<Photo[]> {
return this.photoRepository.find();
}
}

数据库连接是异步的,但 Nest 使此过程对最终用户完全透明。 PhotoRepository 正在等待 db 连接,而 PhotoService 被推迟到仓储准备好使用时。 在每个类实例化时,整个应用程序都可以启动。

这是最终的 PhotoModule

photo.module.ts
import { Module } from '@nestjs/common';
import { DatabaseModule } from '../database/database.module';
import { photoProviders } from './photo.providers';
import { PhotoService } from './photo.service';

@Module({
imports: [DatabaseModule],
providers: [
...photoProviders,
PhotoService,
],
})
export class PhotoModule {}
tip

不要忘记将 PhotoModule 导入到根 AppModule 中。