deno.com
在当前页面

使用 Drizzle ORM 和 Deno 构建数据库应用

Drizzle ORM 是一个 TypeScript ORM,它提供了一种类型安全的方式来与数据库交互。在本教程中,我们将使用 Drizzle ORM 和 Deno 以及 PostgreSQL 来创建、读取、更新和删除恐龙数据:

你可以在 这个 GitHub 仓库 中找到本教程的所有代码。

安装 Drizzle Jump to heading

首先,我们将使用 Deno 的 npm 兼容性安装所需的依赖项。我们将使用 Drizzle 与 Postgres,但你也可以使用 MySQLSQLite。(如果你没有安装 Postgres,可以 在这里安装。)

deno install npm:drizzle-orm npm:drizzle-kit npm:pg npm:@types/pg

这将安装 Drizzle ORM 及其相关工具 —— drizzle-kit 用于模式迁移,pg 用于 PostgreSQL 连接,以及 PostgreSQL 的 TypeScript 类型。这些包将使我们能够以类型安全的方式与数据库交互,同时保持与 Deno 运行时环境的兼容性。

它还将在你的项目根目录中创建一个 deno.json 文件来管理 npm 依赖项:

{
  "imports": {
    "@types/pg": "npm:@types/pg@^8.11.10",
    "drizzle-kit": "npm:drizzle-kit@^0.27.2",
    "drizzle-orm": "npm:drizzle-orm@^0.36.0",
    "pg": "npm:pg@^8.13.1"
  }
}

配置 Drizzle Jump to heading

接下来,让我们在项目根目录中创建一个 drizzle.config.ts 文件。该文件将配置 Drizzle 以与你的 PostgreSQL 数据库一起工作:

import { defineConfig } from "drizzle-kit";

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

