luoyb
2021-03-09 429169689f28647561f775d17eb5250977f7da8c
febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/FilesUploadServiceImpl.java
@@ -2,16 +2,19 @@
import java.io.*;
import java.util.*;
import java.util.stream.Collectors;
import cc.mrbird.febs.common.core.constant.ModuleCode;
import cc.mrbird.febs.common.core.exception.FebsException;
import cc.mrbird.febs.common.core.utils.FebsUtil;
import cc.mrbird.febs.common.core.utils.MyUtil;
import cc.mrbird.febs.common.core.utils.SequenceUtil;
import cc.mrbird.febs.server.hr.entity.EmpBaseInfo;
import cc.mrbird.febs.server.hr.entity.FilesUpload;
import cc.mrbird.febs.server.hr.entity.Folder;
import cc.mrbird.febs.server.hr.mapper.FilesUploadMapper;
import cc.mrbird.febs.server.hr.properties.FebsServerHrProperties;
import cc.mrbird.febs.server.hr.service.IEmpBaseInfoService;
import cc.mrbird.febs.server.hr.service.IFilesUploadService;
import cc.mrbird.febs.server.hr.service.IFolderService;
import cn.hutool.core.io.FileUtil;
@@ -47,6 +50,7 @@
    private final IFolderService iFolderService;
    private final FebsServerHrProperties properties;
    private final IEmpBaseInfoService empBaseInfoService;
    private final String operatorId = Optional.ofNullable(FebsUtil.getCurrentUser())
            .map(u -> u.getUserId().toString())
            .orElse("1");
@@ -108,11 +112,10 @@
        String[] arr = fileids.split(",");
        LambdaUpdateWrapper<FilesUpload> wapper = new LambdaUpdateWrapper<>();
        wapper.in(FilesUpload::getFilesid, arr);
        FilesUpload filesUpload = new FilesUpload();
        filesUpload.setModifier(operatorId);
        filesUpload.setModifytime(new Date());
        filesUpload.setDelFlag(1);
        this.update(filesUpload,wapper);
        wapper.set(FilesUpload::getDelFlag, 1);
        wapper.set(FilesUpload::getModifier, operatorId);
        wapper.set(FilesUpload::getModifytime, new Date());
        this.update(wapper);
        LambdaQueryWrapper<FilesUpload> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.in(FilesUpload::getFilesid, arr);
@@ -129,18 +132,24 @@
        String[] arr = fileids.split(",");
        List<File> files = new ArrayList<>();
        List<String> fileName = new ArrayList<>();
        Arrays.stream(arr).forEach(i -> {
            FilesUpload filesUpload = this.getById(i);
            Folder folder = iFolderService.getById(filesUpload.getFolderid());
            files.add(new File(properties.getUploadCommonPath() + folder.getFoldername() + "/" + filesUpload.getFilesaddress()));
            fileName.add(filesUpload.getFilesname());
            if (fileName.parallelStream().filter(filter -> StrUtil.equals(filesUpload.getFilesname(), filter)).count() == 0) {
                fileName.add(filesUpload.getFilesname());
            } else {
                fileName.add(filesUpload.getFilesname() + fileName.parallelStream().filter(filter -> StrUtil.equals(filesUpload.getFilesname(), filter)).count() + 1);
            }
        });
        MyUtil.download(request, response, files, fileName);
    }
    @Override
    public void mvFiles(String fileids,Long folderid) {
    public void mvFiles(String fileids, Long folderid) {
        String[] arr = fileids.split(",");
        LambdaQueryWrapper<FilesUpload> queryWrapper = new LambdaQueryWrapper<>();
@@ -151,18 +160,19 @@
            Folder newfolder = iFolderService.getById(folderid);
            File oldfile = new File(properties.getUploadCommonPath() + folder.getFoldername() + "/" + i.getFilesaddress());
            File newfile = new File(properties.getUploadCommonPath() + newfolder.getFoldername() + "/" + i.getFilesaddress());
            FileUtil.move(oldfile,newfile,true);
            FileUtil.move(oldfile, newfile, true);
        });
        LambdaUpdateWrapper<FilesUpload> wapper = new LambdaUpdateWrapper<>();
        wapper.in(FilesUpload::getFilesid, arr);
        FilesUpload filesUpload = new FilesUpload();
        filesUpload.setModifier(operatorId);
        filesUpload.setModifytime(new Date());
        filesUpload.setFolderid(folderid);
        this.update(filesUpload,wapper);
        wapper.set(FilesUpload::getFolderid, fileids);
        wapper.set(FilesUpload::getModifier, operatorId);
        wapper.set(FilesUpload::getModifytime, new Date());
        this.update(wapper);
    }
}