示例代码

import * as CryptoJS from 'crypto-js';

export function getTime() {
  const timestamp = new Date().getTime();
  const timestampShort = timestamp.toString().substring(0, 10);
  return parseInt(timestampShort);
}

export function calcSignature(appkey: string, timestamp: number, bodyStr: any) {
  const stringToCheckSum = bodyStr + appkey + timestamp;
  const signature = CryptoJS.SHA512(stringToCheckSum).toString(CryptoJS.enc.Hex);
  return signature;
}

// openApiConfig的数据从接口获取
const openApiConfig = {
  key: 'test',
  secret: 'test',
  chatUri:
    'https://dhb-openapi-int.xiaoice.com/api/v1/openapi/organizations/vhb-org-xxxx/bots/xxxx/sessions/<session_id>/streamChat',
};

const sessionId = '7af3b9d3-34e2-45f4-9475-315a7e2d6515';

export const streamAction = (data: any) => {
  data.userId = 'fake_user_id';
  const key = openApiConfig?.key;
  const secret = openApiConfig?.secret;
  const timestamp = getTime();
  const signature = calcSignature(secret, timestamp, JSON.stringify(data));

  return () =>
    fetch(`${openApiConfig?.chatUri.replace('<session_id>', sessionId || '')}`, {
      method: 'POST',
      body: JSON.stringify(data),
      headers: {
        key,
        timestamp: timestamp.toString(),
        signature: signature,
      },
    });
};

export const initChat = async (
  fetchAction: () => Promise<any>,
  {
    onError,
    onCloseMessage,
  }: {
    onError?: (err: any) => void;
    onCloseMessage?: (result: string) => void;
  } = {}
) => {
  const res = await fetchAction();
  const reader = res.body.getReader();
  let buffer = '';
  let bufferObj: Record<string, any>;
  const decoder = new TextDecoder('utf-8');

  function read() {
    let hasError = false;
    reader?.read().then((result: any) => {
      if (result.done) {
        return;
      }
      if (res.status !== 200 || !res.body) {
        hasError = true;
        onError?.(JSON.parse(decoder.decode(result.value)));
        return;
      }
      buffer += decoder.decode(result.value, { stream: true });
      try {
        const lines = buffer.split('\n\n');
        lines.forEach((message) => {
          if (message.includes('event: close')) {
            try {
              onCloseMessage?.(JSON.parse(message.split('\n')[2].substring(6)));
            } catch {}
          }
          if (message.includes('event: message')) {
            try {
              bufferObj = JSON.parse(message.split('\n')[2].substring(6)) as Record<string, any>;
              console.log(8888, bufferObj);
            } catch (err) {
              // return;
            }
          }
        });
        buffer = lines[lines.length - 1];
      } catch (err) {
        console.log('hasError', err);
        hasError = true;
        return;
      }
      if (!hasError) read();
    });
  }
  read();
};

export const startChat = () => {
  initChat(
    streamAction({
      query: '你好',
    })
  );
};