mb-rrvideo-server 1.0.2

package/src/index.js

@@ -0,0 +1,138 @@
+/**
+ * mb-rrvideo-server 主入口
+ * 独立的 Node.js 转码服务
+ */
+
+const http = require('http');
+const url = require('url');
+const config = require('./config');
+const logger = require('./logger');
+const { handleReceiveRequest } = require('./routes/receive');
+const { handleConvertV1, handleConvertV2 } = require('./routes/convert');
+const { handleMerge } = require('./routes/merge');
+
+// 创建 HTTP 服务器
+const server = http.createServer((req, res) => {
+  try {
+    const parsed = url.parse(req.url || '/');
+    const pathname = parsed.pathname || '/';
+    
+    if (req.method === 'POST') {
+      // 统一接收入口（对应 PHP 的 receiveRequestByRecordIdAct）
+      if (pathname === '/Convert/receiveRequestByRecordIdAct') {
+        return handleReceiveRequest(req, res);
+      }
+      
+      // 直接转码入口（兼容旧版）
+      if (pathname === '/convert') {
+        let body = '';
+        req.on('data', (chunk) => { body += chunk.toString(); });
+        req.on('end', async () => {
+          try {
+            const params = JSON.parse(body || '{}');
+            const logFileName = `${params.record_id || 'unknown'}_convert_${Date.now()}`;
+            
+            // 判断是 V1 还是 V2
+            const isV2 = params.tasks && Array.isArray(params.tasks);
+            if (isV2) {
+              await handleConvertV2(params, res, logFileName);
+            } else {
+              await handleConvertV1(params, res, logFileName);
+            }
+          } catch (e) {
+            res.statusCode = 400;
+            res.setHeader('Content-Type', 'application/json');
+            res.end(JSON.stringify({ result_code: 1003, message: '请求体格式错误' }));
+          }
+        });
+        return;
+      }
+      
+      // 直接合并入口（兼容旧版）
+      if (pathname === '/merge') {
+        let body = '';
+        req.on('data', (chunk) => { body += chunk.toString(); });
+        req.on('end', async () => {
+          try {
+            const params = JSON.parse(body || '{}');
+            const logFileName = `${params.record_id || 'unknown'}_merge_${Date.now()}`;
+            await handleMerge(params, res, logFileName);
+          } catch (e) {
+            res.statusCode = 400;
+            res.setHeader('Content-Type', 'application/json');
+            res.end(JSON.stringify({ result_code: 1003, message: '请求体格式错误' }));
+          }
+        });
+        return;
+      }
+      
+      // 未知路由
+      res.statusCode = 404;
+      res.setHeader('Content-Type', 'application/json');
+      res.end(JSON.stringify({ result_code: 1000, message: '路由不存在' }));
+    } else if (req.method === 'GET') {
+      // 健康检查
+      if (pathname === '/' || pathname === '/health') {
+        res.setHeader('Content-Type', 'application/json');
+        res.end(JSON.stringify({ result_code: 0, message: 'rrvideo-server is running' }));
+        return;
+      }
+      
+      res.statusCode = 404;
+      res.setHeader('Content-Type', 'application/json');
+      res.end(JSON.stringify({ result_code: 1000, message: '路由不存在' }));
+    } else {
+      res.statusCode = 405;
+      res.setHeader('Content-Type', 'application/json');
+      res.end(JSON.stringify({ result_code: 1000, message: '方法不允许' }));
+    }
+  } catch (error) {
+    console.error(`处理请求时出错: ${error.message}`);
+    res.statusCode = 500;
+    res.setHeader('Content-Type', 'application/json');
+    res.end(JSON.stringify({ result_code: 1000, message: '服务器错误' }));
+  }
+});
+
+// 启动服务器
+if (require.main === module) {
+  const PORT = config.get('server.port', 24203);
+  const HOST = config.get('server.host', '0.0.0.0');
+  
+  server.listen(PORT, HOST, () => {
+    console.log(`rrvideo-server listening on ${HOST}:${PORT}`);
+    console.log(`Storage type: ${config.get('storage.type')}`);
+    console.log(`Log level: ${config.get('log.level')}`);
+  });
+  
+  // 程序异常或提前终止时的处理
+  process.on('exit', (code) => {
+    console.log(`进程退出，代码: ${code}`);
+    logger.flushAll();
+  });
+
+  process.on('SIGINT', () => {
+    console.log('收到 SIGINT 信号，正在关闭...');
+    logger.flushAll();
+    process.exit(0);
+  });
+
+  process.on('SIGTERM', () => {
+    console.log('收到 SIGTERM 信号，正在关闭...');
+    logger.flushAll();
+    process.exit(0);
+  });
+
+  process.on('uncaughtException', (err) => {
+    console.error(`未捕获的异常: ${err.message}`);
+    console.error(err.stack);
+    logger.flushAll();
+    process.exit(1);
+  });
+
+  process.on('unhandledRejection', (reason, promise) => {
+    console.error(`未处理的 Promise 拒绝: ${reason}`);
+  });
+}
+
+module.exports = { server };

