330 lines
9.0 KiB
JavaScript
330 lines
9.0 KiB
JavaScript
const express = require('express');
|
|
const cors = require('cors');
|
|
const helmet = require('helmet');
|
|
const dotenv = require('dotenv');
|
|
const rateLimit = require('express-rate-limit');
|
|
const mysql = require('mysql2/promise');
|
|
const path = require('path');
|
|
|
|
// 导入路由模块
|
|
const authModule = require('./routes/auth');
|
|
const usersModule = require('./routes/users');
|
|
const cattleModule = require('./routes/cattle');
|
|
const financeModule = require('./routes/finance');
|
|
const tradingModule = require('./routes/trading');
|
|
const governmentModule = require('./routes/government');
|
|
const mallModule = require('./routes/mall');
|
|
|
|
// 加载环境变量
|
|
dotenv.config();
|
|
|
|
// 数据库连接
|
|
let pool, testDatabaseConnection;
|
|
|
|
if (process.env.DB_TYPE === 'sqlite') {
|
|
// 使用SQLite
|
|
const sqliteDb = require('./database-sqlite');
|
|
pool = sqliteDb.pool;
|
|
testDatabaseConnection = sqliteDb.testDatabaseConnection;
|
|
} else {
|
|
// 使用MySQL
|
|
const mysql = require('mysql2/promise');
|
|
|
|
// 数据库连接配置
|
|
const dbConfig = {
|
|
host: process.env.DB_HOST,
|
|
port: process.env.DB_PORT,
|
|
user: process.env.DB_USER,
|
|
password: process.env.DB_PASSWORD,
|
|
database: process.env.DB_NAME,
|
|
charset: process.env.DB_CHARSET || 'utf8mb4',
|
|
connectionLimit: 10,
|
|
acquireTimeout: 60000,
|
|
timeout: 60000
|
|
};
|
|
|
|
// 创建数据库连接池
|
|
pool = mysql.createPool(dbConfig);
|
|
|
|
// 测试数据库连接
|
|
testDatabaseConnection = async function() {
|
|
try {
|
|
const connection = await pool.getConnection();
|
|
console.log('✅ MySQL数据库连接成功');
|
|
console.log(`📍 连接到: ${process.env.DB_HOST}:${process.env.DB_PORT}/${process.env.DB_NAME}`);
|
|
connection.release();
|
|
return true;
|
|
} catch (error) {
|
|
console.error('❌ 数据库连接失败:', error.message);
|
|
console.log('⚠️ 服务将继续运行，但数据库功能不可用');
|
|
return false;
|
|
}
|
|
};
|
|
}
|
|
|
|
// 创建Express应用
|
|
const app = express();
|
|
const PORT = process.env.PORT || 8888;
|
|
|
|
// 初始化数据库和中间件的异步函数
|
|
async function initializeApp() {
|
|
// 启动时测试数据库连接
|
|
await testDatabaseConnection();
|
|
|
|
// 设置路由模块的数据库连接
|
|
authModule.setPool(pool);
|
|
usersModule.setMiddleware(authModule.authenticateToken, authModule.checkPermission, pool);
|
|
cattleModule.setMiddleware(authModule.authenticateToken, authModule.checkPermission, pool);
|
|
financeModule.setMiddleware(authModule.authenticateToken, authModule.checkPermission, pool);
|
|
tradingModule.setMiddleware(authModule.authenticateToken, authModule.checkPermission, pool);
|
|
governmentModule.setMiddleware(authModule.authenticateToken, authModule.checkPermission, pool);
|
|
mallModule.setMiddleware(authModule.authenticateToken, authModule.checkPermission, pool);
|
|
|
|
console.log('✅ 中间件初始化完成');
|
|
}
|
|
|
|
// 初始化应用（不阻塞服务启动）
|
|
initializeApp().catch(console.error);
|
|
|
|
// 导入错误处理中间件
|
|
const { errorHandler, notFoundHandler, requestLogger, performanceMonitor } = require('./middleware/errorHandler');
|
|
|
|
// 中间件
|
|
app.use(requestLogger); // 请求日志
|
|
app.use(performanceMonitor); // 性能监控
|
|
app.use(helmet()); // 安全头部
|
|
app.use(cors({
|
|
origin: process.env.CORS_ORIGIN || '*',
|
|
credentials: true
|
|
})); // 跨域支持
|
|
app.use(express.json({ limit: '10mb' })); // JSON解析
|
|
app.use(express.urlencoded({ extended: true, limit: '10mb' })); // URL编码解析
|
|
|
|
// 速率限制
|
|
const limiter = rateLimit({
|
|
windowMs: 15 * 60 * 1000, // 15分钟
|
|
max: 100, // 限制每个IP 15分钟内最多100个请求
|
|
message: '请求过于频繁，请稍后再试'
|
|
});
|
|
app.use(limiter);
|
|
|
|
// 基础路由
|
|
app.get('/', (req, res) => {
|
|
res.json({
|
|
message: '欢迎使用锡林郭勒盟地区智慧养殖产业平台API服务',
|
|
version: '1.0.0',
|
|
timestamp: new Date().toISOString()
|
|
});
|
|
});
|
|
|
|
app.get('/health', async (req, res) => {
|
|
try {
|
|
// 测试数据库连接
|
|
const connection = await pool.getConnection();
|
|
const [rows] = await connection.execute('SELECT 1 as test');
|
|
connection.release();
|
|
|
|
res.json({
|
|
status: 'OK',
|
|
timestamp: new Date().toISOString(),
|
|
database: 'Connected',
|
|
environment: process.env.NODE_ENV,
|
|
version: '1.0.0'
|
|
});
|
|
} catch (error) {
|
|
res.status(500).json({
|
|
status: 'ERROR',
|
|
timestamp: new Date().toISOString(),
|
|
database: 'Disconnected',
|
|
error: error.message
|
|
});
|
|
}
|
|
});
|
|
|
|
// API路由
|
|
app.use('/api/v1/auth', authModule.router);
|
|
app.use('/api/v1/users', usersModule.router);
|
|
app.use('/api/v1/cattle', cattleModule.router);
|
|
app.use('/api/v1/finance', financeModule.router);
|
|
app.use('/api/v1/trading', tradingModule.router);
|
|
app.use('/api/v1/government', governmentModule.router);
|
|
app.use('/api/v1/mall', mallModule.router);
|
|
|
|
// 数据库查询接口
|
|
app.get('/api/v1/database/tables', async (req, res) => {
|
|
try {
|
|
const [tables] = await pool.execute(
|
|
'SELECT TABLE_NAME, TABLE_COMMENT FROM information_schema.TABLES WHERE TABLE_SCHEMA = ?',
|
|
[process.env.DB_NAME]
|
|
);
|
|
|
|
res.json({
|
|
success: true,
|
|
data: {
|
|
database: process.env.DB_NAME,
|
|
tables: tables,
|
|
count: tables.length
|
|
},
|
|
timestamp: new Date().toISOString()
|
|
});
|
|
} catch (error) {
|
|
res.status(500).json({
|
|
success: false,
|
|
error: error.message,
|
|
timestamp: new Date().toISOString()
|
|
});
|
|
}
|
|
});
|
|
|
|
// 数据库连接状态接口
|
|
app.get('/api/v1/database/status', async (req, res) => {
|
|
try {
|
|
const [statusResult] = await pool.execute('SHOW STATUS LIKE "Threads_connected"');
|
|
const [versionResult] = await pool.execute('SELECT VERSION() as version');
|
|
|
|
res.json({
|
|
success: true,
|
|
data: {
|
|
connected: true,
|
|
version: versionResult[0].version,
|
|
threads_connected: statusResult[0].Value,
|
|
database: process.env.DB_NAME,
|
|
host: process.env.DB_HOST,
|
|
port: process.env.DB_PORT
|
|
},
|
|
timestamp: new Date().toISOString()
|
|
});
|
|
} catch (error) {
|
|
res.status(500).json({
|
|
success: false,
|
|
error: error.message,
|
|
timestamp: new Date().toISOString()
|
|
});
|
|
}
|
|
});
|
|
app.get('/api/v1/dashboard/map/regions', (req, res) => {
|
|
// 模拟锡林郭勒盟各区域数据
|
|
const regions = [
|
|
{
|
|
id: 'xlg',
|
|
name: '锡林浩特市',
|
|
coordinates: [116.093, 43.946],
|
|
cattle_count: 25600,
|
|
farm_count: 120,
|
|
output_value: 650000000
|
|
},
|
|
{
|
|
id: 'dwq',
|
|
name: '东乌旗',
|
|
coordinates: [116.980, 45.514],
|
|
cattle_count: 18500,
|
|
farm_count: 95,
|
|
output_value: 480000000
|
|
},
|
|
{
|
|
id: 'xwq',
|
|
name: '西乌旗',
|
|
coordinates: [117.615, 44.587],
|
|
cattle_count: 21200,
|
|
farm_count: 108,
|
|
output_value: 520000000
|
|
},
|
|
{
|
|
id: 'abg',
|
|
name: '阿巴嘎旗',
|
|
coordinates: [114.971, 44.022],
|
|
cattle_count: 16800,
|
|
farm_count: 86,
|
|
output_value: 420000000
|
|
},
|
|
{
|
|
id: 'snz',
|
|
name: '苏尼特左旗',
|
|
coordinates: [113.653, 43.859],
|
|
cattle_count: 12400,
|
|
farm_count: 65,
|
|
output_value: 310000000
|
|
}
|
|
];
|
|
|
|
res.json({ regions });
|
|
});
|
|
|
|
app.get('/api/v1/dashboard/map/region/:regionId', (req, res) => {
|
|
const { regionId } = req.params;
|
|
|
|
// 模拟各区域详细数据
|
|
const regionDetails = {
|
|
'xlg': {
|
|
region: {
|
|
id: 'xlg',
|
|
name: '锡林浩特市',
|
|
coordinates: [116.093, 43.946],
|
|
cattle_count: 25600,
|
|
farm_count: 120,
|
|
output_value: 650000000,
|
|
trend: 'up'
|
|
},
|
|
farms: [
|
|
{
|
|
id: 'FARM001',
|
|
name: '锡林浩特市第一牧场',
|
|
coordinates: [116.120, 43.950],
|
|
cattle_count: 2450,
|
|
output_value: 62000000
|
|
},
|
|
{
|
|
id: 'FARM002',
|
|
name: '锡林浩特市第二牧场',
|
|
coordinates: [116.080, 43.930],
|
|
cattle_count: 2100,
|
|
output_value: 53000000
|
|
}
|
|
]
|
|
},
|
|
'dwq': {
|
|
region: {
|
|
id: 'dwq',
|
|
name: '东乌旗',
|
|
coordinates: [116.980, 45.514],
|
|
cattle_count: 18500,
|
|
farm_count: 95,
|
|
output_value: 480000000,
|
|
trend: 'up'
|
|
},
|
|
farms: [
|
|
{
|
|
id: 'FARM003',
|
|
name: '东乌旗牧场A',
|
|
coordinates: [116.990, 45.520],
|
|
cattle_count: 1950,
|
|
output_value: 49000000
|
|
}
|
|
]
|
|
}
|
|
};
|
|
|
|
const detail = regionDetails[regionId];
|
|
if (detail) {
|
|
res.json(detail);
|
|
} else {
|
|
res.status(404).json({ error: '区域未找到' });
|
|
}
|
|
});
|
|
|
|
// 错误处理中间件（必须放在所有路由之后）
|
|
app.use(notFoundHandler); // 404处理
|
|
app.use(errorHandler); // 统一错误处理
|
|
|
|
// 启动服务器
|
|
const HOST = process.env.HOST || '0.0.0.0';
|
|
|
|
app.listen(PORT, HOST, () => {
|
|
console.log(`API服务器正在监听: http://${HOST}:${PORT}`);
|
|
console.log(`服务器绑定到: ${HOST}, 端口: ${PORT}`);
|
|
console.log('尝试使用以下URL访问:');
|
|
console.log(` http://localhost:${PORT}`);
|
|
console.log(` http://127.0.0.1:${PORT}`);
|
|
});
|
|
|
|
module.exports = app; |