industrial_measurement/src/main/java/com/xkrs/service/impl/QcSourceServiceImpl.java

package com.xkrs.service.impl;

import com.xkrs.dao.CraftItemDao;
import com.xkrs.dao.QcItemDao;
import com.xkrs.dao.QcSourceDao;
import com.xkrs.dao.QcSpecDao;
import com.xkrs.encapsulation.PromptMessageEnum;
import com.xkrs.model.entity.CraftItemEntity;
import com.xkrs.model.entity.QcItemEntity;
import com.xkrs.model.entity.QcSourceEntity;
import com.xkrs.model.entity.QcSpecEntity;
import com.xkrs.model.qo.ExportExcelQo;
import com.xkrs.model.qo.QcSourceQoDelete;
import com.xkrs.model.qo.QcSourceQoInsert;
import com.xkrs.service.QcSourceService;
import com.xkrs.util.Constant;
import com.xkrs.util.ExcelUploadUtil;
import com.xkrs.util.LocalDateUtils;
import com.xkrs.util.LocalStringUtils;
import org.apache.http.util.TextUtils;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.util.HSSFColor;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xssf.usermodel.*;
import org.springframework.context.i18n.LocaleContextHolder;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import javax.transaction.Transactional;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.*;

import static com.xkrs.encapsulation.OutputEncapsulation.outputEncapsulationObject;

@Service
public class QcSourceServiceImpl implements QcSourceService {

    private Locale locale = LocaleContextHolder.getLocale();

    @Resource
    private CraftItemDao craftItemDao;

    @Resource
    private QcItemDao qcItemDao;

    @Resource
    private QcSourceDao qcSourceDao;

    @Resource
    private QcSpecDao qcSpecDao;

    /**
     * 导出excel样式一（表头顺序）
     * 批次号 机器号 物料号 模具号 机种号
     */
    List<String> EXCEL_HEAD_STYLE_ONE_NO = Arrays.asList("batchNo", "machineNo", "materialNo", "mouldNo", "varietyNo", "craftItemNo");
    List<String> EXCEL_HEAD_STYLE_ONE_NAME = Arrays.asList("批次", "机器", "物料", "模具", "机种", "工艺项目");

