Drizzle과 Effect PostgreSQL 시작하기

WARNING

이 페이지는 Drizzle 버전 1.0.0-beta.9 이상에서 사용 가능한 기능을 설명합니다.

Effect는 현재 PostgreSQL에서만 사용 가능하며, 곧 다른 모든 데이터베이스에도 구현될 예정입니다.

업그레이드 방법은 여기를 참조하세요.

This guide assumes familiarity with:
  • Effect - 개발자가 복잡한 동기 및 비동기 프로그램을 쉽게 작성할 수 있도록 설계된 강력한 TS 라이브러리 - 자세히 보기
  • dotenv - 환경 변수 관리 패키지 - 자세히 보기
  • tsx - TypeScript 파일 실행 패키지 - 자세히 보기
  • @effect/sql-pg - Effect를 위한 PostgreSQL 툴킷 - 자세히 보기

Drizzle은 @effect/sql-pg 드라이버를 통해 Effect PostgreSQL 연결을 네이티브로 지원합니다.

Basic file structure

This is the basic file structure of the project. In the src/db directory, we have table definition in schema.ts. In drizzle folder there are sql migration file and snapshots.

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

Step 1 - node-postgres 패키지 설치

npm
yarn
pnpm
bun
npm i drizzle-orm effect @effect/sql-pg pg dotenv
npm i -D drizzle-kit tsx @types/pg

Step 2 - 연결 변수 설정

Create a .env file in the root of your project and add your database connection variable:

DATABASE_URL=

PostgreSQL 데이터베이스가 없고 테스트용으로 생성하려면, Docker에서 PostgreSQL 설정 가이드를 사용할 수 있습니다.

Docker에서 PostgreSQL 설정 가이드는 여기에서 확인할 수 있습니다. 가이드를 따라 설정하고 데이터베이스 URL을 생성한 후 다음 단계로 돌아오세요.

Step 3 - 데이터베이스에 Drizzle ORM 연결

Create a index.ts file in the src directory and initialize the connection:

import 'dotenv/config';
import { drizzle } from 'drizzle-orm/effect-postgres';
import { Effect, Redacted } from 'effect';
import { PgClient } from '@effect/sql-pg'
import { types } from 'pg';

const clientLayer = PgClient.layer({
    url: Redacted.make(process.env.DATABASE_URL!);
	  types: {
		  getTypeParser: (typeId, format) => {
			  if ([1184, 1114, 1082, 1186, 1231, 1115, 1185, 1187, 1182].includes(typeId)) { 
          return (val: any) => val;
        }
			  return types.getTypeParser(typeId, format);
		  },
	  },
});

const program = Effect.gen(function*() {
	const client = yield* PgClient.PgClient;
	const db = drizzle(client, { logger: true, /*...*/ });
});

await Effect.runPromise(program).pipe(Effect.provide(clientLayer));

Step 4 - 테이블 생성

Create a schema.ts file in the src/db directory and declare your table:

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(),
});

Step 5 - Drizzle 설정 파일 구성

Drizzle config - a configuration file that is used by Drizzle Kit and contains all the information about your database connection, migration folder and schema files.

Create a drizzle.config.ts file in the root of your project and add the following content:

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!,
  },
});

Step 6 - 데이터베이스에 변경사항 적용

You can directly apply changes to your database using the drizzle-kit push command. This is a convenient method for quickly testing new schema designs or modifications in a local development environment, allowing for rapid iterations without the need to manage migration files:

npx drizzle-kit push

Read more about the push command in documentation.

Tips

Alternatively, you can generate migrations using the drizzle-kit generate command and then apply them using the drizzle-kit migrate command:

Generate migrations:

npx drizzle-kit generate

Apply migrations:

npx drizzle-kit migrate

Read more about migration process in documentation.

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

src/index.ts 파일을 업데이트하여 사용자를 생성, 조회, 업데이트, 삭제하는 쿼리를 추가합니다.

import 'dotenv/config';
import { drizzle } from 'drizzle-orm/effect-postgres';
import { Effect, Redacted } from 'effect';
import { PgClient } from '@effect/sql-pg'
import { types } from 'pg';
import { usersTable } from './db/schema'

const clientLayer = PgClient.layer({
    url: Redacted.make(process.env.DATABASE_URL!);
	  types: {
	  	getTypeParser: (typeId, format) => {
	  		if ([1184, 1114, 1082, 1186, 1231, 1115, 1185, 1187, 1182].includes(typeId)) return (val: any) => val;
	  		return types.getTypeParser(typeId, format);
	  	},
	  },
});

const program = Effect.gen(function*() {
	const client = yield* PgClient.PgClient;
	const db = drizzle(client, { logger: true, /*...*/ });

  const user: typeof usersTable.$inferInsert = {
    name: 'John',
    age: 30,
    email: 'john@example.com',
  };

  yield* db.insert(usersTable).values(user);
  console.log('New user created!')

  const users = yield* db.select().from(usersTable);
  console.log('Getting all users from the database: ', users)
  /*
  const users: {
    id: number;
    name: string;
    age: number;
    email: string;
  }[]
  */

  yield* db
    .update(usersTable)
    .set({
      age: 31,
    })
    .where(eq(usersTable.email, user.email));
  console.log('User info updated!')

  yield* db.delete(usersTable).where(eq(usersTable.email, user.email));
  console.log('User deleted!')
});

await Effect.runPromise(program).pipe(Effect.provide(clientLayer));

Step 8 - index.ts 파일 실행

To run any TypeScript files, you have several options, but let’s stick with one: using tsx

You’ve already installed tsx, so we can run our queries now

Run index.ts script

npm
yarn
pnpm
bun
npx tsx src/index.ts
tips

We suggest using bun to run TypeScript files. With bun, such scripts can be executed without issues or additional settings, regardless of whether your project is configured with CommonJS (CJS), ECMAScript Modules (ESM), or any other module format. To run a script with bun, use the following command:

bun src/index.ts

If you don’t have bun installed, check the Bun installation docs