独立开发省钱指南：Supabase 一个数据库给多个产品使用

// drizzle.config.ts
export default defineConfig({
  // ... 其他配置
  schemaFilter: ['new-project-schema'], // <--- 关键：只管理这个 schema，名称自定义
});

// lib/db/schema.ts
import { pgSchema } from 'drizzle-orm/pg-core';

// 1. 定义 Schema
export const newProjectSchema = pgSchema('new-project-schema'); // <--- 名称自定义，和 drizzle.config.ts 文件里的一致

// 2. 将原来的 pgTable 替换为 newProjectSchema.table
// 旧代码: export const user = pgTable('user', { ... })
// 新代码:
export const user = newProjectSchema.table('user', {
  id: uuid('id').primaryKey().defaultRandom(),
  // ...
});

// 3. 枚举也需要迁移
// 旧代码: export const roleEnum = pgEnum('role', ['admin', 'user'])
// 新代码:
export const roleEnum = newProjectSchema.enum('role', ['admin', 'user']);

import { drizzle } from 'drizzle-orm/postgres-js';
import postgres from 'postgres';
import * as schema from '../lib/db/schema';
import * as dotenv from 'dotenv';
import { getTableConfig } from 'drizzle-orm/pg-core';

dotenv.config();

// 在 .env 配置好旧库和新库的连接串
const { OLD_DB_URL, NEW_DB_URL } = process.env;

// 定义迁移顺序（注意外键依赖：先迁移主表，再迁移子表）
const tables = [
  { name: 'user', table: schema.user }, // 被依赖的基础表
  { name: 'pricing_plans', table: schema.pricingPlans },
  { name: 'orders', table: schema.orders }, // 依赖 user 和 pricing_plans
  // ... 其他表
];

async function main() {
  const sourceClient = postgres(OLD_DB_URL!, { max: 1 });
  const destClient = postgres(NEW_DB_URL!, { max: 1 });
  const destDb = drizzle(destClient, { schema });

  for (const t of tables) {
    console.log(`正在迁移表: ${t.name}...`);

    // 1. 构建字段映射 (db_column -> schema_property)
    const columnMapping: Record<string, string> = {};
    const tableColumns = t.table; 
    for (const key in tableColumns) {
      // @ts-ignore
      const col = tableColumns[key];
      // 这里的逻辑是：Drizzle Schema 对象里的 key 通常就是驼峰属性名 (cardTitle)
      // 而 col.name 是数据库实际列名 (card_title)
      if (col && typeof col === 'object' && 'name' in col) {
          columnMapping[col.name] = key;
      }
    }

    // 2. 从旧库查数据 (假设旧库在 public schema)
    const rows = await sourceClient`SELECT * FROM public.${sourceClient(t.name)}`;
    
    if (rows.length === 0) continue;

    // 3. 转换数据格式 (snake_case -> camelCase)
    const transformRow = (row: any) => {
      const newRow: any = {};
      for (const dbColName in row) {
          const schemaKey = columnMapping[dbColName];
          // 如果 Schema 里定义了这个字段，就用 Schema 的 key
          // 否则保持原样（可能是未定义在 schema 里的字段）
          newRow[schemaKey || dbColName] = row[dbColName];
      }
      return newRow;
    };

    // 4. 批量写入新库
    const batchSize = 100;
    for (let i = 0; i < rows.length; i += batchSize) {
      const batch = rows.slice(i, i + batchSize).map(transformRow);
      await destDb.insert(t.table).values(batch).onConflictDoNothing();
    }
  }

  await sourceClient.end();
  await destClient.end();
  console.log('迁移完成！');
}

main();

DATABASE_URL=new-project-database-url