Skip to content

The Drizzle module integrates drizzle-orm with nestelia's dependency injection system. It is dialect-agnostic — any drizzle-supported database (PostgreSQL, MySQL, SQLite) works out of the box.

Installation

bash
bun add drizzle-orm

Add the driver for your database:

bash
# PostgreSQL
bun add postgres        # postgres.js (recommended)
bun add pg              # node-postgres

# MySQL
bun add mysql2

# SQLite (no extra install needed in Bun)
# bun:sqlite is built into Bun

# LibSQL / Turso
bun add @libsql/client

Setup

Create your drizzle instance and pass it to DrizzleModule.forRoot():

typescript
import { Database } from "bun:sqlite";
import { drizzle } from "drizzle-orm/bun-sqlite";
import { Module } from "nestelia";
import { DrizzleModule } from "nestelia/drizzle";
import * as schema from "./schema";

const sqlite = new Database("app.db");
const db = drizzle(sqlite, { schema });

@Module({
  imports: [
    DrizzleModule.forRoot({ db, isGlobal: true }),
  ],
})
export class AppModule {}

PostgreSQL example

typescript
import postgres from "postgres";
import { drizzle } from "drizzle-orm/postgres-js";
import * as schema from "./schema";

const client = postgres(process.env.DATABASE_URL!);
const db = drizzle(client, { schema });

DrizzleModule.forRoot({ db, isGlobal: true })

Async configuration

Use forRootAsync when the connection string comes from a config service or environment:

typescript
import { DrizzleModule } from "nestelia/drizzle";
import { ConfigService } from "./config.service";
import * as schema from "./schema";

DrizzleModule.forRootAsync({
  isGlobal: true,
  inject: [ConfigService],
  useFactory: (config: ConfigService) => {
    const client = postgres(config.get("DATABASE_URL"));
    return { db: drizzle(client, { schema }) };
  },
})

useClass / useExisting

typescript
import { DrizzleOptionsFactory, DrizzleModuleOptions } from "nestelia/drizzle";
import { Injectable } from "nestelia";

@Injectable()
class DatabaseConfig implements DrizzleOptionsFactory {
  constructor(private readonly config: ConfigService) {}

  createDrizzleOptions(): DrizzleModuleOptions {
    const client = postgres(this.config.get("DATABASE_URL"));
    return { db: drizzle(client, { schema }) };
  }
}

DrizzleModule.forRootAsync({ useClass: DatabaseConfig })

Defining a schema

typescript
// src/schema.ts
import { integer, sqliteTable, text } from "drizzle-orm/sqlite-core";

export const users = sqliteTable("users", {
  id: integer("id").primaryKey({ autoIncrement: true }),
  name: text("name").notNull(),
  email: text("email").notNull().unique(),
});

export type User = typeof users.$inferSelect;
export type NewUser = typeof users.$inferInsert;

Injecting the database

@InjectDrizzle()

Use the @InjectDrizzle() decorator as a shorthand for @Inject(DRIZZLE_INSTANCE):

typescript
import { Injectable } from "nestelia";
import { InjectDrizzle } from "nestelia/drizzle";
import { eq } from "drizzle-orm";
import { drizzle } from "drizzle-orm/bun-sqlite";
import * as schema from "./schema";

type DB = ReturnType<typeof drizzle<typeof schema>>;

@Injectable()
export class UsersService {
  constructor(@InjectDrizzle() private readonly db: DB) {}

  findAll() {
    return this.db.select().from(schema.users).all();
  }

  findOne(id: number) {
    return this.db
      .select()
      .from(schema.users)
      .where(eq(schema.users.id, id))
      .get();
  }

  create(data: schema.NewUser) {
    return this.db
      .insert(schema.users)
      .values(data)
      .returning()
      .get();
  }
}

Using the injection token directly

typescript
import { Injectable, Inject } from "nestelia";
import { DRIZZLE_INSTANCE } from "nestelia/drizzle";

@Injectable()
export class UsersService {
  constructor(@Inject(DRIZZLE_INSTANCE) private readonly db: DB) {}
}

Multiple databases

Register each database with a unique tag and inject it using the same tag:

typescript
// app.module.ts
@Module({
  imports: [
    DrizzleModule.forRoot({ db: primaryDb }),
    DrizzleModule.forRoot({ db: analyticsDb, tag: "analytics" }),
  ],
})
export class AppModule {}
typescript
// report.service.ts
@Injectable()
export class ReportService {
  constructor(
    @InjectDrizzle() private readonly primary: PrimaryDB,
    @InjectDrizzle("analytics") private readonly analytics: AnalyticsDB,
  ) {}
}

Options reference

DrizzleModuleOptions

PropertyTypeDefaultDescription
dbunknownPre-configured drizzle-orm database instance (required)
tagstring | symbolDRIZZLE_INSTANCECustom injection token
isGlobalbooleanfalseRegister as a global module

DrizzleModuleAsyncOptions

PropertyTypeDescription
tagstring | symbolCustom injection token
isGlobalbooleanRegister as a global module
importsany[]Modules to import into this module's scope
useFactory(...args) => DrizzleModuleOptionsFactory function
injectany[]Dependencies for the factory
useClassType<DrizzleOptionsFactory>Class implementing DrizzleOptionsFactory
useExistingType<DrizzleOptionsFactory>Existing provider implementing DrizzleOptionsFactory
extraProvidersProvider[]Additional providers within this module's scope

Exports

ExportDescription
DrizzleModuleThe module class
InjectDrizzle(tag?)Shorthand parameter decorator
DRIZZLE_INSTANCEDefault injection token
DRIZZLE_MODULE_OPTIONSToken for raw options
DrizzleModuleOptionsSync options interface
DrizzleModuleAsyncOptionsAsync options interface
DrizzleOptionsFactoryFactory interface for useClass / useExisting

Released under the MIT License.