微信小程序对接SSE接口记录

微信小程序对接SSE接口记录

• 需求：公司项目对接gpt，gpt产生的结果是分段返回，所以要求在产生结果时，有打字机的效果。原本是由定时器调用，后来优化改为服务端使用SSE接口。
• 小程序使用起来比较方便，但是要求小程序基本库的版本需要在2.20.2以上。文档地址移步这里
• 微信小程序代码

   // 基础库为2.33.0const requestTask = wx.request({url: `xxxxxxxx`, // 需要请求的接口地址enableChunked: true // enableChunked必须为true})// 开发工具存在问题，使用真机测试const listener = data => {// data为返回的数据，可以在此对数据进行处理}// 监听服务端返回的数据requestTask.onChunkReceived(listener)// 移除监听 需传入与监听时同一个的函数对象requestTask.offChunkReceived(listener)

注意：

1. 接收到的结果数据类型固定为arrayBuffer，需要开发者自己进行转换，可以使用下面方法进行转换。

function arrayBufferToString(arr){if(typeof arr === 'string') {  return arr;  }  var dataview = new DataView(arr);var ints = new Uint8Array(arr.byteLength);for(var i=0;i<ints.length;i++){ints[i]=dataview.getUint8(i);}var str = '',  _arr = ints;  for(var i = 0; i < _arr.length; i++) {if (_arr[i]) {var one = _arr[i].toString(2),  v = one.match(/^1+?(?=0)/);  if(v && one.length == 8) {var bytesLength = v[0].length;var store = _arr[i].toString(2).slice(7 - bytesLength);  for(var st = 1; st < bytesLength; st++) {  if ( _arr[st + i]) {store += _arr[st + i].toString(2).slice(2);  }}  str += String.fromCharCode(parseInt(store, 2));  i += bytesLength - 1;  } else {  str += String.fromCharCode(_arr[i]);  }  }}  return str; 
}

2. 微信开发工具中无法转换数据。可能是由于开发工具问题，在服务端返回的字符串中存在中文时，开发工具是无法正常转换的。但是在真机是正常的。如果需要在开发工具中实现转换，可以与服务端协商将数据进行URL编码返回。
3. 服务端一次返回的结果，微信小程序有时会将其截开，并分两次返回。由于截开的位置并不固定，所以可能会存在转换ArrayBuffer时，出现结果异常的问题。使用SSE接口一般有两种需求：一种是将所有的结果累加起来、还有一种就是后面的结果覆盖前面的。在使用第一种时，每次的返回量不会太大，所以应该不会出现微信小程序截开两次返回的情况。但是第二种每次返回的接口都在逐渐增大，可能会出现这种情况，我就是第二种。我是使用下面方法解决

// 我的数据是json字符串，如果出现分开返回，在转json时，会出现报错，所以使用try处理
let timer = null
const listener = data => {try {// 上次结果出现报错 这次正常 清除延时器if (timer) {clearTimeout(timer)timer = null}// 小程序存在数据截开的情况  存五次数据if (arr.length > 4) {arr.shift()}// 这里要存储的是arrayBuffer，不能存储string数据arr.push(data.data)// 数据处理 .......} catch (e) {// 最后一次出现报错 三秒后重组数据timer = setTimeout(() => {const len = arr.lengthlet index = len - 2,data = arr[len - 1],result = nullwhile(index > -1) {// 从后往前 合并data = mergeArrayBuffers(arr[index], data)try{// 数据处理 .......index = -1}catch(e){index -= 1}}}, 3000)}}

// 合并arrayBuffer
function mergeArrayBuffers(buffer1, buffer2) {if (!buffer1) {return buffer2;} else if (!buffer2) {return buffer1;}var tmp = new Uint8Array(buffer1.byteLength + buffer2.byteLength);tmp.set(new Uint8Array(buffer1), 0);tmp.set(new Uint8Array(buffer2), buffer1.byteLength);return tmp.buffer;
}

• 由于是后面的结果覆盖前面的，我只需要处理最后一次结果，如果结果正常则不用处理。不正常再将前面存储的数据一一合并，再做处理。
• 通常一个中文是两个字节，所以可能会出现一个中文恰好被截开的情况，所以需要存储的是原数据

4. 由于SSE特性，需要由用户端断开连接，所以在使用完毕时，需要调用requestTask.abort()断开连接
   ps: 此文章做个人平常记录