    /**
     * 添加检验数据
     */
    @Override
    public String insertQcSource(QcSourceQoInsert insertQo, MultipartFile picture) {
        String batchNo = insertQo.getBatchNo();
        String machineNo = insertQo.getMachineNo();
        String materialNo = insertQo.getMaterialNo();
        String mouldNo = insertQo.getMouldNo();
        String varietyNo = insertQo.getVarietyNo();
        String craftItemNo = insertQo.getCraftItemNo();
        String qcItemNo = insertQo.getQcItemNo();
        String qcValue = insertQo.getQcValue();
        Optional<CraftItemEntity> craftItemByNo = craftItemDao.findByCraftItemNo(craftItemNo);
        if (craftItemByNo.isEmpty()) {
            return outputEncapsulationObject(PromptMessageEnum.DATA_NONE, "不存在该工艺项目，请先添加工艺项目！", locale);
        }
        Optional<QcItemEntity> qcItemByNo = qcItemDao.findByQcItemNo(qcItemNo);
        if (qcItemByNo.isEmpty()) {
            return outputEncapsulationObject(PromptMessageEnum.DATA_NONE, "不存在该检验项目，请先添加检验项目！", locale);
        }
        if (Constant.QC_ITEM_TYPE_STRING.equals(qcItemByNo.get().getQcItemType())) {
            QcSourceEntity entity = new QcSourceEntity();
            entity.setCreateTime(LocalDateUtils.getCurrentSecond());
            entity.setBatchNo(LocalStringUtils.formatEmptyValue(batchNo));
            entity.setMachineNo(LocalStringUtils.formatEmptyValue(machineNo));
            entity.setMaterialNo(LocalStringUtils.formatEmptyValue(materialNo));
            entity.setMouldNo(LocalStringUtils.formatEmptyValue(mouldNo));
            entity.setVarietyNo(LocalStringUtils.formatEmptyValue(varietyNo));
            entity.setCraftItemNo(LocalStringUtils.formatEmptyValue(craftItemNo));
            entity.setQcItemNo(LocalStringUtils.formatEmptyValue(qcItemNo));
            entity.setQcValue(LocalStringUtils.formatEmptyValue(qcValue));
            qcSourceDao.save(entity);
            return outputEncapsulationObject(PromptMessageEnum.SUCCESS, "添加成功！", locale);
        }
        if (Constant.QC_ITEM_TYPE_PICTURE.equals(qcItemByNo.get().getQcItemType())) {
            if (picture == null) {
                return outputEncapsulationObject(PromptMessageEnum.DATA_NONE, "上传图片不能为空！", locale);
            }
            String picturePath;
            try {
                picturePath = ExcelUploadUtil.memoryFile(picture, 2);
            } catch (IOException e) {
                e.printStackTrace();
                return outputEncapsulationObject(PromptMessageEnum.DATA_NONE, "上传图片失败！", locale);
            }
            QcSourceEntity entity = new QcSourceEntity();
            entity.setCreateTime(LocalDateUtils.getCurrentSecond());
            entity.setBatchNo(LocalStringUtils.formatEmptyValue(batchNo));
            entity.setMachineNo(LocalStringUtils.formatEmptyValue(machineNo));
            entity.setMaterialNo(LocalStringUtils.formatEmptyValue(materialNo));
            entity.setMouldNo(LocalStringUtils.formatEmptyValue(mouldNo));
            entity.setVarietyNo(LocalStringUtils.formatEmptyValue(varietyNo));
            entity.setCraftItemNo(LocalStringUtils.formatEmptyValue(craftItemNo));
            entity.setQcItemNo(LocalStringUtils.formatEmptyValue(qcItemNo));
            entity.setQcValue(LocalStringUtils.formatEmptyValue(picturePath));
            qcSourceDao.save(entity);
            return outputEncapsulationObject(PromptMessageEnum.SUCCESS, "添加成功！", locale);
        }
        return outputEncapsulationObject(PromptMessageEnum.DATA_NONE, "添加失败！", locale);
    }

    /**
     * 删除检验数据
     */
    @Transactional(rollbackOn = Exception.class)
    @Override
    public String deleteQcSource(QcSourceQoDelete deleteQo) {
        Integer id = deleteQo.getId();
        Optional<QcSourceEntity> entityById = qcSourceDao.findById(id);
        if (entityById.isEmpty()) {
            return outputEncapsulationObject(PromptMessageEnum.DATA_NONE, "不存在该检验数据，请先添加检验数据！", locale);
        }
        qcSourceDao.deleteById(id);
        return outputEncapsulationObject(PromptMessageEnum.SUCCESS, "删除成功！", locale);
    }

    /**
     * 查询检验数据
     */
    @Override
    public String queryQcSource(String batchNo, String machineNo, String materialNo, String mouldNo, String varietyNo, String craftItemNo) {
        List<CraftItemEntity> craftItemList = craftItemDao.findAll();
        if (craftItemList.isEmpty()) {
            return outputEncapsulationObject(PromptMessageEnum.DATA_NONE, "未添加工艺项目，请先添加工艺项目！", locale);
        }
        List<QcItemEntity> qcItemList = qcItemDao.findAll();
        if (qcItemList.isEmpty()) {
            return outputEncapsulationObject(PromptMessageEnum.DATA_NONE, "未添加检验项目，请先添加检验项目！", locale);
        }
        List<Map<String, Object>> sourceList = qcSourceDao.queryQcSource(LocalStringUtils.formatEmptyValue(batchNo), LocalStringUtils.formatEmptyValue(machineNo), LocalStringUtils.formatEmptyValue(materialNo), LocalStringUtils.formatEmptyValue(mouldNo), LocalStringUtils.formatEmptyValue(varietyNo), LocalStringUtils.formatEmptyValue(craftItemNo));
        List<Map<String, Object>> resultList = generateResultList(sourceList);
        if (resultList.isEmpty()) {
            return outputEncapsulationObject(PromptMessageEnum.DATA_NONE, "暂时没有该检验数据的信息！", locale);
        }
        return outputEncapsulationObject(PromptMessageEnum.SUCCESS, resultList, locale);
    }

