NestJS-登录注册

这里我们后续依赖使用到的几个依赖模块如下（可以提前下载好）

npm install bcrypt // 加密密码
npm install class-validator  //装饰器来验证数据的有效性

注册模块

register.dto.ts注册规则

新建一个register.dto.ts存放我们注册部分规则

// src/modules/user/dto/register.dto.ts
import { IsString, IsEmail, MinLength } from 'class-validator';

export class RegisterDto {
  @IsString()
  username: string;
  @IsString()
  @MinLength(3)
  password: string;
}

user.service.ts

import { Injectable ,ConflictException} from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm';
import { User } from './user.entity';
import { RegisterDto } from './dto/register.dto';
@Injectable()
export class UserService {
  constructor(
    @InjectRepository(User) private userRepository: Repository<User>,
  ) {}
  // 获取所有用户
  async findAll(): Promise<User[]> {
    return this.userRepository.find();
  }
  async register(registerDto: RegisterDto): Promise<User> {
    // 检查用户是否已经存在
    const existingUser = await this.userRepository.findOne({ where: { username: registerDto.username } });
    if (existingUser) {
      throw new ConflictException('Username already exists');
    }
    // 创建新用户
    const user = new User();
    user.username = registerDto.username;
    user.password = registerDto.password;

    // 保存到数据库
    return this.userRepository.save(user);
  }
}

user.controller.ts

更改一下我们使用的注册接口

import { Controller, Get, Post, Param, Body, Put, Delete } from '@nestjs/common';
import { UserService } from './user.service';
import { User } from './user.entity';
import { RegisterDto } from './dto/register.dto';
@Controller()
export class UserController {
  constructor(private readonly userService: UserService) {}
  @Get('users')
  async findAll(): Promise<User[]> {
    return this.userService.findAll();
  }
  @Post('register')
  async register(@Body() registerDto: RegisterDto): Promise<User> {
    return this.userService.register(registerDto);
  }
}

检测注册模块

接下来我们测试一下我们的数据模块,用户存在的时候,这里给我们的提示

{
    "message": "Username already exists",
    "error": "Conflict",
    "statusCode": 409
}

手动设置请求的返回格式

后续优化看代码，这里只是简单返回一下

 return {
  data: [],
  message: '用户已存在！',
  code: 400,
};

再次注册

//存在用户
{
    "code": 400,
    "message": "用户已存在！"
}



//新用户注册
{
    "message": "恭喜你，您的账号注册成功！",
    "code": 200
}

登录模块编写

这里我们直接也是将登录编写进了用户部分，先以实现功能为主

用户dto部分

先编写一个用户对应的规则dto部分

// src/modules/user/dto/login.dto.ts
import { IsString, IsEmail, MinLength } from 'class-validator';
export class LoginDto{
  @IsString()
  username: string;
  @IsString()
  @MinLength(3)
  password: string;
}

user.service.ts之中添加登录方法

import { LoginDto } from './dto/login.dto';

// 异步登录
  async login(loginDto: LoginDto){
  // 查询数据库中是否存在同名用户

  // 1. 查询数据库中是否存在该用户
  const loginUser = await this.userRepository.findOne({
    where: { username: loginDto.username },
  });

   // 2. 如果用户不存在，返回异常信息
   if (!loginUser) {
    throw new HttpException(
      {
        code: 404, // 适当的错误码是 404 (用户未找到)
        message: '用户不存在！',
      },
      HttpStatus.NOT_FOUND, // 使用 404 状态码
    );
  }
  // 3. 比较密码是否正确
  if (loginDto.password  !==  loginUser.password) {
    throw new HttpException(
      {
        code: 401, // 未授权错误码
        message: '密码错误！',
      },
      HttpStatus.UNAUTHORIZED, // 使用 401 状态码
    );
  }

  // 4. 返回登录成功信息，可以添加 JWT 或 Session 等信息
  return {
    message: '登录成功！',
    code: 200,
    data: {
      username: loginUser.username,
      userId: loginUser.userId,
      // 可以附带其他的用户信息，如权限、token等
    },
  };
}

user.controller.ts之中使用

上面的方法服务写完了，但是我们还没有在控制器之中使用，这个时候需要使用一下

可以先导入我们的dto规则部分

import { LoginDto } from './dto/login.dto';

post接口调佣

 @Post('login')
// 登录方法
 login(@Body() loginDto: LoginDto){
  // 调用userService中的login方法
  return this.userService.login(loginDto);
}

核查登录模块

核查登录

接下来我们尝试调一下我们的登录接口

http://localhost:8888/api/login

登录正确时候

{
    "message": "登录成功！",
    "code": 200,
    "data": {
        "username": "admin",
        "userId": 6
    }
}

登录错误时候

{
    "code": 404,
    "message": "用户不存在！"
}

这个时候我们的登录和注册模块功能已经好了