280 lines
9.8 KiB
JavaScript
280 lines
9.8 KiB
JavaScript
const mysql = require('mysql2/promise');
|
||
|
||
// 数据库连接配置
|
||
const dbConfig = {
|
||
host: 'gz-cynosdbmysql-grp-d26pzce5.sql.tencentcdb.com',
|
||
port: 24936,
|
||
user: 'root',
|
||
password: '!Rjb12191',
|
||
database: 'pronode_db',
|
||
charset: 'utf8mb4'
|
||
};
|
||
|
||
// 创建表的 SQL 语句
|
||
const createTablesSQL = [
|
||
// 用户表
|
||
`CREATE TABLE IF NOT EXISTS users (
|
||
id VARCHAR(255) PRIMARY KEY,
|
||
email VARCHAR(255) UNIQUE NOT NULL,
|
||
name VARCHAR(255),
|
||
avatar VARCHAR(500),
|
||
createdAt TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||
updatedAt TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
||
INDEX idx_email (email),
|
||
INDEX idx_created_at (createdAt)
|
||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;`,
|
||
|
||
// 提示词模板表
|
||
`CREATE TABLE IF NOT EXISTS templates (
|
||
id VARCHAR(255) PRIMARY KEY,
|
||
title VARCHAR(255) NOT NULL,
|
||
description TEXT,
|
||
category VARCHAR(100) NOT NULL,
|
||
tags JSON,
|
||
role TEXT NOT NULL,
|
||
task TEXT NOT NULL,
|
||
context TEXT,
|
||
constraints JSON,
|
||
outputFormat VARCHAR(100) NOT NULL,
|
||
variables JSON,
|
||
examples JSON,
|
||
authorId VARCHAR(255) NOT NULL,
|
||
isPublic BOOLEAN DEFAULT TRUE,
|
||
isFeatured BOOLEAN DEFAULT FALSE,
|
||
usageCount INT DEFAULT 0,
|
||
rating DECIMAL(3,2) DEFAULT 0.00,
|
||
ratingCount INT DEFAULT 0,
|
||
createdAt TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||
updatedAt TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
||
FOREIGN KEY (authorId) REFERENCES users(id) ON DELETE CASCADE,
|
||
INDEX idx_category (category),
|
||
INDEX idx_author (authorId),
|
||
INDEX idx_public (isPublic),
|
||
INDEX idx_featured (isFeatured),
|
||
INDEX idx_rating (rating),
|
||
INDEX idx_usage (usageCount),
|
||
INDEX idx_created_at (createdAt)
|
||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;`,
|
||
|
||
// 收藏表
|
||
`CREATE TABLE IF NOT EXISTS favorites (
|
||
id VARCHAR(255) PRIMARY KEY,
|
||
userId VARCHAR(255) NOT NULL,
|
||
templateId VARCHAR(255) NOT NULL,
|
||
createdAt TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||
FOREIGN KEY (userId) REFERENCES users(id) ON DELETE CASCADE,
|
||
FOREIGN KEY (templateId) REFERENCES templates(id) ON DELETE CASCADE,
|
||
UNIQUE KEY unique_user_template (userId, templateId),
|
||
INDEX idx_user (userId),
|
||
INDEX idx_template (templateId),
|
||
INDEX idx_created_at (createdAt)
|
||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;`,
|
||
|
||
// 评论表
|
||
`CREATE TABLE IF NOT EXISTS comments (
|
||
id VARCHAR(255) PRIMARY KEY,
|
||
content TEXT NOT NULL,
|
||
rating INT CHECK (rating >= 1 AND rating <= 5),
|
||
userId VARCHAR(255) NOT NULL,
|
||
templateId VARCHAR(255) NOT NULL,
|
||
createdAt TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||
updatedAt TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
||
FOREIGN KEY (userId) REFERENCES users(id) ON DELETE CASCADE,
|
||
FOREIGN KEY (templateId) REFERENCES templates(id) ON DELETE CASCADE,
|
||
INDEX idx_user (userId),
|
||
INDEX idx_template (templateId),
|
||
INDEX idx_rating (rating),
|
||
INDEX idx_created_at (createdAt)
|
||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;`,
|
||
|
||
// 测试记录表
|
||
`CREATE TABLE IF NOT EXISTS tests (
|
||
id VARCHAR(255) PRIMARY KEY,
|
||
templateId VARCHAR(255) NOT NULL,
|
||
input JSON,
|
||
output TEXT,
|
||
model VARCHAR(100),
|
||
parameters JSON,
|
||
status ENUM('success', 'error', 'pending') DEFAULT 'pending',
|
||
duration INT,
|
||
createdAt TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||
FOREIGN KEY (templateId) REFERENCES templates(id) ON DELETE CASCADE,
|
||
INDEX idx_template (templateId),
|
||
INDEX idx_status (status),
|
||
INDEX idx_model (model),
|
||
INDEX idx_created_at (createdAt)
|
||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;`,
|
||
|
||
// 系统配置表
|
||
`CREATE TABLE IF NOT EXISTS system_configs (
|
||
id VARCHAR(255) PRIMARY KEY,
|
||
config_key VARCHAR(255) UNIQUE NOT NULL,
|
||
value TEXT NOT NULL,
|
||
type VARCHAR(50) DEFAULT 'string',
|
||
createdAt TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||
updatedAt TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
||
INDEX idx_key (config_key),
|
||
INDEX idx_type (type)
|
||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;`
|
||
];
|
||
|
||
async function createTables() {
|
||
let connection;
|
||
|
||
try {
|
||
console.log('🔌 正在连接腾讯云数据库...');
|
||
|
||
// 创建连接
|
||
connection = await mysql.createConnection(dbConfig);
|
||
|
||
console.log('✅ 数据库连接成功！');
|
||
|
||
// 创建数据库（如果不存在）
|
||
try {
|
||
await connection.execute(`CREATE DATABASE IF NOT EXISTS pronode_db CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci`);
|
||
console.log('✅ 数据库 pronode_db 已确认存在');
|
||
} catch (error) {
|
||
console.log('⚠️ 数据库创建跳过（可能已存在）');
|
||
}
|
||
|
||
// 使用数据库
|
||
await connection.execute('USE pronode_db');
|
||
console.log('✅ 已切换到 pronode_db 数据库');
|
||
|
||
// 创建表
|
||
console.log('🏗️ 开始创建数据表...');
|
||
|
||
for (let i = 0; i < createTablesSQL.length; i++) {
|
||
const sql = createTablesSQL[i];
|
||
const tableName = sql.match(/CREATE TABLE IF NOT EXISTS (\w+)/)[1];
|
||
|
||
try {
|
||
await connection.execute(sql);
|
||
console.log(`✅ 表 ${tableName} 创建成功`);
|
||
} catch (error) {
|
||
console.error(`❌ 表 ${tableName} 创建失败:`, error.message);
|
||
}
|
||
}
|
||
|
||
// 验证表是否创建成功
|
||
console.log('🔍 验证表创建结果...');
|
||
const [tables] = await connection.execute('SHOW TABLES');
|
||
console.log('📋 当前数据库表:');
|
||
tables.forEach(table => {
|
||
console.log(` - ${Object.values(table)[0]}`);
|
||
});
|
||
|
||
// 插入初始数据
|
||
console.log('🌱 插入初始数据...');
|
||
await insertInitialData(connection);
|
||
|
||
console.log('🎉 数据库表创建完成！');
|
||
return true;
|
||
|
||
} catch (error) {
|
||
console.error('❌ 数据库操作失败:', error.message);
|
||
return false;
|
||
} finally {
|
||
if (connection) {
|
||
await connection.end();
|
||
console.log('🔌 数据库连接已关闭');
|
||
}
|
||
}
|
||
}
|
||
|
||
async function insertInitialData(connection) {
|
||
try {
|
||
// 创建管理员用户
|
||
const adminUser = {
|
||
id: 'admin_' + Date.now(),
|
||
email: 'admin@promptforge.com',
|
||
name: 'PromptForge Admin',
|
||
avatar: 'https://avatars.githubusercontent.com/u/1234567?v=4'
|
||
};
|
||
|
||
await connection.execute(
|
||
'INSERT IGNORE INTO users (id, email, name, avatar) VALUES (?, ?, ?, ?)',
|
||
[adminUser.id, adminUser.email, adminUser.name, adminUser.avatar]
|
||
);
|
||
console.log('✅ 管理员用户创建成功');
|
||
|
||
// 创建示例模板
|
||
const sampleTemplates = [
|
||
{
|
||
id: 'template_1_' + Date.now(),
|
||
title: 'API设计文档生成器',
|
||
description: '生成完整的API设计文档，包含端点、参数、响应格式等',
|
||
category: 'programming',
|
||
tags: JSON.stringify(['api', 'documentation', 'backend']),
|
||
role: '你是一位资深软件架构师，拥有丰富的API设计经验。',
|
||
task: '我的任务是生成一个关于{{topic}}的API设计文档。',
|
||
context: '这个API将用于{{useCase}}场景，需要支持{{features}}功能。',
|
||
constraints: JSON.stringify([
|
||
{ id: '1', text: '使用RESTful设计原则', category: 'quality' },
|
||
{ id: '2', text: '包含完整的错误处理', category: 'safety' },
|
||
{ id: '3', text: '提供OpenAPI 3.0规范', category: 'format' }
|
||
]),
|
||
outputFormat: 'markdown',
|
||
variables: JSON.stringify([
|
||
{ name: 'topic', type: 'text', required: true, description: 'API主题' },
|
||
{ name: 'useCase', type: 'text', required: true, description: '使用场景' },
|
||
{ name: 'features', type: 'text', required: false, description: '核心功能' }
|
||
]),
|
||
authorId: adminUser.id,
|
||
isPublic: true,
|
||
isFeatured: true,
|
||
usageCount: 156,
|
||
rating: 4.8,
|
||
ratingCount: 23
|
||
}
|
||
];
|
||
|
||
for (const template of sampleTemplates) {
|
||
await connection.execute(
|
||
`INSERT IGNORE INTO templates (
|
||
id, title, description, category, tags, role, task, context,
|
||
constraints, outputFormat, variables, authorId, isPublic,
|
||
isFeatured, usageCount, rating, ratingCount
|
||
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||
[
|
||
template.id, template.title, template.description, template.category,
|
||
template.tags, template.role, template.task, template.context,
|
||
template.constraints, template.outputFormat, template.variables,
|
||
template.authorId, template.isPublic, template.isFeatured,
|
||
template.usageCount, template.rating, template.ratingCount
|
||
]
|
||
);
|
||
}
|
||
console.log('✅ 示例模板创建成功');
|
||
|
||
// 创建系统配置
|
||
const configs = [
|
||
{ key: 'site_name', value: 'PromptForge', type: 'string' },
|
||
{ key: 'site_description', value: '专为大模型提示词系统优化的平台', type: 'string' },
|
||
{ key: 'max_templates_per_user', value: '100', type: 'number' },
|
||
{ key: 'enable_registration', value: 'true', type: 'boolean' }
|
||
];
|
||
|
||
for (const config of configs) {
|
||
await connection.execute(
|
||
'INSERT IGNORE INTO system_configs (id, config_key, value, type) VALUES (?, ?, ?, ?)',
|
||
['config_' + Date.now() + Math.random(), config.key, config.value, config.type]
|
||
);
|
||
}
|
||
console.log('✅ 系统配置创建成功');
|
||
|
||
} catch (error) {
|
||
console.error('❌ 初始数据插入失败:', error.message);
|
||
}
|
||
}
|
||
|
||
// 运行创建表脚本
|
||
createTables().then(success => {
|
||
if (success) {
|
||
console.log('🎉 数据库表创建和初始化完成！');
|
||
} else {
|
||
console.log('💥 数据库表创建失败！');
|
||
}
|
||
process.exit(success ? 0 : 1);
|
||
});
|