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