/*
|
* 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.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.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.chinaztt.mes.quality.entity.ApplyPart;
|
import com.chinaztt.mes.quality.entity.ApplyPartFile;
|
import com.chinaztt.mes.quality.mapper.ApplyPartFileMapper;
|
import com.chinaztt.mes.quality.mapper.ApplyPartMapper;
|
import com.chinaztt.mes.quality.service.ApplyPartFileService;
|
import com.chinaztt.ztt.common.core.util.R;
|
import com.chinaztt.ztt.common.oss.OssProperties;
|
import com.chinaztt.ztt.common.oss.service.OssTemplate;
|
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 shz
|
* @date 2023-03-28 14:29:21
|
*/
|
@Slf4j
|
@Service
|
@AllArgsConstructor
|
@Transactional(rollbackFor = Exception.class)
|
public class ApplyPartFileServiceImpl extends ServiceImpl<ApplyPartFileMapper, ApplyPartFile> implements ApplyPartFileService {
|
|
private final OssProperties ossProperties;
|
private final OssTemplate minioTemplate;
|
private final ApplyPartMapper applyPartMapper;
|
|
@Override
|
public R uploadFile(MultipartFile file, Long applyPartId) {
|
String fileName = IdUtil.simpleUUID() + StrUtil.DOT + FileUtil.extName(file.getOriginalFilename());
|
String url = String.format("/mes/qualityapplypartfile/%s/%s", ossProperties.getBucketName(), fileName);
|
try {
|
minioTemplate.putObject(ossProperties.getBucketName(), fileName, file.getInputStream());
|
|
ApplyPartFile applyPartFile = new ApplyPartFile();
|
applyPartFile.setOriginal(file.getOriginalFilename());
|
applyPartFile.setPath(url);
|
applyPartFile.setFileName(fileName);
|
applyPartFile.setApplyPartId(applyPartId);
|
this.baseMapper.insert(applyPartFile);
|
} catch (Exception e) {
|
log.error("上传失败", e);
|
TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
|
return R.failed(e.getLocalizedMessage());
|
}
|
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 List<ApplyPartFile> getListByApplyPartId(Long applyPartId) {
|
return this.list(Wrappers.<ApplyPartFile>query().lambda().eq(ApplyPartFile::getApplyPartId, applyPartId));
|
}
|
}
|