    /**
     * 生成返回结果
     */
    private List<Map<String, Object>> generateResultList(List<Map<String, Object>> sourceList) {
        List<Map<String, Object>> resultList = new ArrayList<>();
        if (sourceList == null || sourceList.isEmpty()) {
            return resultList;
        }
        for (Map<String, Object> source : sourceList) {
            String batchNo = (String) source.getOrDefault("batchNo", "");
            String machineNo = (String) source.getOrDefault("machineNo", "");
            String materialNo = (String) source.getOrDefault("materialNo", "");
            String mouldNo = (String) source.getOrDefault("mouldNo", "");
            String varietyNo = (String) source.getOrDefault("varietyNo", "");
            String craftItemNo = (String) source.getOrDefault("craftItemNo", "");
            String qcItemNo = (String) source.getOrDefault("qcItemNo", "");
            String qcValue = (String) source.getOrDefault("qcValue", "");
            if (TextUtils.isEmpty(qcItemNo) || TextUtils.isEmpty(qcValue)) {
                continue;
            }
            String formatValue = getFormatValue(varietyNo, craftItemNo, qcItemNo, qcValue);
            Map<String, Object> emptyRoom = findEmptyRoom(resultList, batchNo, qcItemNo);
            if (emptyRoom == null) {
                Map<String, Object> newRoom = new HashMap<>();
                newRoom.put(EXCEL_HEAD_STYLE_ONE_NO.get(0), batchNo);
                newRoom.put(EXCEL_HEAD_STYLE_ONE_NO.get(1), machineNo);
                newRoom.put(EXCEL_HEAD_STYLE_ONE_NO.get(2), materialNo);
                newRoom.put(EXCEL_HEAD_STYLE_ONE_NO.get(3), mouldNo);
                newRoom.put(EXCEL_HEAD_STYLE_ONE_NO.get(4), varietyNo);
                newRoom.put(EXCEL_HEAD_STYLE_ONE_NO.get(5), craftItemNo);
                newRoom.put(qcItemNo, formatValue);
                resultList.add(newRoom);
            } else {
                emptyRoom.put(qcItemNo, formatValue);
            }
        }
        return resultList;
    }

    /**
     * 获取添加了 ##H 或 ##L 后缀的测量值（如果可能的话）
     */
    private String getFormatValue(String varietyNo, String craftItemNo, String qcItemNo, String qcValue) {
        Optional<QcSpecEntity> existsQcSpecOptional = qcSpecDao.findExistsQcSpec(varietyNo, craftItemNo, qcItemNo);
        if (existsQcSpecOptional.isEmpty()) {
            return qcValue;
        }
        QcSpecEntity existsQcSpec = existsQcSpecOptional.get();
        double valueMax = stringToDouble(existsQcSpec.getMax());
        double valueMin = stringToDouble(existsQcSpec.getMin());
        double value = stringToDouble(qcValue);
        if (valueMax >= 0 && value >= 0 && value >= valueMax) {
            return qcValue + "##H";
        }
        if (valueMin >= 0 && value >= 0 && value <= valueMin) {
            return qcValue + "##L";
        }
        return qcValue;
    }

    /**
     * 格式化字符串为Double
     */
    private Double stringToDouble(String stringValue) {
        try {
            return Double.parseDouble(stringValue);
        } catch (Exception e) {
            e.printStackTrace();
            return -1D;
        }
    }

