使用 Drizzle 和 Bun:SQLite 入门

本指南假定您已熟悉：

bun - JavaScript 一体化工具包 - 点击阅读
bun:sqlite - 高性能 SQLite3 驱动的原生实现 - 点击阅读

基本文件结构

这是项目的基本文件结构。在 src/db 目录中，我们在 schema.ts 中有表定义。在 drizzle 文件夹中，有 SQL 迁移文件和快照。

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

步骤 1 - 安装所需的包

npm

yarn

pnpm

bun

npm i drizzle-orm
npm i -D drizzle-kit @types/bun

yarn add drizzle-orm
yarn add -D drizzle-kit @types/bun

pnpm add drizzle-orm
pnpm add -D drizzle-kit @types/bun

bun add drizzle-orm
bun add -D drizzle-kit @types/bun

步骤 2 - 设置连接变量

在你的项目根目录下创建一个 .env 文件，并添加你的数据库连接变量：

DB_FILE_NAME=

重要

例如，如果您想在项目根目录中创建一个 SQLite 数据库文件以进行测试，可以使用以下示例：

DB_FILE_NAME=mydb.sqlite

步骤 3 - 将 Drizzle ORM 连接到数据库

在 src 目录下创建一个 index.ts 文件并初始化连接：

bun:sqlite

bun:sqlite 带配置

import 'dotenv/config';
import { drizzle } from 'drizzle-orm/bun-sqlite';

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

import 'dotenv/config';
import { drizzle } from 'drizzle-orm/bun-sqlite';

// 你可以指定 bun:sql 连接选项中的任何属性
const db = drizzle({ connection: { source: process.env.DB_FILE_NAME! }});

如果你需要提供已有的驱动：

import 'dotenv/config';
import { drizzle } from 'drizzle-orm/bun-sqlite';
import { Database } from 'bun:sqlite';

const sqlite = new Database(process.env.DB_FILE_NAME!);
const db = drizzle({ client: sqlite });

步骤 4 - 创建一个表

在 src/db 目录下创建一个 schema.ts 文件，并声明你的数据表：

import { int, sqliteTable, text } from "drizzle-orm/sqlite-core";

export const usersTable = sqliteTable("users_table", {
  id: int().primaryKey({ autoIncrement: true }),
  name: text().notNull(),
  age: int().notNull(),
  email: text().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: 'sqlite',
  dbCredentials: {
    url: process.env.DB_FILE_NAME!,
  },
});

步骤 6 - 将更改应用到数据库

您可以使用 drizzle-kit push 命令直接将更改应用到数据库中。这是一种方便的方法，适合在本地开发环境中快速测试新的架构设计或修改，允许快速迭代而无需管理迁移文件：

npx drizzle-kit push

在文档中了解更多关于 push 命令的信息。

小贴士

或者，您可以使用 drizzle-kit generate 命令生成迁移文件，然后使用 drizzle-kit migrate 命令应用这些迁移：

生成迁移：

npx drizzle-kit generate

应用迁移：

npx drizzle-kit migrate

在文档中了解更多关于迁移流程的信息。

步骤 7 - 生成并查询数据库

让我们更新 src/index.ts 文件，添加用于创建、读取、更新和删除用户的查询。

import 'dotenv/config';
import { drizzle } from 'drizzle-orm/bun-sqlite';
import { eq } from 'drizzle-orm';
import { usersTable } from './db/schema';
  
const db = drizzle(process.env.DB_FILE_NAME!);

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

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

  const users = await db.select().from(usersTable);
  console.log('Getting all users from the database: ', 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('User info updated!')

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

main();

步骤 8 - 运行 index.ts 文件

要使用 bun 运行脚本，请使用以下命令：

bun src/index.ts