package/src/logger.js

@@ -0,0 +1,144 @@
+const fs = require('fs');
+const path = require('path');
+
+class Logger {
+  constructor() {
+    this.buffers = new Map(); // fileKey -> { logs: [], timer: null }
+    this.flushInterval = 2000; // 2秒
+    this.bufferMaxSize = 500; // 500条
+    this.baseDir = path.join(__dirname, '../logs');
+  }
+
+  ensureDir(dirPath) {
+    if (!fs.existsSync(dirPath)) {
+      fs.mkdirSync(dirPath, { recursive: true });
+    }
+  }
+
+  getCurrentDate() {
+    const date = new Date();
+    const year = date.getFullYear();
+    const month = String(date.getMonth() + 1).padStart(2, '0');
+    const day = String(date.getDate()).padStart(2, '0');
+    return `${year}-${month}-${day}`;
+  }
+
+  getCurrentTime() {
+    const date = new Date();
+    const hour = String(date.getHours()).padStart(2, '0');
+    const minute = String(date.getMinutes()).padStart(2, '0');
+    const second = String(date.getSeconds()).padStart(2, '0');
+    return `${hour}:${minute}:${second}`;
+  }
+
+  getLogFilePath(logType, fileName) {
+    const dateStr = this.getCurrentDate();
+    const logDir = path.join(this.baseDir, logType, dateStr);
+    this.ensureDir(logDir);
+    return path.join(logDir, `${fileName}.log`);
+  }
+
+  flushBuffer(fileKey) {
+    const buf = this.buffers.get(fileKey);
+    if (!buf || buf.logs.length === 0) return;
+
+    try {
+      const { logType, fileName, logs } = buf;
+      const filePath = this.getLogFilePath(logType, fileName);
+      const content = logs.join('\n') + '\n';
+      fs.appendFileSync(filePath, content);
+      buf.logs = [];
+      
+      // 清除定时器
+      if (buf.timer) {
+        clearTimeout(buf.timer);
+        buf.timer = null;
+      }
+    } catch (error) {
+      console.error(`[Logger] 刷新日志失败 [${fileKey}]:`, error.message);
+    }
+  }
+
+  scheduleFlush(fileKey) {
+    const buf = this.buffers.get(fileKey);
+    if (!buf || buf.timer) return;
+
+    buf.timer = setTimeout(() => {
+      this.flushBuffer(fileKey);
+    }, this.flushInterval);
+  }
+
+  log(logType, fileName, message, level = 'INFO') {
+    try {
+      const timestamp = this.getCurrentTime();
+      const lines = String(message).split('\n');
+      const fileKey = `${logType}::${fileName}`;
+
+      if (!this.buffers.has(fileKey)) {
+        this.buffers.set(fileKey, {
+          logType,
+          fileName,
+          logs: [],
+          timer: null
+        });
+      }
+
+      const buf = this.buffers.get(fileKey);
+      
+      for (const line of lines) {
+        if (!line.trim()) continue;
+        const logLine = `[${timestamp}] [${level}] ${line}`;
+        console.log(logLine); // 同时输出到控制台
+        buf.logs.push(logLine);
+      }
+
+      // 缓冲区满立即刷新
+      if (buf.logs.length >= this.bufferMaxSize) {
+        this.flushBuffer(fileKey);
+      } else {
+        this.scheduleFlush(fileKey);
+      }
+    } catch (error) {
+      console.error(`[Logger] 写日志失败:`, error.message);
+    }
+  }
+
+  flushImmediate(logType, fileName) {
+    const fileKey = `${logType}::${fileName}`;
+    this.flushBuffer(fileKey);
+  }
+
+  flushAll() {
+    for (const fileKey of this.buffers.keys()) {
+      this.flushBuffer(fileKey);
+    }
+  }
+}
+
+// 单例模式
+const logger = new Logger();
+
+// 进程退出时刷新所有日志
+process.on('exit', () => {
+  logger.flushAll();
+});
+
+process.on('SIGINT', () => {
+  logger.flushAll();
+  process.exit(0);
+});
+
+process.on('SIGTERM', () => {
+  logger.flushAll();
+  process.exit(0);
+});
+
+// 导出便捷方法
+module.exports = {
+  info: (message, fileName, logType = 'convert') => logger.log(logType, fileName, message, 'INFO'),
+  debug: (message, fileName, logType = 'convert') => logger.log(logType, fileName, message, 'DEBUG'),
+  warn: (message, fileName, logType = 'convert') => logger.log(logType, fileName, message, 'WARN'),
+  error: (message, fileName, logType = 'convert') => logger.log(logType, fileName, message, 'ERROR'),
+  flush: (fileName, logType = 'convert') => logger.flushImmediate(logType, fileName),
+  flushAll: () => logger.flushAll()
+};