98 lines
2.7 KiB
JavaScript
98 lines
2.7 KiB
JavaScript
|
const express = require('express');
|
|||
|
|
const router = express.Router();
|
|||
|
|
const roomStore = require('../store/roomStore');
|
|||
|
|
const http = require('http');
|
|||
|
|
|
|||
|
|
// Java 后端地址 (host.docker.internal 用于从 Docker 容器访问宿主机)
|
|||
|
|
const JAVA_BACKEND_HOST = process.env.JAVA_BACKEND_HOST || 'host.docker.internal';
|
|||
|
|
const JAVA_BACKEND_PORT = process.env.JAVA_BACKEND_PORT || 8081;
|
|||
|
|
|
|||
|
|
// 转发到 Java 后端
|
|||
|
|
function forwardToJava(path, data) {
|
|||
|
|
const postData = JSON.stringify(data);
|
|||
|
|
const options = {
|
|||
|
|
hostname: JAVA_BACKEND_HOST,
|
|||
|
|
port: JAVA_BACKEND_PORT,
|
|||
|
|
path: path,
|
|||
|
|
method: 'POST',
|
|||
|
|
headers: {
|
|||
|
|
'Content-Type': 'application/json',
|
|||
|
|
'Content-Length': Buffer.byteLength(postData)
|
|||
|
|
},
|
|||
|
|
timeout: 5000
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
const req = http.request(options, (res) => {
|
|||
|
|
console.log(`[SRS->Java] ${path} 响应状态: ${res.statusCode}`);
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
req.on('error', (e) => {
|
|||
|
|
console.error(`[SRS->Java] ${path} 转发失败: ${e.message}`);
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
req.on('timeout', () => {
|
|||
|
|
console.error(`[SRS->Java] ${path} 转发超时`);
|
|||
|
|
req.destroy();
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
req.write(postData);
|
|||
|
|
req.end();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// POST /api/srs/on_publish - 推流开始回调
|
|||
|
|
router.post('/on_publish', (req, res) => {
|
|||
|
|
const { app, stream } = req.body;
|
|||
|
|
|
|||
|
|
console.log(`[SRS] 推流开始: app=${app}, stream=${stream}`);
|
|||
|
|
|
|||
|
|
// 更新内存存储
|
|||
|
|
const room = roomStore.setLiveStatus(stream, true);
|
|||
|
|
|
|||
|
|
if (room) {
|
|||
|
|
console.log(`[SRS] 房间 "${room.title}" 开始直播`);
|
|||
|
|
} else {
|
|||
|
|
console.log(`[SRS] 未找到对应房间，streamKey=${stream}`);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 转发到 Java 后端更新数据库
|
|||
|
|
forwardToJava('/api/front/live/srs/on_publish', { stream });
|
|||
|
|
|
|||
|
|
// SRS 要求返回 code: 0 表示成功
|
|||
|
|
res.json({ code: 0 });
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// POST /api/srs/on_unpublish - 推流结束回调
|
|||
|
|
router.post('/on_unpublish', (req, res) => {
|
|||
|
|
const { app, stream } = req.body;
|
|||
|
|
|
|||
|
|
console.log(`[SRS] 推流结束: app=${app}, stream=${stream}`);
|
|||
|
|
|
|||
|
|
// 更新内存存储
|
|||
|
|
const room = roomStore.setLiveStatus(stream, false);
|
|||
|
|
|
|||
|
|
if (room) {
|
|||
|
|
console.log(`[SRS] 房间 "${room.title}" 停止直播`);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 转发到 Java 后端更新数据库
|
|||
|
|
forwardToJava('/api/front/live/srs/on_unpublish', { stream });
|
|||
|
|
|
|||
|
|
res.json({ code: 0 });
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// POST /api/srs/on_play - 观看回调 (可选)
|
|||
|
|
router.post('/on_play', (req, res) => {
|
|||
|
|
const { app, stream } = req.body;
|
|||
|
|
console.log(`[SRS] 观众进入: app=${app}, stream=${stream}`);
|
|||
|
|
res.json({ code: 0 });
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// POST /api/srs/on_stop - 停止观看回调 (可选)
|
|||
|
|
router.post('/on_stop', (req, res) => {
|
|||
|
|
const { app, stream } = req.body;
|
|||
|
|
console.log(`[SRS] 观众离开: app=${app}, stream=${stream}`);
|
|||
|
|
res.json({ code: 0 });
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
module.exports = router;
|