内置数据库支持,使用 SQL 模板字面量。
server.ts
import { defineHandler } from "nitro/h3";
import { useDatabase } from "nitro/database";
export default defineHandler(async () => {
const db = useDatabase();
// Create users table
await db.sql`DROP TABLE IF EXISTS users`;
await db.sql`CREATE TABLE IF NOT EXISTS users ("id" TEXT PRIMARY KEY, "firstName" TEXT, "lastName" TEXT, "email" TEXT)`;
// Add a new user
const userId = String(Math.round(Math.random() * 10_000));
await db.sql`INSERT INTO users VALUES (${userId}, 'John', 'Doe', '')`;
// Query for users
const { rows } = await db.sql`SELECT * FROM users WHERE id = ${userId}`;
return {
rows,
};
});
Nitro 提供了一个内置的数据库层,使用 SQL 模板字面量进行安全的参数化查询。此示例创建了一个 users 表,插入一条记录,并查询该记录。
查询数据库
server.ts
import { defineHandler } from "nitro/h3";
import { useDatabase } from "nitro/database";
export default defineHandler(async () => {
const db = useDatabase();
// 创建 users 表
await db.sql`DROP TABLE IF EXISTS users`;
await db.sql`CREATE TABLE IF NOT EXISTS users ("id" TEXT PRIMARY KEY, "firstName" TEXT, "lastName" TEXT, "email" TEXT)`;
// 添加一个新用户
const userId = String(Math.round(Math.random() * 10_000));
await db.sql`INSERT INTO users VALUES (${userId}, 'John', 'Doe', '')`;
// 查询用户
const { rows } = await db.sql`SELECT * FROM users WHERE id = ${userId}`;
return {
rows,
};
});
使用 useDatabase() 获取数据库实例。通过 db.sql 进行数据库查询,像 ${userId} 这样的变量会自动转义,以防止 SQL 注入。
使用任务运行迁移
Nitro 任务允许你在请求处理程序之外运行操作。对于数据库迁移,在 tasks/ 目录下创建任务文件,然后通过 CLI 运行。这样可以将模式变更与应用代码分离。
tasks/db/migrate.ts
import { defineTask } from "nitro/task";
import { useDatabase } from "nitro/database";
export default defineTask({
meta: {
description: "运行数据库迁移",
},
async run() {
const db = useDatabase();
console.log("正在运行数据库迁移...");
// 创建 users 表
await db.sql`DROP TABLE IF EXISTS users`;
await db.sql`CREATE TABLE IF NOT EXISTS users ("id" TEXT PRIMARY KEY, "firstName" TEXT, "lastName" TEXT, "email" TEXT)`;
return {
result: "数据库迁移完成!",
};
},
});