second commit
This commit is contained in:
279
create-tables.js
Normal file
279
create-tables.js
Normal file
@@ -0,0 +1,279 @@
|
||||
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);
|
||||
});
|
||||
Reference in New Issue
Block a user