jp2demo/src/main/java/com/docus/demo/service/SyncBasicFileImpl.java

package com.docus.demo.service;

import cn.hutool.core.util.ObjectUtil;
import com.docus.demo.dto.BasicSelectDto;
import com.docus.demo.dto.SyncFileDto;
import com.docus.demo.entity.CommonResult;
import com.docus.demo.entity.ScanAssort;
import com.docus.demo.entity.TPicture;
import com.docus.demo.entity.ZdAssort;
import com.docus.demo.facade.ISyncBasicFileService;
import com.docus.demo.mapper.mysql.BasicMapper;
import com.docus.demo.mapper.mysql.ScanAssortMapper;
import com.docus.demo.mapper.sqlserver.PictureMapper;
import com.docus.demo.utils.ImageUtils;
import com.docus.demo.utils.SnowflakeIdWorker;
import com.docus.demo.vo.BasicVo;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.io.File;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.stream.Collectors;

@Slf4j
@Service
public class SyncBasicFileImpl implements ISyncBasicFileService {

    @Autowired
    private BasicMapper basicMapper;
    @Autowired
    private PictureMapper pictureMapper;
    @Autowired
    private ScanAssortMapper scanAssortMapper;

    private SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd");

    private final ExecutorService executor = Executors.newFixedThreadPool(7);


    @Override
    public CommonResult<?> syncBasicFile(SyncFileDto syncFileDto) {

        CompletableFuture.runAsync(() -> {
            try {
                //页码
                int current;
                //每页1000条数据
                int limit = syncFileDto.getLimit();
                //1.先去tbaisc查数据 查询出带cid的数据
                BasicSelectDto basicSelectDto = this.getBaiscSelectDto(syncFileDto);
                List<BasicVo> basicVoList;
                for (current = 1; ; current++) {
                    basicVoList = basicMapper.getBasicVoList((current-1)*limit, limit, basicSelectDto);
                    if (null == basicVoList || basicVoList.size() == 0) {
                        break;
                    }
                    List<ScanAssort> updateOrInsertList = new ArrayList<>(5000);
                    for (BasicVo basicVo : basicVoList) {
                        log.info("开始同步"+basicVo.getInpatientNo());
                        List<ScanAssort> scanAssortList = this.doSyncFile(basicVo);
                        //2.5数据入库
                        if (scanAssortList.size() != 0) {
                            List<ScanAssort> oldScanAssort = scanAssortMapper.getListByPid(basicVo.getPatientId());
                            //根据pid 查询数据库 根据路径和名称做匹配  如果数据存在需要做幂等  不存在需要做数据更新
                            List<ScanAssort> addScanAssortList = this.handleUpdateOrInsert(oldScanAssort,scanAssortList);
                            updateOrInsertList.addAll(addScanAssortList);
                        } else {
                            log.info(basicVo.getPatientId() + " canFindImage ");
                        }
                        log.info("结束同步"+basicVo.getInpatientNo());
                    }
                    scanAssortMapper.insertOrUpdateBatch(updateOrInsertList);
                }
            }catch (Exception e){
                log.info(e.getMessage());
            }
        },executor);

        return CommonResult.success("同步成功");
    }


    private List<ScanAssort> handleUpdateOrInsert(List<ScanAssort> oldScanAssort, List<ScanAssort> scanAssortList) {
        //如果没有旧数据 做全量插入
        if (ObjectUtil.isEmpty(oldScanAssort)){
            return scanAssortList;
        }

        List<ScanAssort> updateOrInsertList = scanAssortList.stream().map(m->{
            //根据路径和文件名称去旧数据做判断 如果存在则需要做更新
            ScanAssort scanAssort = oldScanAssort.stream()
                    .filter(f->ObjectUtil.equal(m.getImagePath(),f.getImagePath())&&ObjectUtil.equal(m.getScanPage(),f.getScanPage()))
                    .findAny()
                    .orElse(null);
            if (scanAssort!=null){
                m.setId(scanAssort.getId());
            }
            return m;
        }).collect(Collectors.toList());
        //做新增或者插入
        return updateOrInsertList;
    }

