闽公网安备 35020302035485号
const express = require('express')
const app = express()
const port = 3000
//允许跨域
app.all('*', function (req, res, next) {
res.setHeader("Access-Control-Allow-Origin", "*");
next();
});
// 堆代码 duidaima.com
app.get('/sse', (req, res) => {
const str = 'hello word!'
// 设置 SSE 相关的响应头
res.setHeader('Content-Type', 'text/event-stream;charset=utf-8');
res.setHeader('Cache-Control', 'no-cache');
res.setHeader('Connection', 'keep-alive');
let index = 0
const timer = setInterval(() => {
if (index < str.length) {
res.write("data: " + JSON.stringify({ content: str[index] }));
index++
} else {
// 当所有数据都发送完毕时,结束响应
clearInterval(timer); // 停止定时器
res.end();
}
}, 100);
});
app.listen(port, () => {
console.log(`Example app listening on port ${port}`)
})
因为我们是作为简单案例讲解,就直接先添加一个通用的中间件,来解决跨域问题。sse传输,我们需要在响应头中将Content-Type和Connection都设置为对应的字段,以此来保证,我们的响应能够被前端正确的处理。我们通过定时器,每次读取一个字符,来模拟每次响应的结果,读取完所有字符后,我们就将响应停止。
最终关注到了两个主要的API:EventSource和Fetch,它们分别具有不同的优势和用途。于是我们分别对两个api做了对比。
import { useEffect, useState } from "react";
import "./App.scss";
function App() {
const [chatText, setChatText] = useState("");
const getRes = async () => {
try {
const res = await fetch("http://localhost:3000/sse", {
method: "get",
});
const reader = res.body?.getReader();
let text = "";
while (reader) {
const { value, done } = await reader.read();
const chars = new TextDecoder().decode(value);
if (done) {
break;
}
const dataArray = chars.trim().split("\n\n");
const jsonObjects = dataArray.map((data) => {
const jsonString = data.substring("data: ".length);
return JSON.parse(jsonString);
});
jsonObjects.forEach((item) => {
text += item.content;
});
setChatText(text);
}
} catch (error) {
console.log("error", error);
}
};
useEffect(() => {
getRes();
}, []);
return <div>{chatText}</div>;
}
在使用fecth的过程中,我们还需要注意getReader这个方法, getReader 方法是用于处理响应体(response body)的一种方式,它返回一个可用于异步读取响应数据的 ReadableStreamDefaultReader 对象。这个方法通常用于处理大型响应或流式数据,以便在数据逐步到达时逐步处理它们,而不是一次性将整个响应数据加载到内存中。