|
package org.springblade.mdm.machinefile.service;
|
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import lombok.AllArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.io.FileUtils;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springblade.core.log.exception.ServiceException;
|
import org.springblade.core.mp.base.BizServiceImpl;
|
import org.springblade.core.mp.support.Condition;
|
import org.springblade.core.oss.OssTemplate;
|
import org.springblade.core.oss.model.BladeFile;
|
import org.springblade.mdm.basesetting.machine.service.MachineService;
|
import org.springblade.mdm.basesetting.machine.entity.Machine;
|
import org.springblade.mdm.commons.service.MachineDirTranslator;
|
import org.springblade.mdm.machinefile.entity.MachineFile;
|
import org.springblade.mdm.machinefile.entity.MachineFileChangeHis;
|
import org.springblade.mdm.machinefile.mapper.MachineFileMapper;
|
import org.springblade.mdm.machinefile.vo.MachineFileQueryVO;
|
import org.springblade.mdm.machinefile.vo.MachineBackFileQueryVO;
|
import org.springblade.mdm.program.vo.CompareDataVO;
|
import org.springblade.mdm.statreport.vo.MachineFileExceptionQueryVO;
|
import org.springblade.mdm.utils.FileContentUtil;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.io.File;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.nio.file.Files;
|
import java.nio.file.Path;
|
import java.nio.file.Paths;
|
import java.util.Optional;
|
|
/**
|
* 机床文件
|
*
|
* @author yangys
|
*/
|
@Slf4j
|
@Service
|
@AllArgsConstructor
|
public class MachineFileService extends BizServiceImpl<MachineFileMapper, MachineFile> {
|
private final MachineService machineService;
|
private final MachineFileChangeHisService machineFileChangeHisService;
|
private final OssTemplate ossTemplate;
|
private MachineDirTranslator machineDirTranslator;
|
/**
|
* 获取文件全路径
|
* @param id 数据id
|
* @return 文件全路径
|
*/
|
public String getFilePath(Long id) {
|
MachineFile machineFile = getById(id);
|
Machine machine = machineService.getByCode(machineFile.getMachineCode());
|
|
return getBasePath(machine, machineFile.getDirType()) + File.separator + machineFile.getName();
|
}
|
|
public String getFilePath(MachineFile machineFile) {
|
Machine machine = machineService.getByCode(machineFile.getMachineCode());
|
return getBasePath(machine, machineFile.getDirType()) + File.separator + machineFile.getName();
|
}
|
@Transactional(readOnly = true)
|
public String getMachineFileContent(Long id) {
|
MachineFile machineFile = getById(id);
|
Machine machine = machineService.getByCode(machineFile.getMachineCode());
|
|
String filePathStr = getBasePath(machine,machineFile.getDirType())+ File.separator+machineFile.getName();
|
|
String content;
|
|
Path filePath = Paths.get(filePathStr);
|
if(!filePath.toFile().exists()){
|
return StringUtils.EMPTY;
|
}
|
try (InputStream inputStream = Files.newInputStream(filePath)) {
|
// 使用输入流读取文件内容
|
content = FileContentUtil.getContentFromStream(inputStream);
|
} catch (IOException e) {
|
log.error("读取文件md5失败",e);
|
return StringUtils.EMPTY;
|
}
|
|
return content;
|
}
|
|
/**
|
* 获取基本路径
|
* @param machine 机床
|
* @param dirType 目录类型
|
* @return 目录基本路径
|
*/
|
public static String getBasePath(Machine machine,String dirType){
|
String dirPath;
|
switch (dirType) {
|
case MachineFile.DIR_TYPE_REC:
|
dirPath = machine.getProgReceiveDir();
|
break;
|
case MachineFile.DIR_TYPE_SEND:
|
dirPath = machine.getProgSendDir();
|
break;
|
case MachineFile.DIR_TYPE_TEMP:
|
dirPath = machine.getProgTempDir();
|
break;
|
default:
|
log.warn("目录类型不匹配:{}",dirType);
|
return null;//不符合任何类型，直接退出
|
}
|
return dirPath;
|
}
|
|
@Transactional
|
public void saveFileContent(Long id, String content) throws IOException {
|
MachineFile machineFile = getById(id);
|
Machine machine = machineService.getByCode(machineFile.getMachineCode());
|
|
String filePathStr = getBasePath(machine,machineFile.getDirType())+ File.separator+machineFile.getName();
|
Path filePath = Paths.get(filePathStr);
|
|
String ossName;
|
BladeFile bfile;
|
try (InputStream inputStream = Files.newInputStream(filePath)) {
|
bfile = ossTemplate.putFile(machineFile.getName(), inputStream);
|
}
|
ossName =bfile.getName();
|
|
try (InputStream inputStream = Files.newInputStream(filePath)) {
|
// 使用输入流读取文件内容
|
String charsetName = FileContentUtil.detectFromInputStream(inputStream);
|
FileUtils.writeStringToFile(filePath.toFile(),content,charsetName);
|
} catch (IOException e) {
|
log.error("读取文件编码失败",e);
|
throw new ServiceException("获取文件编码失败");
|
}
|
|
try (InputStream inputStream = Files.newInputStream(filePath)) {
|
bfile = ossTemplate.putFile(machineFile.getName(), inputStream);
|
}
|
//保存历史记录
|
MachineFileChangeHis his = new MachineFileChangeHis();
|
his.setMachineFileId(id);
|
his.setOssName(ossName);
|
his.setOssNameAfter(bfile.getName());
|
this.machineFileChangeHisService.save(his);
|
}
|
|
/**
|
* 获取文件的输入流
|
* @param machineFile
|
* @return
|
* @throws IOException
|
*/
|
public InputStream getInputStream(MachineFile machineFile) throws IOException {
|
|
Machine machine = machineService.getByCode(machineFile.getMachineCode());
|
|
String filePathStr = getBasePath(machine,machineFile.getDirType())+ File.separator+machineFile.getName();
|
Path filePath = Paths.get(filePathStr);
|
return Files.newInputStream(filePath);
|
|
}
|
|
@Transactional(readOnly = true)
|
public IPage<MachineFile> filePageForAccept(MachineBackFileQueryVO query) {
|
return this.baseMapper.filePageForAccept(Condition.getPage(query),query);
|
}
|
|
public MachineFile getExistsFile(String name, String dirType, String machineCode) {
|
return this.lambdaQuery().eq(MachineFile::getName, name)
|
.eq(MachineFile::getDirType, dirType)
|
.eq(MachineFile::getMachineCode, machineCode).list().stream().findFirst().orElse(null);
|
}
|
|
/**
|
* 回传程序处理 分页查询
|
* @param query
|
* @return
|
*/
|
/*
|
public IPage<MachineAcceptedFileVO> handlePageQuery(MachineAcceptedFileHandleQueryVO query) {
|
IPage<MachineAcceptedFileVO> page = this.getBaseMapper().handlePageQuery(Condition.getPage(query),query);
|
return page;
|
}*/
|
|
/**
|
* 机床文件分页查询
|
* @param query
|
* @return
|
*/
|
public IPage<MachineFile> pageQuery(MachineFileQueryVO query) {
|
return baseMapper.pageQuery(Condition.getPage(query),query);
|
}
|
|
/**
|
* 获取修改记录前后2个文件内容
|
* @param id
|
* @return
|
* @throws IOException
|
*/
|
public CompareDataVO queryHisContentForCompare(Long id) throws IOException {
|
CompareDataVO vo = new CompareDataVO();
|
MachineFileChangeHis his = machineFileChangeHisService.getById(id);
|
vo.setContent1(FileContentUtil.getContentFromStream(ossTemplate.statFileStream(his.getOssName())));
|
vo.setContent2(FileContentUtil.getContentFromStream(ossTemplate.statFileStream(his.getOssNameAfter())));
|
|
return vo;
|
}
|
|
/**
|
* 刷新文件信息，不存在新增，存在则更新
|
* @param mf
|
*/
|
public void refreshFileData(MachineFile mf) {
|
MachineFile existFileInDb = getExistsFile(mf.getName(),mf.getDirType(),mf.getMachineCode());
|
if(existFileInDb == null) {
|
save(mf);
|
}else{
|
log.info("文件在库中已存在:{}",mf.getName());
|
existFileInDb.setFileSize(mf.getFileSize());
|
existFileInDb.setProgramStatus(mf.getProgramStatus());
|
existFileInDb.setFileCreateTime(mf.getFileCreateTime());
|
existFileInDb.setFileModifyTime(mf.getFileModifyTime());
|
existFileInDb.setStatus(MachineFile.STATUS_NORMAL);
|
if(!com.alibaba.excel.util.StringUtils.equals(existFileInDb.getMd5(),mf.getMd5())){
|
//文件内容发生变化了,设置状态为初始状态
|
existFileInDb.setStatus(MachineFile.STATUS_NORMAL);
|
}
|
existFileInDb.setMd5(mf.getMd5());
|
updateById(existFileInDb);
|
}
|
}
|
|
/**
|
* 物理删除文件信息
|
* @param id 数据id
|
*/
|
public void deleteById(Long id) {
|
baseMapper.deleteById(id);
|
}
|
|
/**
|
* 异常文件记录查询
|
* @param query
|
* @return
|
*/
|
@Transactional(readOnly = true)
|
public IPage<MachineFile> exceptionFilePageQuery(MachineFileExceptionQueryVO query) {
|
return baseMapper.exceptionFilePageQuery(Condition.getPage(query),query);
|
}
|
|
/**
|
* 修改文件名称
|
* @param id 文件id
|
* @param name 新名称
|
*/
|
public void changeName(Long id, String name) throws IOException {
|
MachineFile machineFile = getById(id);
|
if(machineFile.getName().equals(name)) {
|
return;
|
}
|
Machine machine = machineService.getByCode(machineFile.getMachineCode());
|
String baseDir = machineDirTranslator.trans(getBasePath(machine,machineFile.getDirType()));
|
String targetFilePath = baseDir + File.separator + name;
|
Path filePath = Paths.get(targetFilePath);
|
|
if(filePath.toFile().exists()) {
|
throw new ServiceException("同名文件已存在："+name);
|
}
|
Files.move(Paths.get(baseDir+File.separator+machineFile.getName()),filePath);
|
baseMapper.deleteById(id);
|
}
|
|
public MachineFile getByNameAndMachineInDir(String name, String machineCode, String dirType) {
|
Optional<MachineFile> optFIle = this.lambdaQuery().eq(MachineFile::getName, name).eq(MachineFile::getMachineCode, machineCode)
|
.eq(MachineFile::getDirType, dirType).eq(MachineFile::getStatus, MachineFile.STATUS_NORMAL).oneOpt();
|
return optFIle.orElse(null);
|
}
|
}
|
