nestjs-drizzle

You can inject the Drizzle instances using their respective tag specified in the configuration

import { Inject, Injectable } from '@nestjs/common';
import * as schema from '../db/schema';
import { LibSQLDatabase } from 'drizzle-orm/libsql';
@Injectable()
export class AppService {
  constructor(
    @Inject('DB_DEV') private drizzleDev: LibSQLDatabase<typeof schema>
  ) {}
  async getData() {
    const books = await this.drizzleDev.query.books.findMany();
    const authors = await this.drizzleDev.query.authors.findMany();
    return {
      books: books,
      authors: authors,
    };
  }
}

In case you need the access to the underlying connection client for tasks such as connection termination or other operations, you must define the session property manually, as Drizzle does not currently expose it. In the following example, we demonstrate how to access the client to close all database connections when our NestJS application is terminated, using the onApplicationShutdown hook.

Note that the specific client type vary depending on the database driver being used, so adjust the type accordingly.

import { MySql2Client, MySql2Database } from 'drizzle-orm/mysql2';

import { Inject, Injectable, OnApplicationShutdown } from '@nestjs/common';

import * as schema from './schema/mysql';

export type ExtendedMySql2Database<T extends Record<string, unknown>> =
  MySql2Database<T> & {
    session: {
      client: MySql2Client;
    };
  };

@Injectable()
export class AppService implements OnApplicationShutdown {
  constructor(
    @Inject('DB')
    private drizzleProd: ExtendedMySql2Database<typeof schema>,
  ) {}

  async getHello() {
    const books = await this.drizzleProd.query.books.findMany();
    const authors = await this.drizzleProd.query.authors.findMany();
    return {
      books: books,
      authors: authors,
    };
  }
  
  async onApplicationShutdown() {
    await this.drizzleProd.session.client.end();
  }
}

You can read the detailed documentation of each of the module from the links given below

Documentation

• @knaadh/nestjs-drizzle-postgres
• @knaadh/nestjs-drizzle-pg
• @knaadh/nestjs-drizzle-mysql2
• @knaadh/nestjs-drizzle-better-sqlite3
• @knaadh/nestjs-drizzle-turso
• @knaadh/nestjs-drizzle-planetscale

License

This package is MIT licensed.