HaHafeng
61cdc97eeb
Summary: - Fix pg-boss queue conflict (duplicate key violation on queue_pkey) - Add global error listener to prevent process crash - Reduce connection pool from 10 to 4 - Add graceful shutdown handling (SIGTERM/SIGINT) - Fix researchWorker recursive call bug in catch block - Make screeningWorker idempotent using upsert Security Standards (v1.1): - Prohibit recursive retry in Worker catch blocks - Prohibit payload bloat (only store fileKey/ID in job.data) - Require Worker idempotency (upsert + unique constraint) - Recommend task-specific expireInSeconds settings - Document graceful shutdown pattern New Features: - PKB signed URL endpoint for document preview/download - pg_bigm installation guide for Docker - Dockerfile.postgres-with-extensions for pgvector + pg_bigm Documentation: - Update Postgres-Only async task processing guide (v1.1) - Add troubleshooting SQL queries - Update safety checklist Tested: Local verification passed
129 lines
3.7 KiB
TypeScript
129 lines
3.7 KiB
TypeScript
import { PrismaClient } from '@prisma/client';
|
||
|
||
const prisma = new PrismaClient();
|
||
|
||
async function main() {
|
||
console.log('🔍 数据库差异分析\n');
|
||
console.log('=' .repeat(60));
|
||
|
||
// 备份文件(2025-12-24)中应该存在的表
|
||
const backupTables = [
|
||
// aia_schema
|
||
'aia_schema.conversations',
|
||
'aia_schema.general_conversations',
|
||
'aia_schema.general_messages',
|
||
'aia_schema.messages',
|
||
'aia_schema.projects',
|
||
// asl_schema
|
||
'asl_schema.fulltext_screening_results',
|
||
'asl_schema.fulltext_screening_tasks',
|
||
'asl_schema.literatures',
|
||
'asl_schema.screening_projects',
|
||
'asl_schema.screening_results',
|
||
'asl_schema.screening_tasks',
|
||
// dc_schema
|
||
'dc_schema.dc_extraction_items',
|
||
'dc_schema.dc_extraction_tasks',
|
||
'dc_schema.dc_health_checks',
|
||
'dc_schema.dc_templates',
|
||
'dc_schema.dc_tool_c_ai_history',
|
||
'dc_schema.dc_tool_c_sessions',
|
||
// pkb_schema
|
||
'pkb_schema.batch_results',
|
||
'pkb_schema.batch_tasks',
|
||
'pkb_schema.documents',
|
||
'pkb_schema.knowledge_bases',
|
||
'pkb_schema.task_templates',
|
||
// platform_schema
|
||
'platform_schema.app_cache',
|
||
'platform_schema.job',
|
||
'platform_schema.job_common', // 可能缺失
|
||
'platform_schema.queue',
|
||
'platform_schema.schedule',
|
||
'platform_schema.subscription',
|
||
'platform_schema.users',
|
||
'platform_schema.version',
|
||
// public
|
||
'public._prisma_migrations',
|
||
'public.admin_logs',
|
||
'public.review_tasks', // 可能被移动到 rvw_schema
|
||
'public.users',
|
||
];
|
||
|
||
console.log('\n📋 检查备份中的表是否在当前数据库中存在：\n');
|
||
|
||
for (const table of backupTables) {
|
||
const [schema, tableName] = table.split('.');
|
||
try {
|
||
const result: any = await prisma.$queryRawUnsafe(
|
||
`SELECT COUNT(*) as count FROM information_schema.tables
|
||
WHERE table_schema = '${schema}' AND table_name = '${tableName}'`
|
||
);
|
||
if (result[0].count === 0n) {
|
||
console.log(` ❌ ${table} - 不存在！`);
|
||
} else {
|
||
console.log(` ✅ ${table} - 存在`);
|
||
}
|
||
} catch (e: any) {
|
||
console.log(` ❌ ${table} - 查询失败: ${e.message}`);
|
||
}
|
||
}
|
||
|
||
// 检查 platform_schema.users 的列结构差异
|
||
console.log('\n\n📋 platform_schema.users 当前列结构：\n');
|
||
const cols: any[] = await prisma.$queryRaw`
|
||
SELECT column_name, data_type, is_nullable, column_default
|
||
FROM information_schema.columns
|
||
WHERE table_schema = 'platform_schema' AND table_name = 'users'
|
||
ORDER BY ordinal_position;
|
||
`;
|
||
|
||
cols.forEach(c => {
|
||
console.log(` ${c.column_name}: ${c.data_type} ${c.is_nullable === 'NO' ? 'NOT NULL' : 'NULLABLE'} ${c.column_default ? `DEFAULT ${c.column_default}` : ''}`);
|
||
});
|
||
|
||
// 备份中 platform_schema.users 应有的列
|
||
const originalUserColumns = ['id', 'email', 'password', 'name', 'avatar_url', 'role', 'status', 'kb_quota', 'kb_used', 'trial_ends_at', 'is_trial', 'last_login_at', 'created_at', 'updated_at'];
|
||
|
||
console.log('\n📋 对比 platform_schema.users 与备份：');
|
||
console.log(' 原始列(备份): ' + originalUserColumns.join(', '));
|
||
console.log(' 当前列: ' + cols.map(c => c.column_name).join(', '));
|
||
|
||
const currentColNames = cols.map(c => c.column_name);
|
||
const missingInCurrent = originalUserColumns.filter(c => !currentColNames.includes(c));
|
||
const newInCurrent = currentColNames.filter(c => !originalUserColumns.includes(c));
|
||
|
||
if (missingInCurrent.length > 0) {
|
||
console.log('\n ⚠️ 备份中有但当前缺失的列: ' + missingInCurrent.join(', '));
|
||
}
|
||
if (newInCurrent.length > 0) {
|
||
console.log(' ➕ 当前新增的列: ' + newInCurrent.join(', '));
|
||
}
|
||
|
||
console.log('\n' + '=' .repeat(60));
|
||
}
|
||
|
||
main()
|
||
.catch(console.error)
|
||
.finally(() => prisma.$disconnect());
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|