一篇文章学会如何在 NestJS 中集成 MongoDB 并实现数据的增删改查操作

2024-01-07 23:34:35

前言

在现代的Web应用程序开发中,无论是在数据存储、检索、还是数据流转的各个环节,数据库都扮演着极其重要的角色。MongoDB是一个基于分布式文件存储的开源数据库系统,以其高性能、高可用性和易扩展性著称。

作为JavaScript社区最受欢迎的后端框架之一,NestJS 提供了优雅的解决方案来与各种数据库进行交互,其中就包括了对MongoDB的支持。

本教程将通过一个简单而通俗的实例,带你在NestJS中集成MongoDB。我们将创建一个简易的用户管理系统,实现用户的增删改查操作。

准备工作

在开始之前,请确保你的开发环境中已经安装了以下软件:

  • Node.js(建议使用LTS版本)
  • npm 或 yarn(作为Node.js的包管理工具)
  • MongoDB(你可以在本机上安装MongoDB,或者使用MongoDB Atlas服务)
  • Nest CLI(通过 npm i -g @nestjs/cli 安装)

准备就绪后,我们可以开始动手编写代码了。

使用步骤

一、创建新的 NestJS 项目

使用 Nest CLI 创建一个新项目:

nest new nest-mongodb-example

进入项目目录:

cd nest-mongodb-example

二、安装 MongoDB 模块

在NestJS中操作MongoDB,我们将使用 @nestjs/mongoose模块,同时也需要安装mongoose:

npm install @nestjs/mongoose mongoose

三、配置 MongoDB 连接

我们需要配置NestJS应用来连接到MongoDB数据库。在 src目录下创建一个名为 configs的文件夹,并在其中创建 mongodb.config.ts文件:

// src/configs/mongodb.config.ts
export default {
  uri: 'mongodb://localhost/nest-mongodb-example',
};

app.module.ts中导入上述配置并初始化:

// src/app.module.ts
import { Module } from '@nestjs/common';
import { MongooseModule } from '@nestjs/mongoose';
import mongodbConfig from './configs/mongodb.config';

@Module({
  imports: [
    MongooseModule.forRoot(mongodbConfig.uri),
    // ...其他模块
  ],
  controllers: [],
  providers: [],
})
export class AppModule {}

四、创建数据模型 Schema

在 MongoDB 中, 我们使用 Schema 定义数据的结构。NestJS配合Mongoose可以让我们方便的操作它。

src目录下创建一个 users文件夹用来组织用户相关的文件。在其中创建 user.schema.ts用来定义用户数据模型:

// src/users/user.schema.ts
import { Prop, Schema, SchemaFactory } from '@nestjs/mongoose';
import { Document } from 'mongoose';

export type UserDocument = User & Document;

@Schema()
export class User {
  @Prop({ required: true })
  username: string;

  @Prop()
  email: string;

  @Prop()
  age: number;
}

export const UserSchema = SchemaFactory.createForClass(User);

五、注册数据模型

然后创建一个对应的服务模块来提供用户的业务逻辑:

nest g module users
nest g service users

这将会自动生成 users.module.tsusers.service.ts文件。

现在,我们将 User模型和 UserSchema导入到 UsersModule中:

// src/users/users.module.ts
import { Module } from '@nestjs/common';
import { MongooseModule } from '@nestjs/mongoose';
import { User, UserSchema } from './user.schema';
import { UsersService } from './users.service';

@Module({
  imports: [
    MongooseModule.forFeature([{ name: User.name, schema: UserSchema }])
  ],
  providers: [UsersService]
})
export class UsersModule {}

六、创建操作 MongoDB 的服务

现在,我们的 UsersService中可以通过依赖注入方式访问 User模型了。在 users.service.ts文件中,我们可以实现一些基本的数据库操作:

// src/users/users.service.ts
import { Injectable, NotFoundException } from '@nestjs/common';
import { InjectModel } from '@nestjs/mongoose';
import { Model } from 'mongoose';
import { User, UserDocument } from './user.schema';
import { CreateUserDto } from './dto/create-user.dto';
import { UpdateUserDto } from './dto/update-user.dto';

@Injectable()
export class UsersService {
  constructor(@InjectModel(User.name) private userModel: Model<UserDocument>) {}

