luoyb
2021-02-19 f52362127751004c996ca9ee2f6f87d7e010b424
febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/EmpAccessoryServiceImpl.java
@@ -12,11 +12,13 @@
import cc.mrbird.febs.server.hr.service.IEmpBaseInfoService;
import cc.mrbird.febs.server.hr.service.ILabelService;
import cc.mrbird.febs.server.hr.vo.EmpAccessoryVO;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.lang.Dict;
import cn.hutool.core.lang.Snowflake;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.ZipUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import org.springframework.stereotype.Service;
@@ -46,7 +48,7 @@
@RequiredArgsConstructor
@Transactional(propagation = Propagation.SUPPORTS, readOnly = true)
public class EmpAccessoryServiceImpl extends ServiceImpl<EmpAccessoryMapper, EmpAccessory> implements IEmpAccessoryService {
    private final ILabelService labelService;
    private final EmpAccessoryMapper empAccessoryMapper;
    private final FebsServerHrProperties properties;
    private final String operatorId = Optional.ofNullable(FebsUtil.getCurrentUser())
@@ -205,4 +207,52 @@
    }
    @Override
    public void mvFiles(String accessoryids,Long labelid,Long empId) {
        String[] arr = accessoryids.split(",");
        LambdaUpdateWrapper<EmpAccessory> wapper = new LambdaUpdateWrapper<>();
        wapper.in(EmpAccessory::getAccessoryid, arr);
        EmpAccessory empAccessory = new EmpAccessory();
        empAccessory.setModifier(operatorId);
        empAccessory.setModifytime(new Date());
        empAccessory.setLabelid(labelid);
        empAccessory.setEmpid(empId);
        this.update(empAccessory,wapper);
    }
    @Override
    public void singledownload(String empIds, Long labelid, HttpServletRequest request, HttpServletResponse response) throws IOException {
        String[] arr = empIds.split(",");
        //主文件夹
        String mianFileStr = properties.getUploadSinglePath()+snowflake.nextIdStr()+"/";
        FileUtil.mkdir(mianFileStr);
        Arrays.stream(arr).forEach(i -> {
            LambdaQueryWrapper<EmpAccessory> lambdaQueryWrapper = new LambdaQueryWrapper<>();
            lambdaQueryWrapper.eq(EmpAccessory::getEmpid,i);
            lambdaQueryWrapper.eq(EmpAccessory::getDelFlag,0);
            if (!labelid.equals(-1L)){
                lambdaQueryWrapper.eq(EmpAccessory::getLabelid,labelid);
            }
            List<EmpAccessory> accessory = this.list(lambdaQueryWrapper);
            EmpBaseInfo empBaseInfo = empBaseInfoService.getById(i);
            String nameAndCertificateNumb = empBaseInfo.getEmpName()+"_"+empBaseInfo.getCertificateNumb()+"/";
            //创建个人文件夹
            String singlefileStr = mianFileStr+nameAndCertificateNumb;
            FileUtil.mkdir(singlefileStr);
            accessory.stream().forEach(a->{
                Label label = labelService.getById(a.getLabelid());
                FileUtil.copy(properties.getUploadSinglePath() + nameAndCertificateNumb + a.getFilesaddress(),singlefileStr+label.getLabelname()+"/"+a.getFilesname(),true);
            });
        });
        ZipUtil.zip(mianFileStr);
        try {
            MyUtil.downloadFile(new File(new StringBuilder().append(mianFileStr, 0, mianFileStr.length() - 1).append(".zip").toString()),response);
        } catch (IOException e) {
            log.error("下载文件异常",e);
        }
        FileUtil.del(new StringBuilder().append(mianFileStr, 0, mianFileStr.length() - 1).append(".zip").toString());
        FileUtil.del(mianFileStr);
    }
}