feat: app module

client/src/service/api/request.ts

@@ -1,122 +1,79 @@
-import axios, { Method, InternalAxiosRequestConfig, AxiosResponse } from 'axios';
+import { sseResponseEventEnum } from '@/constants/chat';
+import { getErrText } from '@/utils/tools';
+import { parseStreamChunk } from '@/utils/adapt';
+import { NextApiResponse } from 'next';
+import { sseResponse } from '../utils/tools';
 
-interface ConfigType {
-  headers?: { [key: string]: string };
-  hold?: boolean;
-}
-interface ResponseDataType {
-  code: number;
-  message: string;
-  data: any;
+interface Props {
+  res: NextApiResponse; // 用于流转发
+  url: string;
+  data: Record<string, any>;
 }
+export const moduleFetch = ({ url, data, res }: Props) =>
+  new Promise<Record<string, any>>(async (resolve, reject) => {
+    try {
+      const baseUrl = `http://localhost:3000/api`;
+      const requestUrl = url.startsWith('/') ? `${baseUrl}${url}` : url;
+      const response = await fetch(requestUrl, {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json'
+        },
+        body: JSON.stringify(data)
+      });
 
-/**
- * 请求开始
- */
-function requestStart(config: InternalAxiosRequestConfig): InternalAxiosRequestConfig {
-  if (config.headers) {
-    config.headers.rootkey = process.env.ROOT_KEY;
-  }
+      if (!response?.body) {
+        throw new Error('Request Error');
+      }
 
-  return config;
-}
+      const responseType = response.headers.get('content-type');
+      if (responseType && responseType.includes('application/json')) {
+        const jsonResponse = await response.json();
+        return resolve(jsonResponse?.data || {});
+      }
 
-/**
- * 请求成功,检查请求头
- */
-function responseSuccess(response: AxiosResponse<ResponseDataType>) {
-  return response;
-}
-/**
- * 响应数据检查
- */
-function checkRes(data: ResponseDataType) {
-  if (data === undefined) {
-    return Promise.reject('服务器异常');
-  } else if (data.code < 200 || data.code >= 400) {
-    return Promise.reject(data);
-  }
-  return data.data;
-}
+      const reader = response.body?.getReader();
 
-/**
- * 响应错误
- */
-function responseError(err: any) {
-  if (!err) {
-    return Promise.reject({ message: '未知错误' });
-  }
-  if (typeof err === 'string') {
-    return Promise.reject({ message: err });
-  }
-  return Promise.reject(err);
-}
+      let chatResponse = {};
 
-/* 创建请求实例 */
-export const instance = axios.create({
-  timeout: 60000, // 超时时间
-  baseURL: `http://localhost:${process.env.PORT || 3000}/api`,
-  headers: {
-    rootkey: process.env.ROOT_KEY
-  }
-});
+      const read = async () => {
+        try {
+          const { done, value } = await reader.read();
+          if (done) {
+            return resolve(chatResponse);
+          }
+          const chunkResponse = parseStreamChunk(value);
 
-/* 请求拦截 */
-instance.interceptors.request.use(requestStart, (err) => Promise.reject(err));
-/* 响应拦截 */
-instance.interceptors.response.use(responseSuccess, (err) => Promise.reject(err));
-
-function request(url: string, data: any, config: ConfigType, method: Method): any {
-  /* 去空 */
-  for (const key in data) {
-    if (data[key] === null || data[key] === undefined) {
-      delete data[key];
+          chunkResponse.forEach((item) => {
+            // parse json data
+            const data = (() => {
+              try {
+                return JSON.parse(item.data);
+              } catch (error) {
+                return {};
+              }
+            })();
+            if (item.event === sseResponseEventEnum.moduleFetchResponse) {
+              chatResponse = {
+                ...chatResponse,
+                ...data
+              };
+            } else if (item.event === sseResponseEventEnum.answer && data?.choices?.[0]?.delta) {
+              sseResponse({
+                res,
+                event: sseResponseEventEnum.answer,
+                data: JSON.stringify(data)
+              });
+            }
+          });
+          read();
+        } catch (err: any) {
+          reject(getErrText(err, '请求异常'));
+        }
+      };
+      read();
+    } catch (err: any) {
+      console.log(err);
+      reject(getErrText(err, '请求异常'));
     }
-  }
-
-  return instance
-    .request({
-      url,
-      method,
-      data: method === 'GET' ? null : data,
-      params: method === 'GET' ? data : null, // get请求不携带data，params放在url上
-      ...config // 用户自定义配置，可以覆盖前面的配置
-    })
-    .then((res) => checkRes(res.data))
-    .catch((err) => responseError(err));
-}
-
-/**
- * api请求方式
- * @param {String} url
- * @param {Any} params
- * @param {Object} config
- * @returns
- */
-export function GET<T = { data: any }>(
-  url: string,
-  params = {},
-  config: ConfigType = {}
-): Promise<T> {
-  return request(url, params, config, 'GET');
-}
-
-export function POST<T = { data: any }>(
-  url: string,
-  data = {},
-  config: ConfigType = {}
-): Promise<T> {
-  return request(url, data, config, 'POST');
-}
-
-export function PUT<T = { data: any }>(
-  url: string,
-  data = {},
-  config: ConfigType = {}
-): Promise<T> {
-  return request(url, data, config, 'PUT');
-}
-
-export function DELETE<T = { data: any }>(url: string, config: ConfigType = {}): Promise<T> {
-  return request(url, {}, config, 'DELETE');
-}
+  });