  async create(createUserDto: CreateUserDto): Promise<User> {
    const createdUser = new this.userModel(createUserDto);
    return createdUser.save();
  }

  async findAll(): Promise<User[]> {
    return this.userModel.find().exec();
  }

  async findOne(id: string): Promise<User> {
    const user = this.userModel.findById(id).exec();
    if (!user) {
      throw new NotFoundException(`User with ID ${id} not found`);
    }
    return user;
  }

  async update(id: string, updateUserDto: UpdateUserDto): Promise<User> {
    const updatedUser = await this.userModel.findByIdAndUpdate(id, updateUserDto, { new: true }).exec();
    if (!updatedUser) {
      throw new NotFoundException(`User with ID ${id} not found`);
    }
    return updatedUser;
  }

  async delete(id: string): Promise<User> {
    const deletedUser = await this.userModel.findByIdAndRemove(id).exec();
    if (!deletedUser) {
      throw new NotFoundException(`User with ID ${id} not found`);
    }
    return deletedUser;
  }
}

七、实现增删改查操作

为了完成我们的CURD操作,我们还需要实现相应的 UsersController来处理HTTP请求。

此时我们同样应该创建DTO(Data Transfer Object)用来安全的传输数据。

首先,创建 create-user.dto.ts:

// src/users/dto/create-user.dto.ts
export class CreateUserDto {
  readonly username: string;
  readonly email?: string;
  readonly age?: number;
}

// src/users/dto/update-user.dto.ts
import { PartialType } from '@nestjs/mapped-types';
import { CreateUserDto } from './create-user.dto';

export class UpdateUserDto extends PartialType(CreateUserDto) {}

然后创建 UsersController:

nest g controller users

users.controller.ts中实现API的接口:

// src/users/users.controller.ts
import { Controller, Get, Post, Body, Put, Param, Delete } from '@nestjs/common';
import { CreateUserDto } from './dto/create-user.dto';
import { UpdateUserDto } from './dto/update-user.dto';
import { UsersService } from './users.service';
import { User } from './user.schema';

@Controller('users')
export class UsersController {
  constructor(private readonly usersService: UsersService) {}

  @Post()
  async create(@Body() createUserDto: CreateUserDto): Promise<User> {
    return this.usersService.create(createUserDto);
  }

  @Get()
  async findAll(): Promise<User[]> {
    return this.usersService.findAll();
  }

  @Get(':id')
  async findOne(@Param('id') id: string): Promise<User> {
    return this.usersService.findOne(id);
  }

  @Put(':id')
  async update(@Param('id') id: string, @Body() updateUserDto: UpdateUserDto): Promise<User> {
    return this.usersService.update(id, updateUserDto);
  }

  @Delete(':id')
  async delete(@Param('id') id: string): Promise<User> {
    return this.usersService.delete(id);
  }
}

现在,我们把 UsersModule加入到 AppModule的imports数组里:

// src/app.module.ts
import { Module } from '@nestjs/common';
import { MongooseModule } from '@nestjs/mongoose';
import mongodbConfig from './configs/mongodb.config';
import { UsersModule } from './users/users.module';

@Module({
  imports: [
    MongooseModule.forRoot(mongodbConfig.uri),
    UsersModule,
    // ...其他模块
  ],
  // ...其他配置
})
export class AppModule {}

到这里,我们已经完成了NestJS与MongoDB的集成,以及一个用户增加和查询功能的简单例子。通过这样的方式,我们便实现了在NestJS中针对MongoDB的增删改查操作。下面是每个操作对应的功能:

  • POST /users 创建用户
  • GET /users 获取所有用户列表
  • GET /users/:id 根据ID获取单个用户
  • PUT /users/:id 根据ID更新用户信息
  • DELETE /users/:id 根据ID删除用户

结语

集成NestJS和MongoDB是一个相对直接的过程,感谢NestJS对于Mongoose的封装,使得我们可以非常便捷地管理数据库操作和数据模型。同时,我们可以享受到TypeScript的类型安全和装饰器带来的元数据功效,使得数据库相关代码更加清晰易懂。希望本教程可以帮助你在项目中快速集成NestJS和MongoDB,并构建出强大的后端服务。

文章来源:https://blog.csdn.net/m0_37890289/article/details/135396632
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。