From 1f173d482408cbeedb84d261898ebd2efb816547 Mon Sep 17 00:00:00 2001
From: “mitao” <“2763622819@qq.com”>
Date: 星期二, 14 四月 2026 09:07:07 +0800
Subject: [PATCH] feat:生产环境配置
---
tcm-training-admin-backend/src/main/java/com/tcm/training/admin/upload/UploadProgressService.java | 12 ++++++++----
1 files changed, 8 insertions(+), 4 deletions(-)
diff --git a/tcm-training-admin-backend/src/main/java/com/tcm/training/admin/upload/UploadProgressService.java b/tcm-training-admin-backend/src/main/java/com/tcm/training/admin/upload/UploadProgressService.java
index 1a3383b..604164f 100644
--- a/tcm-training-admin-backend/src/main/java/com/tcm/training/admin/upload/UploadProgressService.java
+++ b/tcm-training-admin-backend/src/main/java/com/tcm/training/admin/upload/UploadProgressService.java
@@ -12,7 +12,7 @@
private final ConcurrentMap<String, UploadProgressInfo> progressMap = new ConcurrentHashMap<>();
- public void init(String taskId) {
+ public void init(String taskId, long totalBytes) {
if (!StringUtils.hasText(taskId)) {
return;
}
@@ -21,7 +21,7 @@
info.setStatus(UploadProgressInfo.STATUS_UPLOADING);
info.setPercent(0);
info.setTransferredBytes(0L);
- info.setTotalBytes(0L);
+ info.setTotalBytes(Math.max(totalBytes, 0L));
progressMap.put(taskId, info);
}
@@ -33,16 +33,20 @@
info.setStatus(UploadProgressInfo.STATUS_UPLOADING);
info.setPercent(status.getTransferPercentage());
info.setTransferredBytes(status.getTransferredBytes());
- info.setTotalBytes(status.getTotalBytes());
+ if (status.getTotalBytes() > 0) {
+ info.setTotalBytes(status.getTotalBytes());
+ }
}
- public void markSuccess(String taskId, UploadFileResult result) {
+ public void markSuccess(String taskId, UploadFileResult result, long totalBytes) {
if (!StringUtils.hasText(taskId)) {
return;
}
UploadProgressInfo info = progressMap.computeIfAbsent(taskId, this::createUploadingInfo);
info.setStatus(UploadProgressInfo.STATUS_SUCCESS);
info.setPercent(100);
+ info.setTotalBytes(Math.max(totalBytes, 0L));
+ info.setTransferredBytes(Math.max(totalBytes, 0L));
info.setUrl(result.getUrl());
info.setFileName(result.getFileName());
}
--
Gitblit v1.7.1