    /**
     * 尝试找到一个qcItemNo有空位置的Map
     */
    private Map<String, Object> findEmptyRoom(List<Map<String, Object>> mapList, String batchNo, String qcItemNo) {
        if (mapList == null || mapList.isEmpty() || TextUtils.isEmpty(batchNo) || TextUtils.isEmpty(qcItemNo)) {
            return null;
        }
        for (Map<String, Object> room : mapList) {
            if (room.getOrDefault("batchNo", "").equals(batchNo)) {
                if (!room.containsKey(qcItemNo)) {
                    return room;
                }
            }
        }
        return null;
    }

    /**
     * 导出Excel表格
     */
    @Override
    public String exportExcel(ExportExcelQo exportExcelQo) throws Exception {
        String host = exportExcelQo.getHost();
        String batchNo = exportExcelQo.getBatchNo();
        String machineNo = exportExcelQo.getMachineNo();
        String materialNo = exportExcelQo.getMaterialNo();
        String mouldNo = exportExcelQo.getMouldNo();
        String varietyNo = exportExcelQo.getVarietyNo();
        String craftItemNo = exportExcelQo.getCraftItemNo();
        List<CraftItemEntity> craftItemList = craftItemDao.findAll();
        if (craftItemList.isEmpty()) {
            return outputEncapsulationObject(PromptMessageEnum.DATA_NONE, "未添加工艺项目，请先添加工艺项目！", locale);
        }
        List<QcItemEntity> qcItemList = qcItemDao.findAll();
        if (qcItemList.isEmpty()) {
            return outputEncapsulationObject(PromptMessageEnum.DATA_NONE, "未添加检验项目，请先添加检验项目！", locale);
        }
        List<Map<String, Object>> sourceList = qcSourceDao.queryQcSource(LocalStringUtils.formatEmptyValue(batchNo), LocalStringUtils.formatEmptyValue(machineNo), LocalStringUtils.formatEmptyValue(materialNo), LocalStringUtils.formatEmptyValue(mouldNo), LocalStringUtils.formatEmptyValue(varietyNo), LocalStringUtils.formatEmptyValue(craftItemNo));
        List<Map<String, Object>> resultList = generateResultList(sourceList);
        String finalHost = host.endsWith("/") ? host : host + "/";
        List<List<String>> formattedResultList = formatResultList(finalHost, resultList, qcItemList);
        if (formattedResultList.isEmpty()) {
            return outputEncapsulationObject(PromptMessageEnum.DATA_NONE, "暂时没有该检验数据的信息！", locale);
        }
        String excelPath = doExportExcel(formattedResultList);
        return outputEncapsulationObject(PromptMessageEnum.SUCCESS, finalHost + excelPath, locale);
    }

    /**
     * 格式化结果集数据
     */
    private List<List<String>> formatResultList(String host, List<Map<String, Object>> resultList, List<QcItemEntity> qcItemList) {
        List<List<String>> formatResultList = new ArrayList<>();
        if (resultList == null || resultList.isEmpty()) {
            return formatResultList;
        }
        List<String> headList = new ArrayList<>(EXCEL_HEAD_STYLE_ONE_NAME);
        for (QcItemEntity qcItemEntity : qcItemList) {
            headList.add(qcItemEntity.getQcItemName());
        }
        formatResultList.add(headList);
        for (Map<String, Object> dataMap : resultList) {
            List<String> dataList = new ArrayList<>();
            dataList.add((String) dataMap.getOrDefault(EXCEL_HEAD_STYLE_ONE_NO.get(0), ""));
            dataList.add((String) dataMap.getOrDefault(EXCEL_HEAD_STYLE_ONE_NO.get(1), ""));
            dataList.add((String) dataMap.getOrDefault(EXCEL_HEAD_STYLE_ONE_NO.get(2), ""));
            dataList.add((String) dataMap.getOrDefault(EXCEL_HEAD_STYLE_ONE_NO.get(3), ""));
            dataList.add((String) dataMap.getOrDefault(EXCEL_HEAD_STYLE_ONE_NO.get(4), ""));
            dataList.add((String) dataMap.getOrDefault(EXCEL_HEAD_STYLE_ONE_NO.get(5), ""));
            for (QcItemEntity entity : qcItemList) {
                String value = (String) dataMap.getOrDefault(entity.getQcItemNo(), "");
                if (Constant.QC_ITEM_TYPE_STRING.equals(entity.getQcItemType())) {
                    dataList.add(value);
                } else if (Constant.QC_ITEM_TYPE_PICTURE.equals(entity.getQcItemType())) {
                    if (value.startsWith("picture")) {
                        dataList.add(host + value);
                    } else {
                        dataList.add(value);
                    }
                } else {
                    dataList.add(value);
                }
            }
            formatResultList.add(dataList);
        }
        return formatResultList;
    }

