初始化SSE服务端项目

This commit is contained in:
2025-07-01 20:07:24 +08:00
commit 2f899e8792
11 changed files with 1199 additions and 0 deletions

37
routes/push.js Normal file
View File

@ -0,0 +1,37 @@
/**
* @file SSE流路由处理器
* @author Yk <yk_9001@icloud.com>
* @createdAt 2025-07-01
* @lastModifiedAt 2025-07-01
* @description 处理SSE连接
*/
const express = require('express');
const clients = require('../lib/clients');
const router = express.Router();
router.post('/', (req, res) => {
const { message, clientId } = req.body;
if (!message) {
return res.status(400).json({ error: 'Message is required' });
}
const data = { message, time: new Date().toISOString() };
if (clientId) {
const clientRes = clients.get(clientId);
if (!clientRes) {
return res.status(404).json({ error: 'Client not found' });
}
clientRes.write(`data: ${JSON.stringify(data)}\n\n`);
return res.json({ message: 'Message delivered', clientId });
}
const count = clients.broadcast(data);
return res.json({ message: 'Message broadcasted', clients: count });
});
module.exports = router;

35
routes/stream.js Normal file
View File

@ -0,0 +1,35 @@
/**
* @file SSE流路由处理器
* @author Yk <yk_9001@icloud.com>
* @createdAt 2025-07-01
* @lastModifiedAt 2025-07-01
* @description 处理SSE连接建立和维持
*/
const express = require('express');
const { v4: uuidV4 } = require('uuid');
const { setupSSEHeaders } = require('../lib/sse');
const clients = require('../lib/clients');
const router = express.Router();
router.get('/', (req, res) => {
const clientId = req.query.clientId || uuidV4();
setupSSEHeaders(res);
// 立即发送确认
res.write(`data: ${JSON.stringify({
status: 'connected',
clientId,
time: new Date().toISOString()
})}\n\n`);
clients.add(clientId, res);
req.on('close', () => {
clients.remove(clientId);
});
});
module.exports = router;