如何在Drizzle（PostgreSQL）中删除并清空所有数据？

问题背景 #

我发现Drizzle官方文档里没有说明如何清空数据库所有数据，也没有对应的内置函数。尝试了一些代码变体但没生效，还遇到了pg_resetxlog函数不存在的错误（使用PostgreSQL 14.11）。

原代码的问题 #

之前尝试的代码存在两个关键问题：

• 先执行db.delete(posts)和db.delete(users)完全多余，TRUNCATE本身就是高效清空表数据的命令，且CASCADE会自动处理外键关联的表数据删除。
• pg_resetxlog是错误用法：这个函数在PostgreSQL 10之后已更名为pg_resetwal，而且它是离线工具，必须在数据库服务停止时通过命令行执行，用于修复数据库日志损坏，根本不是用来清空数据的，通过SQL调用必然报错。

正确实现方式 #

直接使用PostgreSQL的TRUNCATE TABLE ... CASCADE命令，结合Drizzle的事务批量执行即可，以下是修正后的代码：

async function deleteEverything() {
  try {
    const tableSchema = db._.schema;
    if (!tableSchema) throw new Error("未找到表结构");

    console.log("🗑️  正在清空整个数据库");

    // 生成所有表的TRUNCATE语句，CASCADE处理外键关联
    const queries = Object.values(tableSchema).map(table => 
      sql.raw(`TRUNCATE TABLE "${table.dbName}" CASCADE;`)
    );

    console.log("📨 正在执行清空操作...");

    // 事务中批量执行，保证原子性
    await db.transaction(async (tx) => {
      await Promise.all(queries.map(query => tx.execute(query)));
    });

    console.log("✅ 数据库已清空");
  } catch (e) {
    console.error("清空数据库失败:", e);
  }
}

关键说明 #

• TRUNCATE vs DELETE：TRUNCATE比DELETE高效得多，它直接重置表存储，而非逐行删除数据；CASCADE参数会自动清空所有与当前表有外键关联的表数据，无需手动处理表顺序。
• 表名加引号：用双引号包裹表名可避免PostgreSQL对大小写敏感的问题（若表名是驼峰命名）。
• 事务保证：所有清空操作放在事务中执行，确保要么全部成功，要么全部回滚，避免出现部分表清空的不一致状态。

关于pg_resetwal的补充 #

若确实需要重置WAL日志（通常仅数据库损坏时需用），请先停止PostgreSQL服务，再通过命令行执行：

pg_resetwal -D /path/to/your/postgres/data/directory

该操作非常危险，会丢失未提交事务，仅在数据库无法启动时使用，绝对不要用它清空数据。

内容的提问来源于stack exchange，提问作者psygo