    /**
     * 将格式化后的结果集数据导出为Excel文件
     */
    private String doExportExcel(List<List<String>> dataListList) throws Exception {
        //创建XSSFWorkbook对象(excel的文档对象)
        XSSFWorkbook workbook = new XSSFWorkbook();
        //建立新的sheet对象（excel的表单）
        XSSFSheet sheet = workbook.createSheet("测量数据表");

        XSSFCellStyle highCellStyle = workbook.createCellStyle();
        highCellStyle.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        highCellStyle.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
        XSSFFont highFont = workbook.createFont();
        highFont.setColor(HSSFColor.RED.index);
        highCellStyle.setFont(highFont);

        XSSFCellStyle lowCellStyle = workbook.createCellStyle();
        lowCellStyle.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        lowCellStyle.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
        XSSFFont lowFont = workbook.createFont();
        lowFont.setColor(HSSFColor.BLUE.index);
        lowCellStyle.setFont(lowFont);

        XSSFCellStyle normalCellStyle = workbook.createCellStyle();
        normalCellStyle.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        normalCellStyle.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
        XSSFFont normalFont = workbook.createFont();
        normalFont.setColor(HSSFColor.BLACK.index);
        normalCellStyle.setFont(normalFont);

        //在sheet里创建第一行，参数为行索引(excel的行)，可以是0～65535之间的任何一个
        XSSFRow titleRow = sheet.createRow(0);
        //创建单元格（excel的单元格，参数为列索引，可以是0～255之间的任何一个
        XSSFCell titleCell = titleRow.createCell(0);
        //设置单元格内容
        titleCell.setCellValue("测量数据导出表");
        titleCell.setCellStyle(normalCellStyle);
        //合并单元格CellRangeAddress构造参数依次表示起始行，截至行，起始列， 截至列
        sheet.addMergedRegion(new CellRangeAddress(0, 0, 0, dataListList.get(0).size() - 1));
        int startRowNum = 1;
        for (int i = 0; i < dataListList.size(); i++) {
            List<String> dataList = dataListList.get(i);
            if (dataList != null && dataList.size() > 0) {
                XSSFRow row = sheet.createRow(i + startRowNum);
                for (int j = 0; j < dataList.size(); j++) {
                    XSSFCell cell = row.createCell(j);
                    String data = dataList.get(j);
                    if (data.contains("##H")) {
                        cell.setCellValue(data.split("##")[0]);
                        cell.setCellStyle(highCellStyle);
                    } else if (data.contains("##L")) {
                        cell.setCellValue(data.split("##")[0]);
                        cell.setCellStyle(lowCellStyle);
                    } else {
                        cell.setCellValue(data);
                        cell.setCellStyle(normalCellStyle);
                    }
                }
            }
        }
        //输出Excel文件
        File dirFile = new File("/home/sxy/server/industrial_measurement/excel/");
        if (!dirFile.exists()) {
            dirFile.mkdirs();
        }
        String excelFileName = System.currentTimeMillis() + ".xlsx";
        FileOutputStream output = new FileOutputStream(new File(dirFile, excelFileName));
        workbook.write(output);
        output.flush();
        return "excel/" + excelFileName;
    }


}