Drizzle와 Neon 시작하기

Drizzle는 neon-http와 neon-websockets 드라이버를 통해 Neon 연결을 기본적으로 지원합니다. 이 드라이버들은 내부적으로 neon-serverless 드라이버를 사용합니다.

neon-http와 neon-websockets 드라이버를 사용하면 서버리스 환경에서 TCP 대신 HTTP 또는 WebSockets를 통해 Neon 데이터베이스에 접근할 수 있습니다. 단일 비대화형 트랜잭션의 경우 HTTP를 통한 쿼리가 더 빠릅니다.

세션 또는 대화형 트랜잭션 지원이 필요하거나 pg 드라이버와 완전히 호환되는 대체 드라이버가 필요하다면, WebSocket 기반의 neon-serverless 드라이버를 사용할 수 있습니다. Postgres를 통해 Neon 데이터베이스에 직접 연결할 수도 있습니다.

기본 파일 구조

이 프로젝트의 기본 파일 구조입니다. src/db 디렉토리 안에는 schema.ts 파일에 테이블 정의가 있습니다. drizzle 폴더에는 SQL 마이그레이션 파일과 스냅샷이 있습니다.

📦 <project root>
 ├ 📂 drizzle
 ├ 📂 src
 │   ├ 📂 db
 │   │  └ 📜 schema.ts
 │   └ 📜 index.ts
 ├ 📜 .env
 ├ 📜 drizzle.config.ts
 ├ 📜 package.json
 └ 📜 tsconfig.json

1단계 - @neondatabase/serverless 패키지 설치

npm i drizzle-orm @neondatabase/serverless dotenv
npm i -D drizzle-kit tsx

yarn add drizzle-orm @neondatabase/serverless dotenv
yarn add -D drizzle-kit tsx

pnpm add drizzle-orm @neondatabase/serverless dotenv
pnpm add -D drizzle-kit tsx

bun add drizzle-orm @neondatabase/serverless dotenv
bun add -D drizzle-kit tsx

2단계 - 연결 변수 설정

프로젝트 루트 디렉토리에 .env 파일을 생성하고 데이터베이스 연결 변수를 추가하세요:

DATABASE_URL=

3단계 - Drizzle ORM을 데이터베이스에 연결하기

src/db 디렉토리에 index.ts 파일을 생성하고 데이터베이스 연결을 초기화해 보겠습니다.

import { drizzle } from 'drizzle-orm/neon-http';

const db = drizzle(process.env.DATABASE_URL);

동기식 연결이 필요한 경우, 추가 연결 API를 사용할 수 있습니다. 이때 드라이버 연결을 지정하고 Drizzle 인스턴스에 전달합니다.

import { neon } from '@neondatabase/serverless';
import { drizzle } from 'drizzle-orm/neon-serverless';

const sql = neon(process.env.DATABASE_URL!);
const db = drizzle({ client: sql });

4단계 - 테이블 생성

src/db 디렉토리에 schema.ts 파일을 생성하고 테이블을 선언해 보겠습니다.

import { integer, pgTable, varchar } from "drizzle-orm/pg-core";

export const usersTable = pgTable("users", {
  id: integer().primaryKey().generatedAlwaysAsIdentity(),
  name: varchar({ length: 255 }).notNull(),
  age: integer().notNull(),
  email: varchar({ length: 255 }).notNull().unique(),
});

5단계 - Drizzle 설정 파일 설정하기

Drizzle 설정 파일 - Drizzle Kit에서 사용하는 설정 파일로, 데이터베이스 연결 정보, 마이그레이션 폴더, 스키마 파일 등 모든 정보를 포함합니다.

프로젝트 루트에 drizzle.config.ts 파일을 생성하고 다음 내용을 추가하세요:

import 'dotenv/config';
import { defineConfig } from 'drizzle-kit';

export default defineConfig({
  out: './drizzle',
  schema: './src/db/schema.ts',
  dialect: 'postgresql',
  dbCredentials: {
    url: process.env.DATABASE_URL!,
  },
});

6단계 - 데이터베이스에 변경 사항 적용하기

drizzle-kit push 커맨드를 사용하면 데이터베이스에 직접 변경 사항을 적용할 수 있습니다. 이 방법은 로컬 개발 환경에서 새로운 스키마 디자인이나 수정 사항을 빠르게 테스트할 때 유용합니다. 마이그레이션 파일을 관리할 필요 없이 빠르게 반복 작업을 진행할 수 있습니다:

npx drizzle-kit push

push 커맨드에 대해 더 알아보려면 문서를 참고하세요.

npx drizzle-kit generate

npx drizzle-kit migrate

7단계 - 데이터베이스 시드 및 쿼리

src/index.ts 파일을 사용자 생성, 조회, 수정, 삭제 쿼리로 업데이트해 보겠습니다.

import 'dotenv/config';
import { drizzle } from 'drizzle-orm/neon-http';
import { eq } from 'drizzle-orm';
import { usersTable } from './db/schema';

const db = drizzle(process.env.DATABASE_URL!);

async function main() {
  // 새로운 사용자 데이터 생성
  const user: typeof usersTable.$inferInsert = {
    name: 'John',
    age: 30,
    email: 'john@example.com',
  };

  // 사용자 생성
  await db.insert(usersTable).values(user);
  console.log('새로운 사용자가 생성되었습니다!');

  // 모든 사용자 조회
  const users = await db.select().from(usersTable);
  console.log('데이터베이스에서 모든 사용자를 가져옵니다: ', users);
  /*
  const users: {
    id: number;
    name: string;
    age: number;
    email: string;
  }[]
  */

  // 사용자 정보 업데이트
  await db
    .update(usersTable)
    .set({
      age: 31,
    })
    .where(eq(usersTable.email, user.email));
  console.log('사용자 정보가 업데이트되었습니다!');

  // 사용자 삭제
  await db.delete(usersTable).where(eq(usersTable.email, user.email));
  console.log('사용자가 삭제되었습니다!');
}

main();

8단계 - index.ts 파일 실행하기

TypeScript 파일을 실행하는 방법은 여러 가지가 있지만, 여기서는 tsx를 사용하는 방법을 살펴보겠습니다.

이미 tsx를 설치했으므로, 이제 쿼리를 실행할 수 있습니다.

index.ts 스크립트 실행하기

<Npx>
  tsx src/index.ts
</Npx>

bun src/index.ts