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.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.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.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;

    /**
     * 添加检验数据
     */
    @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<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("batchNo", batchNo);
                newRoom.put("machineNo", machineNo);
                newRoom.put("materialNo", materialNo);
                newRoom.put("mouldNo", mouldNo);
                newRoom.put("varietyNo", varietyNo);
                newRoom.put("craftItemNo", 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;
    }

}