/*
|
* Copyright (c) 2018-2025, ztt All rights reserved.
|
*
|
* Redistribution and use in source and binary forms, with or without
|
* modification, are permitted provided that the following conditions are met:
|
*
|
* Redistributions of source code must retain the above copyright notice,
|
* this list of conditions and the following disclaimer.
|
* Redistributions in binary form must reproduce the above copyright
|
* notice, this list of conditions and the following disclaimer in the
|
* documentation and/or other materials provided with the distribution.
|
* Neither the name of the pig4cloud.com developer nor the names of its
|
* contributors may be used to endorse or promote products derived from
|
* this software without specific prior written permission.
|
* Author: ztt
|
*/
|
package com.chinaztt.mes.quality.service.impl;
|
|
import cn.hutool.core.collection.CollectionUtil;
|
import cn.hutool.core.io.FileUtil;
|
import cn.hutool.core.io.IoUtil;
|
import cn.hutool.core.util.IdUtil;
|
import cn.hutool.core.util.StrUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.chinaztt.mes.basic.entity.Warehouse;
|
import com.chinaztt.mes.common.numgen.NumberGenerator;
|
import com.chinaztt.mes.quality.dto.QualityExceptionDTO;
|
import com.chinaztt.mes.quality.entity.ExceptionAttachment;
|
import com.chinaztt.mes.quality.entity.QualityException;
|
import com.chinaztt.mes.quality.entity.ReportSampleAttachment;
|
import com.chinaztt.mes.quality.mapper.ExceptionAttachmentMapper;
|
import com.chinaztt.mes.quality.mapper.QualityExceptionMapper;
|
import com.chinaztt.mes.quality.service.QualityExceptionService;
|
import com.chinaztt.ztt.common.core.util.R;
|
import com.chinaztt.ztt.common.oss.OssProperties;
|
import com.chinaztt.ztt.common.oss.service.OssTemplate;
|
import com.sun.org.apache.bcel.internal.generic.IF_ACMPEQ;
|
import lombok.AllArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.transaction.interceptor.TransactionAspectSupport;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.InputStream;
|
import java.util.List;
|
|
/**
|
* 生产异常处理
|
*
|
* @author sunxl
|
* @date 2021-07-01 14:48:35
|
*/
|
@Slf4j
|
@Service
|
@AllArgsConstructor
|
@Transactional(rollbackFor = Exception.class)
|
public class QualityExceptionServiceImpl extends ServiceImpl<QualityExceptionMapper, QualityException> implements QualityExceptionService {
|
private NumberGenerator<QualityException> exceptionNumberGenerator;
|
private final OssProperties ossProperties;
|
private final OssTemplate minioTemplate;
|
private ExceptionAttachmentMapper exceptionAttachmentMapper;
|
@Override
|
public boolean fullSave(QualityException exception) {
|
exception.setCode(exceptionNumberGenerator.generateNumberWithPrefix(QualityException.DIGIT, QualityException.PREFIX, QualityException::getCode));
|
baseMapper.insert(exception);
|
return true;
|
}
|
|
@Override
|
public IPage<List<QualityExceptionDTO>> getExceptionPage(Page page, QueryWrapper<QualityExceptionDTO> gen) {
|
return baseMapper.getExceptionPage(page,gen);
|
}
|
|
@Override
|
public QualityExceptionDTO getExceptionPageById(Long id) {
|
return baseMapper.getExceptionPageById(id);
|
}
|
|
@Override
|
public R uploadFile(MultipartFile file, Long exceptionId) {
|
String fileName = IdUtil.simpleUUID() + StrUtil.DOT + FileUtil.extName(file.getOriginalFilename());
|
String url = String.format("/mes/exception/%s/%s", ossProperties.getBucketName(), fileName);
|
try {
|
minioTemplate.putObject(ossProperties.getBucketName(), fileName, file.getInputStream());
|
|
ExceptionAttachment attachment = new ExceptionAttachment();
|
attachment.setOriginal(file.getOriginalFilename());
|
attachment.setUrl(url);
|
attachment.setFileName(fileName);
|
attachment.setExceptionId(exceptionId);
|
exceptionAttachmentMapper.insert(attachment);
|
|
} catch (Exception e) {
|
log.error("上传失败", e);
|
TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
|
return R.failed(e.getLocalizedMessage());
|
}
|
return R.ok();
|
}
|
|
@Override
|
public List<ExceptionAttachment> getExceptionAttachment(Long id) {
|
return exceptionAttachmentMapper.selectList(Wrappers.<ExceptionAttachment>lambdaQuery().eq(ExceptionAttachment::getExceptionId,id));
|
}
|
|
@Override
|
public R deleteFile(String fileName) {
|
ExceptionAttachment exceptionAttachment = exceptionAttachmentMapper.selectOne(Wrappers.<ExceptionAttachment>lambdaQuery().eq(ExceptionAttachment::getFileName,fileName));
|
try {
|
minioTemplate.removeObject(ossProperties.getBucketName(), exceptionAttachment.getFileName());
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
exceptionAttachmentMapper.deleteById(exceptionAttachment.getId());
|
return R.ok();
|
}
|
|
@Override
|
public void getFile(String bucket, String fileName, HttpServletResponse response) {
|
try (InputStream inputStream = minioTemplate.getObject(bucket, fileName)) {
|
response.setContentType("application/octet-stream; charset=UTF-8");
|
IoUtil.copy(inputStream, response.getOutputStream());
|
} catch (Exception e) {
|
log.error("文件读取异常: {}", e.getLocalizedMessage());
|
}
|
}
|
|
@Override
|
public IPage<QualityExceptionDTO> getExceptionFilePage(Page page, QueryWrapper<QualityExceptionDTO> gen) {
|
IPage<QualityExceptionDTO> iPage = baseMapper.getExceptionFilePage(page,gen);
|
if (CollectionUtil.isNotEmpty(iPage.getRecords())){
|
for (QualityExceptionDTO qualityExceptionDTO:iPage.getRecords()){
|
qualityExceptionDTO.setUrlList(exceptionAttachmentMapper.getUrl(qualityExceptionDTO.getId()));
|
}
|
}
|
return iPage;
|
}
|
|
@Override
|
public QualityException addQualityException(QualityException qualityException, List<MultipartFile> file) {
|
qualityException.setCode(exceptionNumberGenerator.generateNumberWithPrefix(QualityException.DIGIT, QualityException.PREFIX, QualityException::getCode));
|
baseMapper.insert(qualityException);
|
for (MultipartFile multipartFile:file){
|
uploadFile(multipartFile,qualityException.getId());
|
}
|
return qualityException;
|
}
|
|
@Override
|
public QualityException editQualityException(QualityExceptionDTO qualityException, List<MultipartFile> file) {
|
baseMapper.updateById(qualityException);
|
List<ExceptionAttachment> exceptionAttachmentList = exceptionAttachmentMapper.getUrl(qualityException.getId());
|
if (CollectionUtil.isNotEmpty(exceptionAttachmentList)){
|
A:for (ExceptionAttachment exceptionAttachment:exceptionAttachmentList){
|
if (CollectionUtil.isNotEmpty(qualityException.getUrlList())){
|
for (ExceptionAttachment newExceptionAttachment:qualityException.getUrlList()){
|
if (exceptionAttachment.getUrl().equals(newExceptionAttachment.getUrl())){
|
continue A;
|
}
|
}
|
exceptionAttachmentMapper.deleteById(exceptionAttachment.getId());
|
}
|
}
|
}
|
if (CollectionUtil.isNotEmpty(file)){
|
for (MultipartFile multipartFile:file){
|
uploadFile(multipartFile,qualityException.getId());
|
}
|
}
|
return qualityException;
|
}
|
}
|