    private List<ScanAssort> doSyncFile(BasicVo basicVo) {
        String[] disDate = dateFormat.format(basicVo.getDisDate()).split("-");
        String cyYear = disDate[0];
        String cyMonth = disDate[1];
        String cyDay = disDate[2];
        StringBuilder inpatientNo = new StringBuilder(basicVo.getInpatientNo());
        for (int s =inpatientNo.length() ;s<8;s++){
            inpatientNo.insert(0, "0");
        }

        String rootDir = "Z:\\"
                + cyYear + File.separator
                + cyYear + cyMonth + File.separator
                + cyYear + cyMonth + cyDay + File.separator
                + inpatientNo + cyYear + cyMonth + cyDay + File.separator;
        String outDir = "F:\\lianzhong" + File.separator
                + cyYear + File.separator
                + cyMonth + File.separator
                + cyDay + File.separator
                + basicVo.getInpatientNo();


        String cuid = basicVo.getBColumn5();
        List<ScanAssort> scanAssortList = new ArrayList<>(100);
        //      2.根据cid查询联众表t_picture  走索引大概零点几秒
        List<TPicture> tPictureList = pictureMapper.getPictureInfoByCid(cuid);
        //      2.1根据病案信息拼接地址
        int sort = 0;
        for (TPicture tpicture : tPictureList) {

            String picName = tpicture.getPicName().split(".jpg")[0];
            String inPutFile = rootDir + picName + ".jp2";
            String outFile = outDir + File.separator + picName + ".jpg";

            File file  = new File(inPutFile);
            if (!file.exists()){
                log.info(basicVo.getInpatientNo()+"文件未找到"+inPutFile);
            }
            boolean savePicFlag ;
            boolean rotateFlag ;
            //     2.2jp2转化jpg  图片通过文件流写到挂在的盘符
            savePicFlag =  ImageUtils.getInstance().savePic(inPutFile, outFile, tpicture.getRotateDegree());
            if (!savePicFlag){
                //需要同步tif文件成jpg过来
                inPutFile  = rootDir + picName + ".tif";
                savePicFlag =  ImageUtils.getInstance().savePic(inPutFile, outFile, tpicture.getRotateDegree());
            }
            //       2.4组合文件信息
            ScanAssort scanAssort = new ScanAssort();
            scanAssort.setId(SnowflakeIdWorker.idWorker.nextId());
            scanAssort.setPatientId(basicVo.getPatientId());
            scanAssort.setAssortId(tpicture.getPicKind());
            scanAssort.setFileTitle(picName);
            scanAssort.setImagePath(outDir);
            scanAssort.setCreateTime(new Date());
            scanAssort.setScanPage(picName + ".jpg");
            scanAssort.setSort(sort);
            scanAssort.setFileSource(2);
            scanAssort.setFileStorageType(1);
            scanAssort.setFilePages(1);

            if (savePicFlag){
                scanAssort.setCreater("auto2");
            }else {
                scanAssort.setCreater("auto4");
            }

            scanAssortList.add(scanAssort);
            sort+=1;
        }


        return scanAssortList;
    }

    private BasicSelectDto getBaiscSelectDto(SyncFileDto syncFileDto) {

        List<String> inpatientNoList = Arrays.asList(syncFileDto.getInpatientNo().split(","));

        BasicSelectDto basicSelectDto = new BasicSelectDto();
        basicSelectDto.setStartDate(syncFileDto.getStartDate());
        basicSelectDto.setEndDate(syncFileDto.getEndDate());
        basicSelectDto.setInpatientNoList(inpatientNoList);

        return basicSelectDto;
    }

    private String getRootDir(BasicVo basicInfo, String cyYear, String cyMonth, String cyDay) {
        String inpatientNo = basicInfo.getInpatientNo().trim();
        String rootDir = "";

         rootDir = "Z:\\"
                + cyYear + File.separator
                + cyYear + cyMonth + File.separator
                + cyYear + cyMonth + cyDay;
        File file = new File(rootDir);
        File[] subdirectories = file.listFiles(File::isDirectory);
        if (subdirectories != null) {
            //判断当前日期下是否有路径
            for (File subFile : subdirectories) {
                if (subFile.getName().contains(inpatientNo)) {
                    rootDir = rootDir + File.separator + subFile.getName() + File.separator;
                    return rootDir;
                }
            }
        }

        for (int day = 1; day < 32; day++) {
            String otherDay = String.format("%02d", day);
            rootDir = "Z:\\"
                    + cyYear + File.separator
                    + cyYear + cyMonth + File.separator
                    + cyYear + cyMonth + otherDay;
            file = new File(rootDir);
            subdirectories = file.listFiles(File::isDirectory);
            if (subdirectories != null) {
                //判断当前日期下是否有路径
                for (File subFile : subdirectories) {
                    if (subFile.getName().contains(inpatientNo)) {
                        rootDir = rootDir + File.separator + subFile.getName() + File.separator;
                        return rootDir;
                    }
                }
            }
        }


        return rootDir;
    }
}