这些配置设置决定了:

  • 迁移文件的输出位置(./drizzle
  • 模式定义的位置(./src/db/schema.ts
  • PostgreSQL 作为你的数据库方言,以及
  • 如何使用存储在环境变量中的 URL 连接到你的数据库

drizzle-kit 将使用此配置来管理你的数据库模式并自动生成 SQL 迁移。

我们还需要在项目根目录中创建一个包含 DATABASE_URL 连接字符串的 .env 文件:

DATABASE_URL=postgresql://[user[:password]@][host][:port]/[dbname]

请确保将登录凭据替换为你的。

接下来,让我们连接到数据库并使用 Drizzle 填充我们的表。

定义模式 Jump to heading

有两种方法可以使用 Drizzle 定义你的表模式。如果你已经定义了 Postgres 表,可以使用 pull 推断它们;否则,你可以在代码中定义它们,然后使用 Drizzle 创建一个新表。我们将在下面探讨这两种方法。

使用 pull 推断模式 Jump to heading

如果你在添加 Drizzle 之前已经定义了 Postgres 表,那么你可以使用 npm:drizzle-kit pull 命令自动生成 TypeScript 类型和表定义。这在处理现有数据库或希望确保代码与数据库结构保持同步时特别有用。

假设我们当前的数据库已经具有以下表模式:

Postgres 中的表模式图

我们将运行以下命令来推断数据库并填充 ./drizzle 目录下的多个文件:

deno --env -A --node-modules-dir npm:drizzle-kit pull

Failed to find Response internal state key
No config path provided, using default 'drizzle.config.ts'
Reading config file '/private/tmp/deno-drizzle-example/drizzle.config.ts'
Pulling from ['public'] list of schemas

Using 'pg' driver for database querying
[] 2 tables fetched
[] 8 columns fetched
[] 0 enums fetched
[] 0 indexes fetched
[] 1 foreign keys fetched
[] 0 policies fetched
[] 0 check constraints fetched
[] 0 views fetched

[i] No SQL generated, you already have migrations in project
[] You schema file is ready ➜ drizzle/schema.ts 🚀
[] You relations file is ready ➜ drizzle/relations.ts 🚀
我们使用 --env 标志来读取包含数据库 URL 的 .env 文件,并使用 --node-modules-dir 标志创建一个 node_modules 文件夹,以便正确使用 drizzle-kit

上述命令将在 ./drizzle 目录中创建多个文件,这些文件定义了模式、跟踪更改并提供数据库迁移所需的信息:

  • drizzle/schema.ts:该文件使用 Drizzle ORM 的模式定义语法定义数据库模式。
  • drizzle/relations.ts:该文件旨在使用 Drizzle ORM 的关系 API 定义表之间的关系。
  • drizzle/0000_long_veda.sql:一个 SQL 迁移文件,包含创建数据库表的 SQL 代码。代码被注释掉了 —— 如果你希望在新的环境中运行此迁移以创建表,可以取消注释此代码。
  • drizzle/meta/0000_snapshot.json:一个快照文件,表示当前数据库模式的状态。
  • drizzle/meta/_journal.json:该文件跟踪已应用于数据库的迁移。它帮助 Drizzle ORM 知道哪些迁移已经运行,哪些迁移仍需应用。

首先在 Drizzle 中定义模式 Jump to heading

如果你还没有在 Postgres 中定义现有表(例如,你正在启动一个全新的项目),你可以在代码中定义表和类型,并让 Drizzle 创建它们。

让我们创建一个新目录 ./src/db/,并在其中创建一个 schema.ts 文件,我们将用以下内容填充它:

// schema.ts
import {
  boolean,
  foreignKey,
  integer,
  pgTable,
  serial,
  text,
  timestamp,
} from "drizzle-orm/pg-core";

export const dinosaurs = pgTable("dinosaurs", {
  id: serial().primaryKey().notNull(),
  name: text(),
  description: text(),
});

export const tasks = pgTable("tasks", {
  id: serial().primaryKey().notNull(),
  dinosaurId: integer("dinosaur_id"),
  description: text(),
  dateCreated: timestamp("date_created", { mode: "string" }).defaultNow(),
  isComplete: boolean("is_complete"),
}, (table) => {
  return {
    tasksDinosaurIdFkey: foreignKey({
      columns: [table.dinosaurId],
      foreignColumns: [dinosaurs.id],
      name: "tasks_dinosaur_id_fkey",
    }),
  };
});
上述代码表示两个表 dinosaurstasks 及其关系。了解更多关于使用 Drizzle 定义模式及其关系的信息

一旦我们定义了 ./src/db/schema.ts,我们就可以通过创建迁移来创建表及其指定的关系:

deno -A --node-modules-dir npm:drizzle-kit generate

Failed to find Response internal state key
No config path provided, using default 'drizzle.config.ts'
Reading config file '/private/tmp/drizzle/drizzle.config.ts'
2 tables
dinosaurs 3 columns 0 indexes 0 fks
tasks 5 columns 0 indexes 1 fks

上述命令将创建一个包含迁移脚本和日志的 ./drizzle/ 文件夹。

与数据库交互 Jump to heading

现在我们已经设置了 Drizzle ORM,我们可以使用它来简化 Postgres 数据库中的数据管理。首先,Drizzle 建议将 schema.tsrelations.ts 复制到 ./src/db 目录中,以便在应用程序中使用。

让我们创建一个 ./src/db/db.ts,它导出一些辅助函数,使我们更容易与数据库交互:

import { drizzle } from "drizzle-orm/node-postgres";
import { dinosaurs as dinosaurSchema, tasks as taskSchema } from "./schema.ts";
import { dinosaursRelations, tasksRelations } from "./relations.ts";
import pg from "pg";
import { integer } from "drizzle-orm/sqlite-core";
import { eq } from "drizzle-orm/expressions";

// 使用 pg 驱动。
const { Pool } = pg;

// 使用 pg 驱动和模式实例化 Drizzle 客户端。
export const db = drizzle({
  client: new Pool({
    connectionString: Deno.env.get("DATABASE_URL"),
  }),
  schema: { dinosaurSchema, taskSchema, dinosaursRelations, tasksRelations },
});

// 插入恐龙。
export async function insertDinosaur(dinosaurObj: typeof dinosaurSchema) {
  return await db.insert(dinosaurSchema).values(dinosaurObj);
}

// 插入任务。
export async function insertTask(taskObj: typeof taskSchema) {
  return await db.insert(taskSchema).values(taskObj);
}

// 按 id 查找恐龙。
export async function findDinosaurById(dinosaurId: typeof integer) {
  return await db.select().from(dinosaurSchema).where(
    eq(dinosaurSchema.id, dinosaurId),
  );
}

// 按名称查找恐龙。
export async function findDinosaurByName(name: string) {
  return await db.select().from(dinosaurSchema).where(
    eq(dinosaurSchema.name, name),
  );
}

// 根据恐龙 id 查找任务。
export async function findDinosaurTasksByDinosaurId(
  dinosaurId: typeof integer,
) {
  return await db.select().from(taskSchema).where(
    eq(taskSchema.dinosaurId, dinosaurId),
  );
}

// 更新恐龙。
export async function updateDinosaur(dinosaurObj: typeof dinosaurSchema) {
  return await db.update(dinosaurSchema).set(dinosaurObj).where(
    eq(dinosaurSchema.id, dinosaurObj.id),
  );
}

// 更新任务。
export async function updateTask(taskObj: typeof taskSchema) {
  return await db.update(taskSchema).set(taskObj).where(
    eq(taskSchema.id, taskObj.id),
  );
}

// 按 id 删除恐龙。
export async function deleteDinosaurById(id: typeof integer) {
  return await db.delete(dinosaurSchema).where(
    eq(dinosaurSchema.id, id),
  );
}

// 按 id 删除任务。
export async function deleteTask(id: typeof integer) {
  return await db.delete(taskSchema).where(eq(taskSchema.id, id));
}

现在我们可以将这些辅助函数导入到一个脚本中,在该脚本中我们可以对数据库执行一些简单的 CRUD 操作。让我们创建一个新文件 ./src/script.ts

import {
  deleteDinosaurById,
  findDinosaurByName,
  insertDinosaur,
  insertTask,
  updateDinosaur,
} from "./db/db.ts";

// 创建一个新的恐龙。
await insertDinosaur({
  name: "Denosaur",
  description: "Dinosaurs should be simple.",
});

// 按名称查找该恐龙。
const res = await findDinosaurByName("Denosaur");

// 使用该恐龙的 id 创建一个任务。
await insertTask({
  dinosaurId: res.id,
  description: "Remove unnecessary config.",
  isComplete: false,
});

// 使用新的描述更新恐龙。
const newDeno = {
  id: res.id,
  name: "Denosaur",
  description: "The simplest dinosaur.",
};
await updateDinosaur(newDeno);

// 删除恐龙(及其所有任务)。
await deleteDinosaurById(res.id);

我们可以运行它,它将对数据库执行所有操作:

deno -A --env ./src/script.ts

下一步是什么? Jump to heading

Drizzle ORM 是一个流行的数据映射工具,用于简化和维护数据模型以及与数据库的交互。希望本教程能让你了解如何在 Deno 项目中使用 Drizzle。

现在你已经基本了解了如何在 Deno 中使用 Drizzle ORM,你可以:

  1. 添加更复杂的数据库关系
  2. 使用 Hono 实现 REST API 来提供恐龙数据
  3. 为数据库操作添加验证和错误处理
  4. 为数据库交互编写测试
  5. 将你的应用程序部署到云端

🦕 祝你在使用 Deno 和 Drizzle ORM 时编码愉快!这种堆栈的类型安全性和简单性使其成为构建现代 Web 应用程序的绝佳选择。

你找到需要的内容了吗?

隐私政策