在Nest.js框架中使用Drizzle ORM时，很多开发者可能已经习惯了Prisma的使用方式，但Drizzle ORM的集成似乎不如Prisma那样直观。本文将探讨如何在Nest.js中更优雅地使用Drizzle ORM，提供一种类似于PrismaService的使用体验。
背景介绍
Drizzle ORM是PostgreSQL的一个轻量级ORM，旨在提供简洁的SQL查询构建和类型安全的数据库操作。与Prisma不同，Drizzle ORM的集成通常需要通过Nest.js的providers来完成，这可能看起来不那么直观。
传统的Drizzle ORM集成
通常，我们会通过以下方式在Nest.js中集成Drizzle ORM：
import{Module}from'@nestjs/common';import{ConfigService}from'@nestjs/config';import{drizzle}from'drizzle-orm/postgres-js';importpostgresfrom'postgres';import*asschemafrom'./schema';exportconstPG_CONNECTION="PG_CONNECTION";@Module({providers:[{provide:PG_CONNECTION,inject:[ConfigService],useFactory:async(config:ConfigService)=>{constconnection=postgres(config.get("DATABASE_URL"));returndrizzle(connection,{schema});},},],exports:[PG_CONNECTION],})exportclassDrizzleModule{}然后，在服务中注入这个连接：
import{Inject,Injectable}from'@nestjs/common';import{PG_CONNECTION}from'../drizzle/drizzle.module';import{dbType}from'drizzle-orm/postgres-js';import*asschemafrom'../drizzle/schema';@Injectable()exportclassUsersService{constructor(@Inject(PG_CONNECTION)privatedrizzle:dbType<typeofschema>){}asyncfindAll(){returnawaitthis.drizzle.query.users.findMany();}}这种方式虽然有效，但每次需要注入PG_CONNECTION并导入类型和常量，这显得有点繁琐。
优雅的集成方式
为了实现类似于Prisma的简洁体验，我们可以创建一个DrizzleService来封装这个连接：
@Injectable()exportclassDrizzleService{constructor(@Inject(PG_CONNECTION)readonlydb:dbType<typeofschema>){}}然后在DrizzleModule中导出这个服务：
@Module({providers:[DrizzleService],exports:[DrizzleService],})exportclassDrizzleModule{}这样，在其他服务中，我们可以这样注入和使用Drizzle ORM：
import{Inject,Injectable}from'@nestjs/common';import{DrizzleService}from'../drizzle/drizzle.service';@Injectable()exportclassUsersService{constructor(privatereadonlydrizzle:DrizzleService){}asyncfindAll(){returnawaitthis.drizzle.db.query.users.findMany();}}实例解析
假设我们有一个用户数据库表，我们可以通过UsersService来查询用户数据：
asyncfindUserById(id:number){constuser=awaitthis.drizzle.db.query.users.findFirst({where:(users,{eq})=>eq(users.id,id),});returnuser;}这种方式不仅使代码更加简洁，也更符合Nest.js的设计哲学，即通过服务注入来管理依赖。
总结
通过创建一个DrizzleService来封装Drizzle ORM的连接，我们可以在Nest.js项目中更优雅地使用Drizzle ORM。这样的设计不仅使代码结构更加清晰，也降低了使用复杂度，让开发者能够更加专注于业务逻辑的实现，而不必过多考虑数据库连接的细节。希望这种方法能帮助更多开发者在Nest.js项目中高效地使用Drizzle ORM。
