阅读量:0
遇到这样一个需求,通过ai动态生成大纲文本，前端在接收到后端的文本内容时，接收的文本是“动态”的，视角表现层面上逐字生成渲染的，效果如下，那么如何实现这种效果？这里就要用到流式接口，也就是流式传输。
什么是流式传输
先说说传统传输吧，我们日常接触的大多数是传统传输方式，就也是整段传输，前后端将数据一次性传送给对方；相比于传统传输方式，流式传输则采用分段的方式将要传输的数据分层n段，后端一次性传一段给前端，直到传输完成为止，当然在传输的过程中，前端也可以提前中断传输，后端收到中断传输的消息后，也不再继续往前端传输剩下没传完的数据段。 当然，前端流式传输需要在后端支持分块传输的情况下才能实现。
前端实现流式传输的几种主流方式
查阅了资料，目前前端实现流式传输主要有以下几种方式：fetch、SSE、websocket，详情移步另一位博主，没错我查阅的资料就是他的文章，链接chatGPT流式输出前端实现fetch、SSE、websocket_fetch sse-CSDN博客
我用的是fetch，fetch 本身不直接支持流式输出，但你可以使用fetch ai 中的 ReadableStream实现流式数据处理。
什么是fetch?
简单来说fetch 是一种 HTTP 数据请求的方式， XMLHttpRequest（以下简称 XHR）的一种替代方案。与基于回调的API的XMLHttpRequest不同，fetch是基于Promise的，可以链式分块化地处理数据，更重要的是Fetch API能够处理流式响应。
更多关于fetch 的文档可以参考以下文档，我们重点讲fetch 处理流式响应
fetch 官方文档： 官方文档
fetch处理流式响应
要实现 fetch 的流式输出，关键在于如何正确地处理返回的 ReadableStream 对象。ReadableStream 是 HTML 标准的一部分，它代表了一个可以从内部读取数据的源头。在 Fetch API 中，Response 对象的 body 属性就是一个 ReadableStream 实例。
return fetch(input, { ...init, headers }) .then((res) => { console.log('res.body', res.body); if (res.ok) return Promise.resolve(res); })
实现步骤：
1.创建一个阅读器并将流锁定到它。
//创建一个阅读器并将流锁定到它。当流被锁定时，在这个阅读器发布之前，无法获得其他阅读器 const reader = res_.body.getReader(); return reader.read().then(({ value, done }) => { console.log('value', value); });
可以看到阅读器一次只接收一次响应数据，且接收到的数据为utf-8编码数据，这时我们需要让阅读器重复获取数据（发布这个阅读器），直到数据获取完或者手动中止为止，并且还要对 utf-8编码数据进行转码
2. 发布阅读器我们 可以使用递归函数去重复发布阅读器
const reader = res_.body.getReader(); return reader.read().then(function push({ value, done }) { console.log('done, value', done, value); return reader.read().then(push); });3. 转码的话我们需要使用 文本解码器（TextDecoder）进行转码 ，详情可以看我另一篇文章浅识TextDecoder-CSDN博客
const reader = res_.body.getReader(); const utf8Decoder = new TextDecoder('utf-8'); return reader.read().then(function push({ value, done }) { let _value = value ? utf8Decoder.decode(value, { stream: true }) : ''; console.log('转码后的数据', _value); return reader.read().then(push); });
4.中止fetch
要中止不完整的fetch()操作，请使用AbortController和AbortSignal接口。
定义
const controllerRef = useRef(null); controllerRef.current = new AbortController(); const signal = controllerRef.current.signal; fetch(input, { headers, signal })使用
<button onClick={() => controllerRef.current.abort()}>停止生成 </button> 5.完整代码
//存储中断器 const controllerRef = useRef(null); const getStreamData = () => { const input = 'xxxxxx'; //fetch 其它配置项 const token = utils.getToken(); const headers = { ...(token ? { Authorization: 'Bearer ' + token } : null), }; controllerRef.current = new AbortController(); const signal = controllerRef.current.signal; //存储拿到的数据 let outline = ''; fetch(input, { headers, signal }) .then((res) => { //res.ok 表示成功状态 if (res.ok) return Promise.resolve(res); }) .then((res_) => { //创建一个阅读器并将流锁定到它。当流被锁定时，在这个阅读器发布之前，无法获得其他阅读器 const reader = res_.body.getReader(); const utf8Decoder = new TextDecoder('utf-8'); //定义一个对象存储每次拿到的数据 return reader.read().then(function push({ value, done }) { if (done) { //表示数据流结束 return outline; } else { let _value = value ? utf8Decoder.decode(value, { stream: true }) : ''; let _data = _value?.split('\n\n') || []; for (const it of _data) { let _it = it.replace(/^(data: \[DONE\])|^(data:)/, ''); if (_it) { let res: { code; data; msg } = JSON.parse(_it); if (res?.code == 0) { outline = outline + res?.data?.text; console.log('接收到的数据为:', outline); } else { return Promise.reject(res); } } } } return reader.read().then(push); }); }); };
