pg

server/prisma/check-migration-dependencies.js

@@ -0,0 +1,175 @@
+const fs = require('fs');
+const path = require('path');
+
+/**
+ * 检查迁移文件的外键依赖关系
+ */
+
+// 存储表创建和引用信息的映射
+const tableCreationOrder = {};
+const foreignKeyReferences = {};
+
+/**
+ * 解析迁移文件中的表创建和外键引用
+ */
+function parseMigrationFile(filePath) {
+  try {
+    const sqlContent = fs.readFileSync(filePath, 'utf8');
+    const lines = sqlContent.split('\n');
+
+    let currentTable = null;
+    let inCreateTable = false;
+
+    lines.forEach((line, index) => {
+      const trimmedLine = line.trim();
+
+      // 检测 CREATE TABLE 语句
+      if (trimmedLine.startsWith('CREATE TABLE')) {
+        const tableMatch = trimmedLine.match(/CREATE TABLE "([^"]+)"/);
+        if (tableMatch) {
+          currentTable = tableMatch[1];
+          inCreateTable = true;
+          tableCreationOrder[currentTable] = path.basename(path.dirname(filePath));
+        }
+      }
+
+      // 检测外键约束
+      if (trimmedLine.includes('FOREIGN KEY') && trimmedLine.includes('REFERENCES')) {
+        const fkMatch = trimmedLine.match(/FOREIGN KEY \("[^"]+"\)\s+REFERENCES\s+"([^"]+)"/);
+        if (fkMatch && currentTable) {
+          const referencedTable = fkMatch[1];
+          if (!foreignKeyReferences[currentTable]) {
+            foreignKeyReferences[currentTable] = [];
+          }
+          foreignKeyReferences[currentTable].push({
+            referencedTable,
+            line: index + 1
+          });
+        }
+      }
+
+      // 检测表创建结束
+      if (inCreateTable && trimmedLine === ');') {
+        inCreateTable = false;
+        currentTable = null;
+      }
+    });
+
+    return true;
+  } catch (error) {
+    console.error(`解析文件失败: ${filePath}`, error.message);
+    return false;
+  }
+}
+
+/**
+ * 递归处理目录中的所有迁移文件
+ */
+function processMigrationsDirectory(migrationsDir) {
+  const items = fs.readdirSync(migrationsDir);
+
+  items.forEach(item => {
+    const itemPath = path.join(migrationsDir, item);
+    const stat = fs.statSync(itemPath);
+
+    if (stat.isDirectory() && item !== 'migration_lock.toml') {
+      // 递归处理子目录
+      processMigrationsDirectory(itemPath);
+    } else if (item === 'migration.sql') {
+      // 处理迁移文件
+      parseMigrationFile(itemPath);
+    }
+  });
+}
+
+/**
+ * 检查依赖关系
+ */
+function checkDependencies() {
+  console.log('=== 迁移文件外键依赖关系检查 ===\n');
+
+  const issues = [];
+
+  // 检查每个表的外键引用
+  Object.entries(foreignKeyReferences).forEach(([table, references]) => {
+    references.forEach(({ referencedTable, line }) => {
+      // 检查被引用的表是否在当前迁移或更早的迁移中创建
+      if (tableCreationOrder[referencedTable]) {
+        const currentMigration = tableCreationOrder[table];
+        const referencedMigration = tableCreationOrder[referencedTable];
+
+        // 如果被引用的表在更新的迁移中创建，这可能有问题
+        if (currentMigration < referencedMigration) {
+          issues.push({
+            table,
+            referencedTable,
+            issue: `表 ${table} 引用了在更晚迁移中创建的表 ${referencedTable}`,
+            migration: currentMigration
+          });
+        }
+      } else {
+        issues.push({
+          table,
+          referencedTable,
+          issue: `表 ${table} 引用了不存在的表 ${referencedTable}`,
+          migration: tableCreationOrder[table]
+        });
+      }
+    });
+  });
+
+  if (issues.length === 0) {
+    console.log('✅ 没有发现外键依赖关系问题');
+  } else {
+    console.log('❌ 发现以下外键依赖关系问题:');
+    issues.forEach(issue => {
+      console.log(`  表: ${issue.table}`);
+      console.log(`  引用表: ${issue.referencedTable}`);
+      console.log(`  问题: ${issue.issue}`);
+      console.log(`  迁移: ${issue.migration}`);
+      console.log('');
+    });
+  }
+
+  return issues;
+}
+
+/**
+ * 显示表创建顺序
+ */
+function showTableCreationOrder() {
+  console.log('\n=== 表创建顺序 ===');
+
+  const sortedTables = Object.entries(tableCreationOrder)
+    .sort(([,a], [,b]) => a.localeCompare(b))
+    .map(([table, migration]) => `  ${table} (${migration})`);
+
+  sortedTables.forEach(table => console.log(table));
+}
+
+/**
+ * 主函数
+ */
+function main() {
+  const migrationsDir = path.join(__dirname, 'migrations');
+
+  console.log('开始检查迁移文件外键依赖关系...\n');
+
+  processMigrationsDirectory(migrationsDir);
+
+  showTableCreationOrder();
+  const issues = checkDependencies();
+
+  return {
+    tableCreationOrder,
+    foreignKeyReferences,
+    issues: issues.length
+  };
+}
+
+// 运行检查
+if (require.main === module) {
+  main();
+}
+
+module.exports = { parseMigrationFile, processMigrationsDirectory, checkDependencies };

server/prisma/migrations/20230921191814_init/migration.sql

@@ -1,4 +1,4 @@
--- CreateTable
+-- CreateTable (基础表 - 无外键依赖)
 CREATE TABLE "api_keys" (
     "id" SERIAL PRIMARY KEY,
     "secret" TEXT,
@@ -7,30 +7,6 @@ CREATE TABLE "api_keys" (
     "lastUpdatedAt" TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
 );
 
--- CreateTable
-CREATE TABLE "workspace_documents" (
-    "id" SERIAL PRIMARY KEY,
-    "docId" TEXT NOT NULL,
-    "filename" TEXT NOT NULL,
-    "docpath" TEXT NOT NULL,
-    "workspaceId" INTEGER NOT NULL,
-    "metadata" TEXT,
-    "createdAt" TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "lastUpdatedAt" TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    CONSTRAINT "workspace_documents_workspaceId_fkey" FOREIGN KEY ("workspaceId") REFERENCES "workspaces" ("id") ON DELETE RESTRICT ON UPDATE CASCADE
-);
-
--- CreateTable
-CREATE TABLE "invites" (
-    "id" SERIAL PRIMARY KEY,
-    "code" TEXT NOT NULL,
-    "status" TEXT NOT NULL DEFAULT 'pending',
-    "claimedBy" INTEGER,
-    "createdAt" TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "createdBy" INTEGER NOT NULL,
-    "lastUpdatedAt" TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
-);
-
 -- CreateTable
 CREATE TABLE "system_settings" (
     "id" SERIAL PRIMARY KEY,
@@ -52,11 +28,26 @@ CREATE TABLE "users" (
 );
 
 -- CreateTable
-CREATE TABLE "document_vectors" (
+CREATE TABLE "workspaces" (
     "id" SERIAL PRIMARY KEY,
-    "docId" TEXT NOT NULL,
-    "vectorId" TEXT NOT NULL,
+    "name" TEXT NOT NULL,
+    "slug" TEXT NOT NULL,
+    "vectorTag" TEXT,
+    "createdAt" TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "openAiTemp" REAL,
+    "openAiHistory" INTEGER NOT NULL DEFAULT 20,
+    "lastUpdatedAt" TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "openAiPrompt" TEXT
+);
+
+-- CreateTable
+CREATE TABLE "invites" (
+    "id" SERIAL PRIMARY KEY,
+    "code" TEXT NOT NULL,
+    "status" TEXT NOT NULL DEFAULT 'pending',
+    "claimedBy" INTEGER,
     "createdAt" TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "createdBy" INTEGER NOT NULL,
     "lastUpdatedAt" TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
 );
 
@@ -70,19 +61,28 @@ CREATE TABLE "welcome_messages" (
 );
 
 -- CreateTable
-CREATE TABLE "workspaces" (
+CREATE TABLE "document_vectors" (
     "id" SERIAL PRIMARY KEY,
-    "name" TEXT NOT NULL,
-    "slug" TEXT NOT NULL,
-    "vectorTag" TEXT,
+    "docId" TEXT NOT NULL,
+    "vectorId" TEXT NOT NULL,
+    "createdAt" TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "lastUpdatedAt" TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
+);
+
+-- CreateTable (有外键依赖的表 - 依赖 workspaces)
+CREATE TABLE "workspace_documents" (
+    "id" SERIAL PRIMARY KEY,
+    "docId" TEXT NOT NULL,
+    "filename" TEXT NOT NULL,
+    "docpath" TEXT NOT NULL,
+    "workspaceId" INTEGER NOT NULL,
+    "metadata" TEXT,
     "createdAt" TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "openAiTemp" REAL,
-    "openAiHistory" INTEGER NOT NULL DEFAULT 20,
     "lastUpdatedAt" TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "openAiPrompt" TEXT
+    CONSTRAINT "workspace_documents_workspaceId_fkey" FOREIGN KEY ("workspaceId") REFERENCES "workspaces" ("id") ON DELETE RESTRICT ON UPDATE CASCADE
 );
 
--- CreateTable
+-- CreateTable (有外键依赖的表 - 依赖 workspaces 和 users)
 CREATE TABLE "workspace_chats" (
     "id" SERIAL PRIMARY KEY,
     "workspaceId" INTEGER NOT NULL,
@@ -95,7 +95,7 @@ CREATE TABLE "workspace_chats" (
     CONSTRAINT "workspace_chats_user_id_fkey" FOREIGN KEY ("user_id") REFERENCES "users" ("id") ON DELETE CASCADE ON UPDATE CASCADE
 );
 
--- CreateTable
+-- CreateTable (有外键依赖的表 - 依赖 workspaces 和 users)
 CREATE TABLE "workspace_users" (
     "id" SERIAL PRIMARY KEY,
     "user_id" INTEGER NOT NULL,
@@ -109,9 +109,6 @@ CREATE TABLE "workspace_users" (
 -- CreateIndex
 CREATE UNIQUE INDEX "api_keys_secret_key" ON "api_keys"("secret");
 
--- CreateIndex
-CREATE UNIQUE INDEX "workspace_documents_docId_key" ON "workspace_documents"("docId");
-
 -- CreateIndex
 CREATE UNIQUE INDEX "invites_code_key" ON "invites"("code");
 
@@ -123,3 +120,6 @@ CREATE UNIQUE INDEX "users_username_key" ON "users"("username");
 
 -- CreateIndex
 CREATE UNIQUE INDEX "workspaces_slug_key" ON "workspaces"("slug");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "workspace_documents_docId_key" ON "workspace_documents"("docId");