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

  • 安装 Drizzle
  • 配置 Drizzle
  • 定义模式
  • 与数据库交互
  • 下一步是什么?

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

安装 Drizzle

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

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

此操作安装 Drizzle ORM 及其相关工具——[drizzle-kit(https://orm.drizzle.team/docs/kit-overview)] 用于模式迁移、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

接下来,在您的项目根目录下创建一个 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 迁移。

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

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

请确保将登录凭证替换为您的。

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

定义模式

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

推断模式与 pull

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

假设我们当前的数据库已经包含以下表结构:

我们将运行以下命令来检查数据库并在 ./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 标志创建一个允许我们正确使用 drizzle-kitnode_modules 文件夹。

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

  • drizzle/schema.ts : 此文件使用 Drizzle ORM 的架构定义语法定义数据库架构。
  • drizzle/relations.ts : 此文件旨在使用 Drizzle ORM 的关系 API 定义表之间的关系。
  • drizzle/0000_long_veda.sql : 一个包含创建数据库表(s)的 SQL 代码的 SQL 迁移文件。代码已被注释掉——如果您想在新环境中运行此迁移以创建表(s),可以取消注释此代码。
  • drizzle/meta/0000_snapshot.json : 表示您数据库模式当前状态的快照文件。
  • drizzle/meta/_journal.json : 此文件跟踪已应用到您数据库中的迁移。它帮助 Drizzle ORM 了解已运行的迁移和尚未应用的迁移。

在 Drizzle 中定义模式

如果您还没有在 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 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/ 文件夹。

与数据库交互

现在我们已经设置了 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";

// Use pg driver.
const { Pool } = pg;

// Instantiate Drizzle client with pg driver and schema.
export const db = drizzle({
  client: new Pool({
    connectionString: Deno.env.get("DATABASE_URL"),
  }),
  schema: { dinosaurSchema, taskSchema, dinosaursRelations, tasksRelations },
});

// Insert dinosaur.
export async function insertDinosaur(dinosaurObj: typeof dinosaurSchema) {
  return await db.insert(dinosaurSchema).values(dinosaurObj);
}

// Insert task.
export async function insertTask(taskObj: typeof taskSchema) {
  return await db.insert(taskSchema).values(taskObj);
}

// Find dinosaur by id.
export async function findDinosaurById(dinosaurId: typeof integer) {
  return await db.select().from(dinosaurSchema).where(
    eq(dinosaurSchema.id, dinosaurId),
  );
}

// Find dinosaur by name.
export async function findDinosaurByName(name: string) {
  return await db.select().from(dinosaurSchema).where(
    eq(dinosaurSchema.name, name),
  );
}

// Find tasks based on dinosaur id.
export async function findDinosaurTasksByDinosaurId(
  dinosaurId: typeof integer,
) {
  return await db.select().from(taskSchema).where(
    eq(taskSchema.dinosaurId, dinosaurId),
  );
}

// Update dinosaur.
export async function updateDinosaur(dinosaurObj: typeof dinosaurSchema) {
  return await db.update(dinosaurSchema).set(dinosaurObj).where(
    eq(dinosaurSchema.id, dinosaurObj.id),
  );
}

// Update task.
export async function updateTask(taskObj: typeof taskSchema) {
  return await db.update(taskSchema).set(taskObj).where(
    eq(taskSchema.id, taskObj.id),
  );
}

// Delete dinosaur by id.
export async function deleteDinosaurById(id: typeof integer) {
  return await db.delete(dinosaurSchema).where(
    eq(dinosaurSchema.id, id),
  );
}

// Delete task by 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";

// Create a new dinosaur.
await insertDinosaur({
  name: "Denosaur",
  description: "Dinosaurs should be simple.",
});

// Find that dinosaur by name.
const res = await findDinosaurByName("Denosaur");

// Create a task with that dinosaur by its id.
await insertTask({
  dinosaurId: res.id,
  description: "Remove unnecessary config.",
  isComplete: false,
});

// Update a dinosaur with a new description.
const newDeno = {
  id: res.id,
  name: "Denosaur",
  description: "The simplest dinosaur.",
};
await updateDinosaur(newDeno);

// Delete the dinosaur (and any tasks it has).
await deleteDinosaurById(res.id);

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

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

下一步是什么?

Drizzle ORM 是一个流行的数据映射工具,用于简化管理和维护数据模型以及与数据库的交互。希望这篇教程能帮助你开始在 Deno 项目中使用 Drizzle。

现在您已经了解了如何使用 Drizzle ORM 与 Deno,您可以:

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

🦕 使用 Deno 和 Drizzle ORM 开发愉快!这个堆栈的类型安全和简洁性使其成为构建现代网络应用的绝佳选择。