feat: 添加PostgreSQL数据库集成和机台管理功能
- 新增PostgreSQL数据库支持，包含内置和外部数据库两种部署模式 - 实现机台管理功能，包括保护状态控制和VHD关键词设置 - 添加用户认证系统，支持会话管理和权限控制 - 更新Docker配置，支持数据库持久化和健康检查 - 扩展API接口，提供完整的RESTful API文档 - 更新客户端代码以支持机台ID和保护检查功能 - 添加数据库初始化脚本和配置指南
This commit is contained in:
Lannamokia
@@ -0,0 +1,195 @@
|
||||
const { Pool } = require('pg');
|
||||
require('dotenv').config();
|
||||
|
||||
// 数据库配置
|
||||
const dbConfig = {
|
||||
host: process.env.DB_HOST || 'localhost',
|
||||
port: parseInt(process.env.DB_PORT) || 5432,
|
||||
database: process.env.DB_NAME || 'vhd_select',
|
||||
user: process.env.DB_USER || 'postgres',
|
||||
password: process.env.DB_PASSWORD || 'password',
|
||||
max: parseInt(process.env.DB_MAX_CONNECTIONS) || 20, // 最大连接数
|
||||
idleTimeoutMillis: parseInt(process.env.DB_IDLE_TIMEOUT) || 30000, // 空闲连接超时时间
|
||||
connectionTimeoutMillis: parseInt(process.env.DB_CONNECTION_TIMEOUT) || 5000, // 连接超时时间
|
||||
ssl: process.env.DB_SSL === 'true' ? { rejectUnauthorized: false } : false,
|
||||
};
|
||||
|
||||
// 创建PostgreSQL连接池
|
||||
const pool = new Pool(dbConfig);
|
||||
|
||||
// 数据库操作类
|
||||
class Database {
|
||||
constructor() {
|
||||
this.pool = pool;
|
||||
this.initializeDatabase();
|
||||
}
|
||||
|
||||
// 初始化数据库连接
|
||||
async initializeDatabase() {
|
||||
const useEmbeddedDb = process.env.USE_EMBEDDED_DB === 'true';
|
||||
console.log(`数据库配置: ${useEmbeddedDb ? '内置数据库' : '外部数据库'}`);
|
||||
console.log(`连接地址: ${dbConfig.host}:${dbConfig.port}/${dbConfig.database}`);
|
||||
|
||||
try {
|
||||
const client = await this.pool.connect();
|
||||
console.log('PostgreSQL数据库连接成功');
|
||||
|
||||
// 测试查询
|
||||
const result = await client.query('SELECT NOW()');
|
||||
console.log('数据库时间:', result.rows[0].now);
|
||||
|
||||
// 检查表是否存在
|
||||
const tableCheck = await client.query(`
|
||||
SELECT EXISTS (
|
||||
SELECT FROM information_schema.tables
|
||||
WHERE table_schema = 'public'
|
||||
AND table_name = 'machines'
|
||||
);
|
||||
`);
|
||||
|
||||
if (tableCheck.rows[0].exists) {
|
||||
console.log('数据库表结构已存在');
|
||||
} else {
|
||||
console.log('警告: machines表不存在，请确保数据库已正确初始化');
|
||||
}
|
||||
|
||||
client.release();
|
||||
this.isConnected = true;
|
||||
} catch (error) {
|
||||
console.error('数据库连接失败:', error.message);
|
||||
console.error('错误详情:', error);
|
||||
|
||||
if (useEmbeddedDb) {
|
||||
console.error('内置数据库连接失败，请检查Docker配置');
|
||||
} else {
|
||||
console.error('外部数据库连接失败，请检查数据库服务器状态和连接配置');
|
||||
}
|
||||
|
||||
this.isConnected = false;
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
// 获取机台信息
|
||||
async getMachine(machineId) {
|
||||
try {
|
||||
const client = await this.pool.connect();
|
||||
const result = await client.query(
|
||||
'SELECT * FROM machines WHERE machine_id = $1',
|
||||
[machineId]
|
||||
);
|
||||
client.release();
|
||||
|
||||
return result.rows[0] || null;
|
||||
} catch (error) {
|
||||
console.error('获取机台信息失败:', error.message);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
// 创建或更新机台信息
|
||||
async upsertMachine(machineId, isProtected = false, vhdKeyword = 'SDEZ') {
|
||||
try {
|
||||
const client = await this.pool.connect();
|
||||
const result = await client.query(`
|
||||
INSERT INTO machines (machine_id, protected, vhd_keyword)
|
||||
VALUES ($1, $2, $3)
|
||||
ON CONFLICT (machine_id)
|
||||
DO UPDATE SET
|
||||
protected = EXCLUDED.protected,
|
||||
vhd_keyword = EXCLUDED.vhd_keyword,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
RETURNING *
|
||||
`, [machineId, isProtected, vhdKeyword]);
|
||||
client.release();
|
||||
|
||||
return result.rows[0];
|
||||
} catch (error) {
|
||||
console.error('更新机台信息失败:', error.message);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
// 更新机台保护状态
|
||||
async updateMachineProtection(machineId, isProtected) {
|
||||
try {
|
||||
const client = await this.pool.connect();
|
||||
const result = await client.query(`
|
||||
UPDATE machines
|
||||
SET protected = $2, updated_at = CURRENT_TIMESTAMP
|
||||
WHERE machine_id = $1
|
||||
RETURNING *
|
||||
`, [machineId, isProtected]);
|
||||
client.release();
|
||||
|
||||
return result.rows[0] || null;
|
||||
} catch (error) {
|
||||
console.error('更新机台保护状态失败:', error.message);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
// 更新机台VHD关键词
|
||||
async updateMachineVhdKeyword(machineId, vhdKeyword) {
|
||||
try {
|
||||
const client = await this.pool.connect();
|
||||
const result = await client.query(`
|
||||
UPDATE machines
|
||||
SET vhd_keyword = $2, updated_at = CURRENT_TIMESTAMP
|
||||
WHERE machine_id = $1
|
||||
RETURNING *
|
||||
`, [machineId, vhdKeyword]);
|
||||
client.release();
|
||||
|
||||
return result.rows[0] || null;
|
||||
} catch (error) {
|
||||
console.error('更新机台VHD关键词失败:', error.message);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
// 获取所有机台信息
|
||||
async getAllMachines() {
|
||||
try {
|
||||
const client = await this.pool.connect();
|
||||
const result = await client.query(
|
||||
'SELECT * FROM machines ORDER BY machine_id'
|
||||
);
|
||||
client.release();
|
||||
|
||||
return result.rows;
|
||||
} catch (error) {
|
||||
console.error('获取所有机台信息失败:', error.message);
|
||||
return [];
|
||||
}
|
||||
}
|
||||
|
||||
// 删除机台
|
||||
async deleteMachine(machineId) {
|
||||
try {
|
||||
const client = await this.pool.connect();
|
||||
const result = await client.query(
|
||||
'DELETE FROM machines WHERE machine_id = $1 RETURNING *',
|
||||
[machineId]
|
||||
);
|
||||
client.release();
|
||||
|
||||
return result.rows[0] || null;
|
||||
} catch (error) {
|
||||
console.error('删除机台失败:', error.message);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
// 关闭数据库连接池
|
||||
async close() {
|
||||
try {
|
||||
await this.pool.end();
|
||||
console.log('数据库连接池已关闭');
|
||||
} catch (error) {
|
||||
console.error('关闭数据库连接池失败:', error.message);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = new Database();
|
||||
Reference in New Issue
Block a user