From 957607035233956b52104bfee9a6b40ca702935b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=A9=AC=E8=B6=85?= Date: Tue, 12 May 2020 22:58:25 +0800 Subject: [PATCH] =?UTF-8?q?QW-177=20=E4=BF=AE=E5=A4=8D=E8=A7=A3=E5=8E=8B?= =?UTF-8?q?=E7=BC=A9=E5=90=8E=E7=BD=91=E7=9B=98=E5=AE=B9=E9=87=8F=E4=BF=A1?= =?UTF-8?q?=E6=81=AF=E6=B2=A1=E6=9C=89=E5=8F=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../main/java/com/mac/scp/service/FileService.java | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/file-web/src/main/java/com/mac/scp/service/FileService.java b/file-web/src/main/java/com/mac/scp/service/FileService.java index dce9047..d17f726 100644 --- a/file-web/src/main/java/com/mac/scp/service/FileService.java +++ b/file-web/src/main/java/com/mac/scp/service/FileService.java @@ -28,7 +28,21 @@ public class FileService implements IFileService { @Override public void batchInsertFile(List fileBeanList) { + UserBean sessionUserBean = (UserBean) SecurityUtils.getSubject().getPrincipal(); + StorageBean storageBean = filetransferService.selectStorageBean(new StorageBean(sessionUserBean.getUserId())); + long fileSizeSum = 0; + for (FileBean fileBean : fileBeanList) { + if (fileBean.getIsdir() == 0) { + fileSizeSum += fileBean.getFilesize(); + } + } fileMapper.batchInsertFile(fileBeanList); + if (storageBean != null) { + long updateFileSize = storageBean.getStoragesize() + fileSizeSum; + + storageBean.setStoragesize(updateFileSize); + filetransferService.updateStorageBean(storageBean); + } } @Override