【NestJS】构建可复用的数据存储模块 - 动态模块

动态模块

Nest 中的动态模块允许你创建可在运行时配置的模块,而不是使用静态的固定配置。当你需要提供灵活、可定制的模块时,这尤其有用。


使用场景

在项目中有用户(UserModule)、书籍(BookModule)等模块,我希望通过 json 文件来存储对应模块数据。 这个时候就可以定义一个存储模块(DbModule),用来定义存储的读取和写入方法,其他模块在使用存储模块时,传入 json 文件路径。

在 UserModule 里用的时候,path 是 users.json,在 BookModule 用的时候,path 是 books.json。

这个 DbModule 就是动态模块


创建动态模块

1. 定义动态模块

ts 复制代码
import { DynamicModule, Module } from '@nestjs/common';
import { DbService } from './db.service';
​
export interface DbModuleOptions {
  path: string
}
​
@Module({})
export class DbModule {
  static register(options: DbModuleOptions): DynamicModule {
    return {
      module: DbModule,
      providers: [
        {
          provide: "OPTIONS",
          useValue: options
        },
        DbService,
      ],
      exports: [DbService]
    }
  }
}

该方法返回一个 DynamicModule 对象,包含以下属性:

  • module: 指定这个动态模块对应的类

  • providers: 定义该模块提供的服务

    • 把传入的 options 用 useValue 来声明为 provider;token 为 OPTIONS。

      记住 provide 定义的值,后面要考(会用到)

    • DbService 中实现读取和写入方法

  • exports: 指定哪些服务可以被导入该模块的其他模块使用

2. 定义动态模块对应的 service

service 中实现读取和写入方法,注入

ts 复制代码
import { Inject, Injectable } from '@nestjs/common';
import { DbModuleOptions } from './db.module';
import { access, readFile, writeFile } from 'fs/promises';
​
@Injectable()
export class DbService {
  @Inject('OPTIONS')
  private options: DbModuleOptions;
​
  // 读取
  async read() {
    const filePath = this.options.path;
    // 读取文件内容
    const str = await readFile(filePath, { encoding: 'utf-8' });
    // 转为对象
    return JSON.parse(str);
  }
​
  // 写入
  async write(obj: Record<string, any>) {
    await writeFile(this.options.path, JSON.stringify(obj || []), {
      encoding: 'utf-8',
    });
  }
}
​

使用动态模块

在 UserModule 中使用

ts 复制代码
import { Module } from '@nestjs/common';
import { UserService } from './user.service';
import { UserController } from './user.controller';
import { DbModule } from 'src/db/db.module';
​
@Module({
  imports: [
    DbModule.register({
      path: 'users.json',
    }),
  ],
  controllers: [UserController],
  providers: [UserService],
})
export class UserModule {}
​

在 UserService 中使用就和静态模块一样了

js 复制代码
import { Inject, Injectable } from '@nestjs/common';
import { DbService } from '../db/db.service';
import { User } from './entities/user.entity';
​
@Injectable()
export class UserService {
  @Inject(DbService)
  dbService: DbService;
​
  async register() {
    // 读取
    const users = await this.dbService.read();
​
    // ...
​
    // 写入
    await this.dbService.write(user)
    return user
  }
}
相关推荐
干就完了15 小时前
js数组方法,其实也就这么多东西,一篇全搞懂
前端·javascript
JIE_5 小时前
【Hero动画】用一个指令实现Vue跨路由/组件动画
前端
aidingni8885 小时前
Comet浏览器不为人知的故事
前端·javascript
武子康5 小时前
大数据-91 Spark广播变量:高效共享只读数据的最佳实践 RDD+Scala编程
大数据·后端·spark
努力的小郑5 小时前
MySQL索引(二):覆盖索引、最左前缀原则与索引下推详解
后端·mysql
阿拉伦5 小时前
智能交通拥堵治理柔性设计实践复盘小结
后端
用户4099322502125 小时前
如何在 FastAPI 中优雅地模拟多模块集成测试?
后端·ai编程·trae
一枝花算不算浪漫5 小时前
线上频繁FullGC?慌得一比!竟是Log4j2的这个“特性”坑了我
jvm·后端
Cache技术分享5 小时前
182. Java 包 - 创建和使用 Java 包
前端·后端