285 lines
8.9 KiB
JavaScript
285 lines
8.9 KiB
JavaScript
const express = require('express');
|
||
const http = require('http');
|
||
const cors = require('cors');
|
||
const dotenv = require('dotenv');
|
||
const multer = require('multer');
|
||
const swaggerUi = require('swagger-ui-express');
|
||
const swaggerSpec = require('./config/swagger');
|
||
const { sequelize } = require('./config/database-simple');
|
||
const webSocketManager = require('./utils/websocket');
|
||
const logger = require('./utils/logger');
|
||
const {
|
||
apiRateLimiter,
|
||
inputSanitizer,
|
||
sessionTimeoutCheck,
|
||
securityHeaders
|
||
} = require('./middleware/security');
|
||
const { autoOperationLogger } = require('./middleware/autoOperationLogger');
|
||
|
||
// 加载环境变量
|
||
dotenv.config();
|
||
|
||
// 创建Express应用和HTTP服务器
|
||
const app = express();
|
||
const server = http.createServer(app);
|
||
const PORT = process.env.PORT || 3001;
|
||
|
||
// 配置文件上传
|
||
const storage = multer.diskStorage({
|
||
destination: function (req, file, cb) {
|
||
cb(null, 'uploads/') // 上传文件保存目录
|
||
},
|
||
filename: function (req, file, cb) {
|
||
// 生成唯一文件名
|
||
const uniqueSuffix = Date.now() + '-' + Math.round(Math.random() * 1E9);
|
||
cb(null, file.fieldname + '-' + uniqueSuffix + '.' + file.originalname.split('.').pop());
|
||
}
|
||
});
|
||
|
||
const upload = multer({
|
||
storage: storage,
|
||
limits: {
|
||
fileSize: 10 * 1024 * 1024 // 限制文件大小为10MB
|
||
},
|
||
fileFilter: function (req, file, cb) {
|
||
// 只允许Excel文件
|
||
const allowedTypes = [
|
||
'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
|
||
'application/vnd.ms-excel'
|
||
];
|
||
|
||
if (allowedTypes.includes(file.mimetype)) {
|
||
cb(null, true);
|
||
} else {
|
||
cb(new Error('只允许上传Excel文件(.xlsx或.xls格式)'), false);
|
||
}
|
||
}
|
||
});
|
||
|
||
// 安全中间件
|
||
app.use(securityHeaders);
|
||
app.use(apiRateLimiter);
|
||
app.use(inputSanitizer);
|
||
app.use(sessionTimeoutCheck);
|
||
|
||
// 基础中间件
|
||
app.use(cors());
|
||
app.use(express.json({ charset: 'utf8' }));
|
||
app.use(express.urlencoded({ extended: true, charset: 'utf8' }));
|
||
|
||
// API统一中间件
|
||
const { apiMiddleware, errorHandler, notFoundHandler } = require('./middleware/apiMiddleware');
|
||
app.use('/api', apiMiddleware);
|
||
|
||
// 静态文件服务 - 提供前端构建文件
|
||
app.use(express.static(__dirname + '/dist'));
|
||
|
||
// 自动操作日志中间件（在所有API路由之前）
|
||
app.use('/api', autoOperationLogger);
|
||
|
||
// 设置响应头确保UTF-8编码（仅对API请求）
|
||
app.use('/api', (req, res, next) => {
|
||
res.setHeader('Content-Type', 'application/json; charset=utf-8');
|
||
next();
|
||
});
|
||
|
||
// Swagger 文档路由配置
|
||
const swaggerOptions = {
|
||
explorer: true,
|
||
swaggerOptions: {
|
||
docExpansion: 'list',
|
||
defaultModelsExpandDepth: 2,
|
||
defaultModelExpandDepth: 2,
|
||
displayRequestDuration: true,
|
||
filter: true,
|
||
showExtensions: true,
|
||
showCommonExtensions: true,
|
||
tryItOutEnabled: true,
|
||
persistAuthorization: true,
|
||
layout: 'BaseLayout',
|
||
deepLinking: true,
|
||
displayOperationId: false,
|
||
syntaxHighlight: {
|
||
activate: true,
|
||
theme: 'tomorrow-night'
|
||
}
|
||
},
|
||
customCss: `
|
||
.swagger-ui .topbar { background: linear-gradient(135deg, #1890ff 0%, #722ed1 100%); }
|
||
.swagger-ui .topbar .download-url-wrapper { display: none; }
|
||
.swagger-ui .info { margin: 20px 0; }
|
||
.swagger-ui .info .title { color: #1890ff; font-size: 2.5em; }
|
||
.swagger-ui .scheme-container { background: #f8f9fa; padding: 15px; border-radius: 8px; }
|
||
.swagger-ui .btn.authorize { background: #52c41a; border-color: #52c41a; }
|
||
.swagger-ui .btn.authorize:hover { background: #73d13d; border-color: #73d13d; }
|
||
.swagger-ui .opblock.opblock-post { border-color: #1890ff; background: rgba(24, 144, 255, 0.1); }
|
||
.swagger-ui .opblock.opblock-get { border-color: #52c41a; background: rgba(82, 196, 26, 0.1); }
|
||
.swagger-ui .opblock.opblock-put { border-color: #fa8c16; background: rgba(250, 140, 22, 0.1); }
|
||
.swagger-ui .opblock.opblock-delete { border-color: #f5222d; background: rgba(245, 34, 45, 0.1); }
|
||
.swagger-ui .parameters-col_description p { font-size: 13px; line-height: 1.4; }
|
||
.swagger-ui .response-col_description { font-size: 13px; }
|
||
.swagger-ui .model-box { background: #fafafa; border: 1px solid #d9d9d9; border-radius: 6px; }
|
||
.swagger-ui .model-title { color: #1890ff; font-weight: 600; }
|
||
.swagger-ui .prop-type { color: #722ed1; font-weight: 500; }
|
||
.swagger-ui .prop-format { color: #fa8c16; }
|
||
.swagger-ui .opblock-summary-description { font-size: 14px; color: #595959; }
|
||
`,
|
||
customSiteTitle: '宁夏智慧养殖监管平台 - API文档',
|
||
customfavIcon: '/favicon.ico'
|
||
};
|
||
|
||
app.use('/api-docs', swaggerUi.serve, swaggerUi.setup(swaggerSpec, swaggerOptions));
|
||
|
||
// 基础路由
|
||
app.get('/', (req, res) => {
|
||
res.json({
|
||
message: '宁夏智慧养殖监管平台API服务',
|
||
version: '1.0.0',
|
||
docs: '/api-docs'
|
||
});
|
||
});
|
||
|
||
// 数据库同步
|
||
sequelize.sync()
|
||
.then(() => {
|
||
console.log('数据库同步成功');
|
||
})
|
||
.catch(err => {
|
||
console.error('数据库同步失败:', err);
|
||
});
|
||
|
||
// 认证相关路由
|
||
app.use('/api/auth', require('./routes/auth'));
|
||
|
||
// 用户相关路由
|
||
app.use('/api/users', require('./routes/users'));
|
||
|
||
// 产品相关路由
|
||
app.use('/api/products', require('./routes/products'));
|
||
|
||
// 订单相关路由
|
||
app.use('/api/orders', require('./routes/orders'));
|
||
|
||
// 农场相关路由
|
||
app.use('/api/farms', require('./routes/farms'));
|
||
|
||
// 动物相关路由
|
||
app.use('/api/animals', require('./routes/animals'));
|
||
|
||
// 牛只档案相关路由（基于iot_cattle表）
|
||
app.use('/api/iot-cattle', require('./routes/iot-cattle'));
|
||
|
||
// 栏舍相关路由
|
||
app.use('/api/pens', require('./routes/pens'));
|
||
app.use('/api/cattle-pens', require('./routes/cattle-pens'));
|
||
app.use('/api/cattle-batches', require('./routes/cattle-batches'));
|
||
app.use('/api/cattle-transfer-records', require('./routes/cattle-transfer-records'));
|
||
app.use('/api/cattle-exit-records', require('./routes/cattle-exit-records'));
|
||
|
||
// 牛只基础数据路由
|
||
app.use('/api/cattle-user', require('./routes/cattle-user'));
|
||
app.use('/api/cattle-type', require('./routes/cattle-type'));
|
||
|
||
// 设备相关路由
|
||
app.use('/api/devices', require('./routes/devices'));
|
||
|
||
// 智能设备相关路由
|
||
app.use('/api/smart-devices', require('./routes/smart-devices'));
|
||
app.use('/api/iot-jbq-client', require('./routes/iot-jbq-client'));
|
||
|
||
// 智能预警相关路由
|
||
app.use('/api/smart-alerts', require('./routes/smart-alerts'));
|
||
|
||
// 电子围栏相关路由
|
||
app.use('/api/electronic-fence', require('./routes/electronic-fence'));
|
||
|
||
// 电子围栏坐标点相关路由
|
||
app.use('/api/electronic-fence-points', require('./routes/electronic-fence-points'));
|
||
|
||
// 预警相关路由
|
||
app.use('/api/alerts', require('./routes/alerts'));
|
||
|
||
// 统计数据相关路由
|
||
app.use('/api/stats', require('./routes/stats'));
|
||
|
||
// 百度地图API相关路由
|
||
app.use('/api/map', require('./routes/map'));
|
||
|
||
// 报表管理相关路由
|
||
app.use('/api/reports', require('./routes/reports'));
|
||
|
||
// 系统管理相关路由
|
||
app.use('/api/system', require('./routes/system'));
|
||
|
||
// 数据备份相关路由
|
||
app.use('/api/backup', require('./routes/backup'));
|
||
|
||
// 菜单管理相关路由
|
||
app.use('/api/menus', require('./routes/menus'));
|
||
|
||
// 角色权限管理相关路由
|
||
app.use('/api/role-permissions', require('./routes/role-permissions'));
|
||
|
||
// 表单日志相关路由
|
||
app.use('/api/form-logs', require('./routes/formLogs'));
|
||
|
||
// 操作日志相关路由
|
||
app.use('/api/operation-logs', require('./routes/operationLogs'));
|
||
|
||
// 绑定信息相关路由
|
||
app.use('/api/binding', require('./routes/binding'));
|
||
|
||
// API演示路由
|
||
app.use('/api/demo', require('./routes/api-demo'));
|
||
|
||
// 统一错误处理中间件
|
||
app.use(errorHandler);
|
||
|
||
// 404处理中间件
|
||
app.use(notFoundHandler);
|
||
|
||
// 前端路由 - 处理所有非API请求，返回前端应用
|
||
app.get('*', (req, res) => {
|
||
// 如果是API请求，跳过
|
||
if (req.path.startsWith('/api')) {
|
||
return res.status(404).json({
|
||
success: false,
|
||
message: 'API接口不存在'
|
||
});
|
||
}
|
||
|
||
// 返回前端应用
|
||
res.sendFile(__dirname + '/dist/index.html');
|
||
});
|
||
|
||
// 错误处理中间件
|
||
app.use((err, req, res, next) => {
|
||
console.error('服务器错误:', err);
|
||
res.status(500).json({
|
||
success: false,
|
||
message: '服务器错误'
|
||
});
|
||
});
|
||
|
||
// 初始化WebSocket
|
||
webSocketManager.init(server);
|
||
|
||
// 初始化实时数据推送服务
|
||
const realtimeService = require('./services/realtimeService');
|
||
|
||
// 启动服务器
|
||
server.listen(PORT, '0.0.0.0', () => {
|
||
console.log(`服务器运行在端口 ${PORT}`);
|
||
console.log(`服务器监听所有网络接口 (0.0.0.0:${PORT})`);
|
||
console.log(`API 文档地址: http://localhost:${PORT}/api-docs`);
|
||
console.log(`WebSocket 服务已启动`);
|
||
|
||
// 启动实时数据推送服务
|
||
realtimeService.start();
|
||
console.log(`实时数据推送服务已启动`);
|
||
|
||
logger.info(`宁夏智慧养殖监管平台服务器启动成功，端口: ${PORT}`);
|
||
});
|
||
|
||
// 导出app和webSocketManager供其他模块使用
|
||
module.exports = { app, webSocketManager }; |