@tmsfe/tms-core 0.0.197 → 0.0.199

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tmsfe/tms-core",
-  "version": "0.0.197",
+  "version": "0.0.199",
   "description": "tms运行时框架",
   "repository": {
     "type": "git",

package/src/encrypt/index.ts

@@ -61,7 +61,8 @@ function proxyWxRequest(): void {
       // 如果用户自定义了dataType或者responseType，则不做处理
       if (dataType || responseType) {
         util.reportFunc(url, traceparent, '用户自定义了dataType和responseType');
-        originalRequestApi.call(this, {
+        // 出行业务请求流式接口时要求必须传responseType，所以这里返回 wx.request 的 requestTask，用于监听流式数据
+        return originalRequestApi.call(this, {
           ...originalOptions,
           success: (res) => {
             encryptUtil.dealEncryptionSwitch(url, traceId, res.header).then((msg) => {
@@ -71,7 +72,6 @@ function proxyWxRequest(): void {
           },
           header: { ...header, Traceparent: traceparent },
         });
-        return;
       }
       // 加密请求数据
       const { data: formatData, header: formatHeader, cryptoKeyInfo } = util

package/src/request.js

@@ -191,6 +191,9 @@ export default class Request {
   secretKey = '';
   withAuth = true;
   baseParam = {};
+  enableChunked = false;
+  responseType = '';
+  enableHttp2 = false;
 
   /**
    * Request 构造函数
@@ -208,6 +211,15 @@ export default class Request {
       this.withAuth = !!config.withAuth;
     }
     this.baseParam = config.baseParam || {};
+    if (config.enableChunked) {
+      this.enableChunked = !!config.enableChunked;
+    }
+    if (config.responseType) {
+      this.responseType = config.responseType;
+    }
+    if (config.enableHttp2) {
+      this.enableHttp2 = !!config.enableHttp2;
+    }
   }
 
   /**
@@ -299,6 +311,21 @@ export default class Request {
     return this.createRequestTask(path, param, 'POST', header);
   }
 
+  /**
+   * 发送post方式的请求（返回 wx.request 的 requestTask，用于监听流式数据）
+   * @memberof Request
+   * @param {string} path 请求接口路径
+   * @param {object} params 参数
+   * @param {object} params.param 业务参数
+   * @param {string} params.method 请求方法
+   * @param {object} params.header 自定义请求头
+   * @param {object} params.callBacks 监听流式数据的回调函数
+   * @returns {Promise} Promise<WechatMiniprogram.RequestTask>
+   */
+  execPostTask(path, params) {
+    return this.createStreamRequestTask(path, params);
+  }
+
   /**
    * @memberof Request
    * @param {String} path 请求接口路径
@@ -530,8 +557,52 @@ export default class Request {
       });
     });
   }
-}
 
+  /**
+     * 创建发送请求任务（返回 requestTask，用于流式场景）
+     * 支持流式输出，同时提供 success、fail 回调
+     * @memberof Request
+     * @param {string} path 请求接口路径
+     * @param {object} params 参数对象
+     * @param {object} params.param 业务参数
+     * @param {string} params.method 请求方法 get/post
+     * @param {object} params.header 自定义的请求头
+     * @param {object} params.callbacks 回调函数对象
+     * @param {function} data.callbacks.onSuccess 请求成功回调
+     * @param {function} data.callbacks.onFail 请求失败回调
+     * @param {function} data.callbacks.onComplete 请求完成回调
+     * @returns {Promise} 返回 wx.request 的 requestTask
+     */
+  async createStreamRequestTask(path, params = {}) {
+    const { param = {}, method = 'POST', header = {}, callbacks = {} } = params;
+    const requestParam = await composeParam(param, this.withAuth, this.baseParam);
+    const data = sign(requestParam, this.secretKey);
+
+    const { onSuccess, onFail, onComplete } = callbacks;
+
+    // 创建 requestTask，支持流式输出 提供onChunkReceived监听、abort中断请求方法
+    const requestTask = wx.request({
+      url: this.makeUrl(path),
+      header,
+      method,
+      data,
+      enableHttp2: this.enableHttp2,
+      enableChunked: this.enableChunked,
+      responseType: this.responseType,
+      success: (res) => {
+        onSuccess?.(res);
+      },
+      fail: (err) => {
+        onFail?.(err);
+      },
+      complete: (res) => {
+        onComplete?.(res);
+      },
+    });
+
+    return requestTask;
+  }
+}
 /**
  * 判断指定路径是否需要token失效重试
  * @param {string} path 请求路径