贝利信息

TypeORM与NestJS应用中密码自动哈希的实现指南

日期:2025-12-05 00:00 / 作者:花韻仙語

本文详细介绍了在TypeORM与NestJS应用中,如何利用TypeORM实体生命周期钩子自动对用户密码进行哈希处理。通过在实体内部集成`@BeforeInsert()`和`@BeforeUpdate()`装饰器,结合`bcrypt`库,我们能够确保在用户模型持久化到数据库前,密码始终以安全哈希的形式存储,从而增强应用程序的安全性。

在构建任何涉及用户认证的应用程序时,密码的安全性是至关重要的。直接存储明文密码是严重的安全漏洞。最佳实践是存储密码的哈希值,并且这种哈希过程应该在密码被保存到数据库之前自动完成。在TypeORM和NestJS的组合应用中,我们可以利用TypeORM提供的实体生命周期钩子(Entity Lifecycle Hooks)来实现这一自动化过程。

1. 为什么需要自动哈希密码?

密码哈希是一种单向加密过程,它将密码转换为一串固定长度的字符,且无法逆向还原。即使数据库被泄露,攻击者也无法直接获取用户的原始密码。自动哈希确保了开发人员不必在每个保存用户的地方手动调用哈希函数,从而减少了错误并提高了代码的一致性和安全性。

2. TypeORM实体生命周期钩子

TypeORM提供了多种装饰器,允许我们在实体生命周期的特定阶段执行自定义逻辑。对于密码哈希,最常用的是:

这些钩子在实体实例上操作,允许我们修改即将持久化的数据。

3. 环境准备

首先,我们需要安装一个强大的密码哈希库,这里推荐使用bcrypt:

npm install bcrypt
npm install -D @types/bcrypt

4. 在实体中实现自动密码哈希

我们将以一个User实体为例,演示如何集成密码哈希逻辑。

4.1 用户实体结构

假设我们有一个User实体,其中包含username和password字段。password字段将用于存储哈希后的密码。

// src/user/user.entity.ts
import { Entity, PrimaryGeneratedColumn, Column, BeforeInsert, BeforeUpdate } from 'typeorm';
import * as bcrypt from 'bcrypt';

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

  @Column({ unique: true })
  username: string;

  @Column()
  password: string;

  // ... 其他字段
}

4.2 实现@BeforeInsert()钩子

当创建新用户时,我们需要在密码保存到数据库之前对其进行哈希。

// src/user/user.entity.ts (添加在 User 类内部)
// ...
export class User {
  // ... 其他字段和属性

  @BeforeInsert()
  async hashPasswordOnInsert() {
    if (this.password) {
      this.password = await bcrypt.hash(this.password, 10); // 10 是 saltRounds
    }
  }
}

解释:

4.3 实现@BeforeUpdate()钩子

当用户更新其密码时,我们也需要对新密码进行哈希。

// src/user/user.entity.ts (添加在 User 类内部)
// ...
export class User {
  // ... 其他字段和属性

  @BeforeInsert()
  async hashPasswordOnInsert() {
    if (this.password) {
      this.password = await bcrypt.hash(this.password, 10);
    }
  }

  @BeforeUpdate()
  async hashPasswordOnUpdate() {
    // 只有当密码字段被修改时才重新哈希
    // 注意:TypeORM的@BeforeUpdate钩子不会自动告诉你哪些字段被修改。
    // 如果你在服务层手动设置了新的明文密码,这个钩子会捕获到。
    // 如果你只更新了其他字段,而没有触碰password字段,则不会重新哈希。
    // 为了更精确地判断密码是否改变,可能需要额外逻辑(如在服务层比较旧密码或设置一个标志)。
    if (this.password && this.password.length < 60) { // 简单判断是否为明文密码(哈希密码通常更长)
      this.password = await bcrypt.hash(this.password, 10);
    }
  }
}

解释:

5. 完整用户实体示例

结合上述所有部分,完整的User实体将如下所示:

// src/user/user.entity.ts
import { Entity, PrimaryGeneratedColumn, Column, BeforeInsert, BeforeUpdate } from 'typeorm';
import * as bcrypt from 'bcrypt';

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

  @Column({ unique: true })
  username: string;

  @Column()
  password: string;

  @BeforeInsert()
  async hashPasswordOnInsert() {
    if (this.password) {
      this.password = await bcrypt.hash(this.password, 10);
    }
  }

  @BeforeUpdate()
  async hashPasswordOnUpdate() {
    // 只有当密码字段被修改且看起来是明文时才重新哈希
    // 这里使用长度判断作为一种简单机制,实际应用中可能需要更复杂的逻辑
    if (this.password && this.password.length < 60) {
      this.password = await bcrypt.hash(this.password, 10);
    }
  }

  // 辅助方法:用于验证密码,不直接存储在数据库中
  async validatePassword(password: string): Promise {
    return bcrypt.compare(password, this.password);
  }
}

6. 在服务层使用

在服务层创建或更新用户时,你只需将明文密码赋值给实体实例的password属性,TypeORM的钩子会自动处理哈希。

// src/user/user.service.ts
import { Injectable } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm';
import { User } from './user.entity';
import { CreateUserDto } from './dto/create-user.dto';
import { UpdateUserDto } from './dto/update-user.dto';

@Injectable()
export class UserService {
  constructor(
    @InjectRepository(User)
    private usersRepository: Repository,
  ) {}

  async create(createUserDto: CreateUserDto): Promise {
    const newUser = this.usersRepository.create(createUserDto);
    // 在这里设置明文密码,@BeforeInsert 会自动哈希
    // newUser.password = createUserDto.password; // create方法通常会直接映射dto,所以这行可能不需要显式写
    return this.usersRepository.save(newUser);
  }

  async update(id: number, updateUserDto: UpdateUserDto): Promise {
    const user = await this.usersRepository.findOneBy({ id });
    if (!user) {
      throw new Error('User not found');
    }
    // 如果 updateUserDto 包含新密码,则会更新 user.password
    // @BeforeUpdate 会自动处理新密码的哈希
    Object.assign(user, updateUserDto);
    return this.usersRepository.save(user);
  }

  async findOneByUsername(username: string): Promise {
    return this.usersRepository.findOne({ where: { username } });
  }
}

7. 密码验证

在用户登录时,你需要验证用户输入的明文密码是否与数据库中存储的哈希密码匹配。bcrypt.compare()方法是专门为此设计的。

// src/auth/auth.service.ts (示例)
import { Injectable, UnauthorizedException } from '@nestjs/common';
import { UserService } from '../user/user.service';
import { LoginDto } from './dto/login.dto';

@Injectable()
export class AuthService {
  constructor(private userService: UserService) {}

  async validateUser(username: string, pass: string): Promise {
    const user = await this.userService.findOneByUsername(username);
    if (user && await user.validatePassword(pass)) { // 使用实体中的辅助方法验证密码
      const { password, ...result } = user;
      return result;
    }
    return null;
  }

  async login(loginDto: LoginDto) {
    const user = await this.validateUser(loginDto.username, loginDto.password);
    if (!user) {
      throw new UnauthorizedException('Invalid credentials');
    }
    // ... 生成 JWT token 等后续操作
    return user;
  }
}

8. 注意事项与最佳实践

总结

通过在TypeORM实体中使用@BeforeInsert()和@BeforeUpdate()生命周期钩子,并结合bcrypt库,我们可以优雅且高效地在NestJS应用程序中实现密码的自动哈希。这种方法不仅简化了开发流程,更重要的是,它极大地增强了应用程序处理用户敏感数据的安全性,是现代Web应用开发中不可或缺的一环。