如果您还没有 MySQL 数据库并想要创建一个用于测试,您可以使用我们关于如何在 Docker 中设置 MySQL 的指南。
Docker 中的 MySQL 指南可以在 这里 找到。去设置一下,生成一个数据库 URL(指南中有解释),然后回来进行下一步
要使用 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.jsonnpm i drizzle-orm mysql2 dotenv
npm i -D drizzle-kit tsx
在你的项目根目录下创建一个 .env 文件,并添加你的数据库连接变量:
DATABASE_URL=如果您还没有 MySQL 数据库并想要创建一个用于测试,您可以使用我们关于如何在 Docker 中设置 MySQL 的指南。
Docker 中的 MySQL 指南可以在 这里 找到。去设置一下,生成一个数据库 URL(指南中有解释),然后回来进行下一步
在 src/db 目录下创建一个 index.ts 文件并初始化连接:
import 'dotenv/config';
import { drizzle } from "drizzle-orm/mysql2";
const db = drizzle(process.env.DATABASE_URL);对于内置的带有 DDL 迁移的 migrate 功能,我们和驱动强烈建议你使用单一的 client 连接。
在查询时,可以根据你的业务需求自由选择使用 client 或 pool。
在 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(),
});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: 'mysql',
dbCredentials: {
url: process.env.DATABASE_URL!,
},
});您可以使用 drizzle-kit push 命令直接将更改应用到数据库中。这是一种方便的方法,适合在本地开发环境中快速测试新的架构设计或修改,允许快速迭代而无需管理迁移文件:
npx drizzle-kit push在 文档 中了解更多关于 push 命令的信息。
或者,您可以使用 drizzle-kit generate 命令生成迁移文件,然后使用 drizzle-kit migrate 命令应用这些迁移:
生成迁移:
npx drizzle-kit generate应用迁移:
npx drizzle-kit migrate在 文档 中了解更多关于迁移流程的信息。
让我们更新 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: '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();要运行任何 TypeScript 文件,你有几种选择,但我们这里先专注于一种:使用 tsx
你已经安装了 tsx,所以现在可以运行我们的查询了
运行 index.ts 脚本
npx tsx src/index.ts
我们建议使用 bun 来运行 TypeScript 文件。使用 bun,无论你的项目是配置为 CommonJS(CJS)、ECMAScript 模块(ESM)还是其他模块格式,这类脚本都可以顺利运行,无需额外设置或调整。
使用 bun 运行脚本,执行以下命令:
bun src/index.ts如果你还没有安装 bun,请查看 Bun 安装文档