chatGPT前端流式输出js实现三种方法—fetch、SSE、websocket

function streamOutput(msg) {
  // 发送 POST 请求
  fetch('url', {
    method:"POST",
    body:JSON.stringify({ "content": msg}),
    timeout: 0,
    dataType:"text/event-stream",
    headers:{
      "Content-Type":"application/json"
    },
  }).then(response => {
    // 检查响应是否成功
    if (!response.ok) {
      throw new Error('Network response was not ok');
    }
    // 返回一个可读流
    return response.body;
  }).then(body => {
    disableLoading();
    const reader = body.getReader();
    // 读取数据流
    function read() {
      return reader.read().then(({ done, value }) => {
        // 检查是否读取完毕
        if (done) {
          console.log('已传输完毕');
          return;
        }
        // 处理每个数据块
        console.log('收到的数据:', value);
        // 继续读取下一个数据块
        read();
      });
    }
    // 开始读取数据流
    read();
  }).catch(error => {console.error('Fetch error:', error);});
}

npm install event-source-polyfill

yarn add event-source-polyfill

import EventSource from 'event-source-polyfill';

<script type="text/javascript" src="/path/eventsource.js"></script>

function streamOutput() {
	  // 创建 EventSourcePolyfill连接，如果不需要发送请求头可以使用EventSource即可
	const eventSource = new EventSourcePolyfill('url',{
		headers:{
      		"Content-Type":"application/json"
    	}
	});
	// 处理 SSE 消息
	eventSource.onmessage = function (event) {
	  console.log('接收SSE的消息:', event.data);
	  // 在这里处理接收到的流式数据
	};
	// 处理 SSE 连接打开事件
	eventSource.onopen = function (event) {
	  console.log('SSE连接完成:', event);
	};
	// 处理 SSE 连接关闭事件
	eventSource.onclose = function (event) {
	  console.log('SSE连接关闭:', event);
	};
	// 处理 SSE 错误事件
	eventSource.onerror = function (error) {
	  console.error('SSE EventSource error:', error);
	};
}

function streamOutput(msg) {
	 const socket = new WebSocket('url');
	// 连接打开时触发
	socket.addEventListener('open', event => {
	  console.log('WebSocket连接完成:', event);
	  // 处理接收到的消息
	  socket.addEventListener('message', event => {
	    console.log('接收消息:', event.data);
	    // 在这里处理接收到的流式数据
	  });
	});
	// 连接关闭时触发
	socket.addEventListener('close', event => {
	  console.log('WebSocket连接关闭:', event);
	});
	// 处理错误时触发
	socket.addEventListener('error', error => {
	  console.error('WebSocket error:', error);
	});
}