ls-pro-common 3.1.58 → 3.1.60

package/lib/http/uploadByChunk.js

@@ -0,0 +1,385 @@
+import _asyncToGenerator from "@babel/runtime/helpers/esm/asyncToGenerator";
+import _regeneratorRuntime from "@babel/runtime/regenerator";
+import { httpUpload } from './upload';
+import { httpPost } from './ajax';
+import { urlDownloadDomain, setUrlQuery } from '../utils';
+/** 分片上传接口地址 */
+var BASE_URL = '/petrel/petrel-file-center-api/lesoon/oss/file/chunk/upload';
+/**
+ * 分片上传文件
+ *
+ * @param config 分片上传配置
+ * @returns 返回包含 promise、cancel 方法和 uploadId 的对象
+ */
+export var httpUploadByChunk = function httpUploadByChunk(config) {
+  var file = config.file,
+    createTransactionUrl = config.createTransactionUrl,
+    uploadChunkUrl = config.uploadChunkUrl,
+    commitUrl = config.commitUrl,
+    cancelUrl = config.cancelUrl,
+    _config$chunkSize = config.chunkSize,
+    chunkSizeMB = _config$chunkSize === void 0 ? 2 : _config$chunkSize,
+    _config$needGateWay = config.needGateWay,
+    needGateWay = _config$needGateWay === void 0 ? true : _config$needGateWay,
+    _config$useDownloadDo = config.useDownloadDomain,
+    useDownloadDomain = _config$useDownloadDo === void 0 ? true : _config$useDownloadDo,
+    onProgress = config.onProgress,
+    _config$maxRetries = config.maxRetries,
+    maxRetries = _config$maxRetries === void 0 ? 3 : _config$maxRetries,
+    _config$code = config.code,
+    code = _config$code === void 0 ? 'RETAIL' : _config$code,
+    _config$catalogName = config.catalogName,
+    catalogName = _config$catalogName === void 0 ? 'mdm' : _config$catalogName,
+    _config$batchSize = config.batchSize,
+    batchSize = _config$batchSize === void 0 ? 1 : _config$batchSize;
+  var uploadId;
+  var isCancelled = false;
+  var cancelFunctions = [];
+  var chunkSize = chunkSizeMB * 1024 * 1024;
+  // 计算总分片数
+  var totalChunks = Math.ceil(file.size / chunkSize);
+  // 进度管理器：记录每个分片的上传进度（0-1）
+  var chunkProgressMap = new Map();
+  var lastReportedProgress = 0;
+  /** 更新并报告进度，确保进度只增不减 */
+  var updateProgress = function updateProgress() {
+    // 计算所有分片的累计进度
+    var totalProgress = 0;
+    chunkProgressMap.forEach(function (progress) {
+      totalProgress += progress;
+    });
+    // 计算总体进度百分比（0-100）
+    var currentProgress = Math.min(Math.round(totalProgress / totalChunks * 100), 100);
+    // 确保进度只增不减
+    var finalProgress = Math.max(currentProgress, lastReportedProgress);
+    // 只有当进度增加时才回调
+    if (finalProgress > lastReportedProgress) {
+      lastReportedProgress = finalProgress;
+      onProgress === null || onProgress === void 0 ? void 0 : onProgress(finalProgress);
+    }
+  };
+  /** 创建上传事务 */
+  var createTransaction = /*#__PURE__*/function () {
+    var _ref = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee() {
+      var _result$flag;
+      var params, api, finalUrl, result, _result$flag2;
+      return _regeneratorRuntime.wrap(function _callee$(_context) {
+        while (1) switch (_context.prev = _context.next) {
+          case 0:
+            params = {
+              chunkNumbers: totalChunks,
+              fileName: encodeURIComponent(file.name),
+              contentType: encodeURIComponent(file.type),
+              fileSize: file.size
+            };
+            api = setUrlQuery(createTransactionUrl || "".concat(BASE_URL, "/transaction/").concat(code, "?catalogName=").concat(catalogName), params);
+            finalUrl = useDownloadDomain ? urlDownloadDomain(api) : api;
+            _context.next = 5;
+            return httpPost(finalUrl, {}, true, needGateWay);
+          case 5:
+            result = _context.sent;
+            if (!(((_result$flag = result.flag) === null || _result$flag === void 0 ? void 0 : _result$flag.retCode) === '0' && result.data)) {
+              _context.next = 10;
+              break;
+            }
+            return _context.abrupt("return", result.data);
+          case 10:
+            throw new Error(((_result$flag2 = result.flag) === null || _result$flag2 === void 0 ? void 0 : _result$flag2.retMsg) || '创建上传事务失败');
+          case 11:
+          case "end":
+            return _context.stop();
+        }
+      }, _callee);
+    }));
+    return function createTransaction() {
+      return _ref.apply(this, arguments);
+    };
+  }();
+  /**
+   * 上传单个分片
+   *
+   * @param chunk Blob 分片
+   * @param chunkIndex 分片索引（从 1 开始）
+   * @param retryCount 当前重试次数
+   */
+  var uploadChunk = /*#__PURE__*/function () {
+    var _ref2 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee2(chunk, chunkIndex) {
+      var retryCount,
+        params,
+        api,
+        _result$flag3,
+        uploadResult,
+        result,
+        _result$flag4,
+        _args2 = arguments;
+      return _regeneratorRuntime.wrap(function _callee2$(_context2) {
+        while (1) switch (_context2.prev = _context2.next) {
+          case 0:
+            retryCount = _args2.length > 2 && _args2[2] !== undefined ? _args2[2] : 0;
+            if (!isCancelled) {
+              _context2.next = 3;
+              break;
+            }
+            throw new Error('上传已取消');
+          case 3:
+            if (uploadId) {
+              _context2.next = 5;
+              break;
+            }
+            throw new Error('上传事务 ID 不存在');
+          case 5:
+            params = {
+              uploadId: uploadId,
+              chunkSequence: chunkIndex,
+              size: chunk.size
+            };
+            api = setUrlQuery(uploadChunkUrl || "".concat(BASE_URL, "/").concat(code, "?catalogName=").concat(catalogName), params);
+            _context2.prev = 7;
+            // 初始化分片进度为 0
+            chunkProgressMap.set(chunkIndex, 0);
+            // 使用 httpUpload 上传分片
+            uploadResult = httpUpload(api, new File([chunk], file.name, {
+              type: file.type
+            }), {}, 'file', needGateWay, function (progress) {
+              // 更新当前分片的上传进度（0-1）
+              chunkProgressMap.set(chunkIndex, progress / 100);
+              // 更新总体进度
+              updateProgress();
+            }, useDownloadDomain); // 保存取消函数
+            cancelFunctions.push(uploadResult.cancel);
+            // 等待上传完成
+            _context2.next = 13;
+            return uploadResult.promise;
+          case 13:
+            result = _context2.sent;
+            if (!(((_result$flag3 = result.flag) === null || _result$flag3 === void 0 ? void 0 : _result$flag3.retCode) === '0')) {
+              _context2.next = 19;
+              break;
+            }
+            // 分片上传成功，标记为完成（进度为 1）
+            chunkProgressMap.set(chunkIndex, 1);
+            // 更新总体进度
+            updateProgress();
+            _context2.next = 21;
+            break;
+          case 19:
+            // 上传失败，重置进度
+            chunkProgressMap.set(chunkIndex, 0);
+            throw new Error(((_result$flag4 = result.flag) === null || _result$flag4 === void 0 ? void 0 : _result$flag4.retMsg) || '分片上传失败');
+          case 21:
+            _context2.next = 30;
+            break;
+          case 23:
+            _context2.prev = 23;
+            _context2.t0 = _context2["catch"](7);
+            if (!(retryCount < maxRetries && !isCancelled)) {
+              _context2.next = 28;
+              break;
+            }
+            // 重置进度后重试
+            chunkProgressMap.set(chunkIndex, 0);
+            return _context2.abrupt("return", uploadChunk(chunk, chunkIndex, retryCount + 1));
+          case 28:
+            // 重试失败，重置进度
+            chunkProgressMap.set(chunkIndex, 0);
+            throw _context2.t0;
+          case 30:
+          case "end":
+            return _context2.stop();
+        }
+      }, _callee2, null, [[7, 23]]);
+    }));
+    return function uploadChunk(_x, _x2) {
+      return _ref2.apply(this, arguments);
+    };
+  }();
+  /** 提交合并请求 */
+  var commitUpload = /*#__PURE__*/function () {
+    var _ref3 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee3() {
+      var _result$flag5;
+      var params, api, finalUrl, result, _result$flag6;
+      return _regeneratorRuntime.wrap(function _callee3$(_context3) {
+        while (1) switch (_context3.prev = _context3.next) {
+          case 0:
+            if (uploadId) {
+              _context3.next = 2;
+              break;
+            }
+            throw new Error('上传事务 ID 不存在');
+          case 2:
+            params = {
+              uploadId: uploadId
+            };
+            api = setUrlQuery(commitUrl || "".concat(BASE_URL, "/commit/normal/").concat(code), params);
+            finalUrl = useDownloadDomain ? urlDownloadDomain(api) : api;
+            _context3.next = 7;
+            return httpPost(finalUrl, {}, true, needGateWay);
+          case 7:
+            result = _context3.sent;
+            if (!(((_result$flag5 = result.flag) === null || _result$flag5 === void 0 ? void 0 : _result$flag5.retCode) === '0')) {
+              _context3.next = 12;
+              break;
+            }
+            return _context3.abrupt("return", result);
+          case 12:
+            throw new Error(((_result$flag6 = result.flag) === null || _result$flag6 === void 0 ? void 0 : _result$flag6.retMsg) || '合并文件失败');
+          case 13:
+          case "end":
+            return _context3.stop();
+        }
+      }, _callee3);
+    }));
+    return function commitUpload() {
+      return _ref3.apply(this, arguments);
+    };
+  }();
+  /** 取消上传 */
+  var cancel = /*#__PURE__*/function () {
+    var _ref4 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee4() {
+      var api, finalUrl;
+      return _regeneratorRuntime.wrap(function _callee4$(_context4) {
+        while (1) switch (_context4.prev = _context4.next) {
+          case 0:
+            if (!isCancelled) {
+              _context4.next = 2;
+              break;
+            }
+            return _context4.abrupt("return");
+          case 2:
+            isCancelled = true;
+            // 取消所有正在上传的分片
+            cancelFunctions.forEach(function (cancelFn) {
+              try {
+                cancelFn();
+              } catch (e) {
+                // 忽略取消错误
+              }
+            });
+            cancelFunctions.length = 0;
+            // 如果已有上传事务 ID，发送取消请求
+            if (!uploadId) {
+              _context4.next = 15;
+              break;
+            }
+            _context4.prev = 6;
+            // 处理 URL 参数，支持 ? 和 & 开头
+            api = setUrlQuery(cancelUrl || "".concat(BASE_URL, "/cancel/").concat(code), {
+              uploadId: uploadId
+            });
+            finalUrl = useDownloadDomain ? urlDownloadDomain(api) : api;
+            _context4.next = 11;
+            return httpPost(finalUrl, {}, true, needGateWay);
+          case 11:
+            _context4.next = 15;
+            break;
+          case 13:
+            _context4.prev = 13;
+            _context4.t0 = _context4["catch"](6);
+          case 15:
+          case "end":
+            return _context4.stop();
+        }
+      }, _callee4, null, [[6, 13]]);
+    }));
+    return function cancel() {
+      return _ref4.apply(this, arguments);
+    };
+  }();
+  /** 执行分片上传流程 */
+  var promise = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee5() {
+    var chunks, chunkIndex, start, end, chunk, i, batch, result;
+    return _regeneratorRuntime.wrap(function _callee5$(_context5) {
+      while (1) switch (_context5.prev = _context5.next) {
+        case 0:
+          _context5.prev = 0;
+          _context5.next = 3;
+          return createTransaction();
+        case 3:
+          uploadId = _context5.sent;
+          if (!isCancelled) {
+            _context5.next = 6;
+            break;
+          }
+          throw new Error('上传已取消');
+        case 6:
+          // 第二步：计算所有分片
+          chunks = [];
+          chunkIndex = 1;
+          for (start = 0; start < file.size; start += chunkSize) {
+            end = Math.min(start + chunkSize, file.size);
+            chunk = file.slice(start, end);
+            chunks.push({
+              blob: chunk,
+              index: chunkIndex
+            });
+            chunkIndex++;
+          }
+          // 初始化所有分片的进度为 0
+          chunks.forEach(function (chunkInfo) {
+            chunkProgressMap.set(chunkInfo.index, 0);
+          });
+          // 第三步：分批并发上传分片
+          i = 0;
+        case 11:
+          if (!(i < chunks.length)) {
+            _context5.next = 20;
+            break;
+          }
+          if (!isCancelled) {
+            _context5.next = 14;
+            break;
+          }
+          throw new Error('上传已取消');
+        case 14:
+          // 获取当前批次的分片
+          batch = chunks.slice(i, i + batchSize); // 并发上传当前批次的所有分片
+          _context5.next = 17;
+          return Promise.all(batch.map(function (chunkInfo) {
+            return uploadChunk(chunkInfo.blob, chunkInfo.index);
+          }));
+        case 17:
+          i += batchSize;
+          _context5.next = 11;
+          break;
+        case 20:
+          // 确保所有分片上传完成，进度为 100%
+          if (chunks.length > 0) {
+            chunks.forEach(function (chunkInfo) {
+              chunkProgressMap.set(chunkInfo.index, 1);
+            });
+            updateProgress();
+          }
+          if (!isCancelled) {
+            _context5.next = 23;
+            break;
+          }
+          throw new Error('上传已取消');
+        case 23:
+          _context5.next = 25;
+          return commitUpload();
+        case 25:
+          result = _context5.sent;
+          // 上传完成，确保进度为 100%
+          lastReportedProgress = 100;
+          onProgress === null || onProgress === void 0 ? void 0 : onProgress(100);
+          return _context5.abrupt("return", result);
+        case 31:
+          _context5.prev = 31;
+          _context5.t0 = _context5["catch"](0);
+          // 如果出错，确保进度回调被调用
+          if (!isCancelled) {
+            onProgress === null || onProgress === void 0 ? void 0 : onProgress(0);
+          }
+          throw _context5.t0;
+        case 35:
+        case "end":
+          return _context5.stop();
+      }
+    }, _callee5, null, [[0, 31]]);
+  }))();
+  return {
+    promise: promise,
+    cancel: cancel,
+    uploadId: uploadId
+  };
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ls-pro-common",
-  "version": "3.1.58",
+  "version": "3.1.60",
   "description": "ls-pro-common",
   "license": "MIT",
   "sideEffects": [