MySQL

新数据库

了解 Drizzle

入门

Drizzle 和 MySQL 入门

本指南假定您熟悉

dotenv - 用于管理环境变量的包 - 点击此处阅读
tsx - 用于运行 TypeScript 文件的包 - 点击此处阅读
mysql2 - 用于查询 MySQL 数据库的包 - 在此处阅读

要将 Drizzle 与 MySQL 数据库一起使用，您应该使用 mysql2 驱动。

根据官方网站，mysql2 是一个专注于性能的 Node.js MySQL 客户端。

Drizzle ORM 原生支持 mysql2，通过 drizzle-orm/mysql2 包。

基本文件结构

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

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

第 1 步 - 安装 mysql2 包

npm

yarn

pnpm

bun

npm i drizzle-orm mysql2 dotenv
npm i -D drizzle-kit tsx

yarn add drizzle-orm mysql2 dotenv
yarn add -D drizzle-kit tsx

pnpm add drizzle-orm mysql2 dotenv
pnpm add -D drizzle-kit tsx

bun add drizzle-orm mysql2 dotenv
bun add -D drizzle-kit tsx

步骤 2 - 设置连接变量

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

DATABASE_URL=

提示

如果您还没有 MySQL 数据库并想创建一个用于测试，可以使用我们关于如何在 Docker 中设置 MySQL 的指南。

Docker 中的 MySQL 指南可在此处获取。请按照说明设置它，生成数据库 URL（指南中有解释），然后回来进行后续步骤。

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

在 src/db 目录下创建 index.ts 文件并初始化连接

mysql2

带配置的 mysql2

您的 mysql2 驱动

import 'dotenv/config';
import { drizzle } from "drizzle-orm/mysql2";

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

import 'dotenv/config';
import { drizzle } from "drizzle-orm/mysql2";

// You can specify any property from the mysql2 connection options
const db = drizzle({ connection: { uri: process.env.DATABASE_URL }});

import 'dotenv/config';
import { drizzle } from "drizzle-orm/mysql2";
import mysql from "mysql2/promise";
  
const poolConnection = mysql.createPool({
  host: "host",
  user: "user",
  database: "database",
});
const db = drizzle({ client: poolConnection });

// or if you need client connection
async function main() {
  const connection = await mysql.createConnection({
    host: "host",
    user: "user",
    database: "database",
  });
  const db = drizzle({ client: connection });
}
main();

重要提示

对于内置的 migrate 函数和 DDL 迁移，我们和驱动程序强烈建议您使用单个 client 连接。

出于查询目的，您可以根据业务需求自由选择使用 client 或 pool。

第 4 步 - 创建表

在 src/db 目录中创建 schema.ts 文件并声明你的表

import { int, mysqlTable, serial, varchar } from 'drizzle-orm/mysql-core';

export const usersTable = mysqlTable('users_table', {
  id: serial().primaryKey(),
  name: varchar({ length: 255 }).notNull(),
  age: int().notNull(),
  email: varchar({ length: 255 }).notNull().unique(),
});

第 5 步 - 设置 Drizzle 配置文件

Drizzle config - 由 Drizzle Kit 使用的配置文件，包含有关你的数据库连接、迁移文件夹和 schema 文件的所有信息。

在项目的根目录中创建 drizzle.config.ts 文件并添加以下内容

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

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

第 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/mysql2';
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: '[email protected]',
  };

  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 文件

要运行任何 TypeScript 文件，你有多种选择，但我们只使用一种：使用 tsx

你已经安装了 tsx，所以我们现在可以运行查询了

运行 index.ts 脚本

npm

yarn

pnpm

bun

npx tsx src/index.ts

yarn tsx src/index.ts

pnpm tsx src/index.ts

bun tsx src/index.ts

提示

我们建议使用 bun 来运行 TypeScript 文件。使用 bun，无论您的项目是配置为 CommonJS (CJS)、ECMAScript Modules (ESM) 还是任何其他模块格式，此类脚本都可以顺利执行，无需额外设置。要使用 bun 运行脚本，请使用以下命令：

bun src/index.ts

如果你没有安装 bun，请查看 Bun 安装文档