package org.jeecg.modules.doc.service.impl;
|
|
import cn.hutool.core.util.IdUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
|
import org.apache.commons.codec.digest.DigestUtils;
|
import org.apache.commons.io.FileUtils;
|
import org.apache.commons.io.FilenameUtils;
|
import org.apache.commons.io.IOUtils;
|
import org.jeecg.common.util.DateUtils;
|
import org.jeecg.modules.doc.component.*;
|
import org.jeecg.modules.doc.constant.Constant;
|
import org.jeecg.modules.doc.entity.DocFile;
|
import org.jeecg.modules.doc.entity.DocFilePath;
|
import org.jeecg.modules.doc.mapper.DocFileMapper;
|
import org.jeecg.modules.doc.mapper.DocFilePathMapper;
|
import org.jeecg.modules.doc.service.IDocFileService;
|
import org.jeecg.modules.doc.vo.CopyFile;
|
import org.jeecg.modules.doc.vo.DownloadFile;
|
import org.jeecg.modules.doc.exception.QiwenException;
|
import org.jeecg.modules.doc.util.UFOPUtils;
|
import org.jeecg.modules.doc.vo.QiwenFile;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import org.springframework.util.FileCopyUtils;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.annotation.Resource;
|
import java.io.*;
|
import java.util.*;
|
|
/**
|
* @Description: doc_file
|
* @Author: jeecg-boot
|
* @Date: 2022-07-04
|
* @Version: V1.0
|
*/
|
@Service
|
public class DocFileServiceImpl extends ServiceImpl<DocFileMapper, DocFile> implements IDocFileService {
|
|
|
@Resource
|
DocFilePathMapper docFilePathMapper;
|
@Resource
|
AsyncTaskComp asyncTaskComp;
|
|
@Autowired
|
DocFileDealComp fileDealComp;
|
|
|
|
/**
|
* 上传文件到目的根路径，先计算文件MD5值，查询已上传过同一文件，有相同文件返回文件id，否则上传新文件并返回id
|
*
|
* @param f 待上传文件
|
* @param identifier
|
* @return
|
*/
|
@Override
|
public DocFile saveFile(MultipartFile f, String identifier) throws IOException {
|
String dfId;
|
DocFile docFile = new DocFile();
|
|
dfId = IdUtil.getSnowflakeNextIdStr();
|
docFile.setFileId(dfId);
|
|
String date = DateUtils.getDate("yyyyMMdd");
|
File savepath = new File(UFOPUtils.getStaticPath() + date);
|
if (!savepath.exists()) {
|
savepath.mkdirs();// 创建文件根目录
|
}
|
int index = f.getOriginalFilename().lastIndexOf(".");
|
String extend = index == -1?"":f.getOriginalFilename().substring(index+1);
|
String fileUrl = UFOPUtils.getUploadFileUrl(identifier,extend);
|
File saveFile = new File(UFOPUtils.getStaticPath() + fileUrl);
|
try {
|
f.transferTo(saveFile);
|
} catch (IOException e) {
|
throw new RuntimeException(e);
|
}
|
|
docFile.setFileUrl(fileUrl);
|
docFile.setFileStatus(1);
|
docFile.setStorageType(0);
|
docFile.setFileSize(f.getSize());
|
docFile.setIdentifier(identifier);
|
baseMapper.insert(docFile);
|
// }
|
|
return docFile;
|
}
|
|
|
@Override
|
public DocFile saveFile(File f, String identifier) throws IOException {
|
String dfId = null;
|
String fileUrl = null;
|
DocFile docFile = new DocFile();
|
List<DocFile> fileList = baseMapper.selectList(new LambdaQueryWrapper<DocFile>().eq(DocFile::getIdentifier,identifier));
|
|
if (fileList!=null && fileList.size()>0) {
|
docFile = fileList.get(0);
|
fileUrl = docFile.getFileUrl();
|
|
} else {
|
|
dfId = IdUtil.getSnowflakeNextIdStr();
|
docFile.setFileId(dfId);
|
|
String date = DateUtils.getDate("yyyyMMdd");
|
File savepath = new File(UFOPUtils.getStaticPath() + "/研发技术文档/" + date);
|
if (!savepath.exists()) {
|
savepath.mkdirs();// 创建文件根目录
|
}
|
int index = f.getName().lastIndexOf(".");
|
String extend = index == -1?"":f.getName().substring(index+1);
|
fileUrl = UFOPUtils.getUploadFileUrl(identifier,extend);
|
}
|
|
|
String saveUrl = fileUrl.replace("研发技术文档", "//192.168.0.9/研发技术文档");
|
File saveFile = new File(saveUrl);
|
try {
|
FileCopyUtils.copy(f, saveFile);
|
} catch (IOException e) {
|
throw new RuntimeException(e);
|
}
|
|
docFile.setFileUrl(fileUrl.replace("研发技术文档", "upload"));
|
docFile.setFileStatus(1);
|
docFile.setStorageType(0);
|
docFile.setFileSize(f.length());
|
docFile.setIdentifier(identifier);
|
if (dfId != null) {
|
baseMapper.insert(docFile);
|
} else if (docFile.getFileId() != null) {
|
baseMapper.update(null, new LambdaUpdateWrapper<DocFile>().set(DocFile::getFileStatus, 1).eq(DocFile::getFileId, docFile.getFileId()));
|
}
|
|
// }
|
|
return docFile;
|
}
|
@Override
|
public Long getFilePointCount(String fileId) {
|
LambdaQueryWrapper<DocFilePath> lambdaQueryWrapper = new LambdaQueryWrapper<>();
|
lambdaQueryWrapper.eq(DocFilePath::getFileId, fileId);
|
long count = docFilePathMapper.selectCount(lambdaQueryWrapper);
|
return count;
|
}
|
@Override
|
public void updateFileDetail(String userFileId, String identifier, long fileSize, String modifyUserId) {
|
DocFilePath userFile = docFilePathMapper.selectById(userFileId);
|
DocFile fileBean = new DocFile();
|
fileBean.setIdentifier(identifier);
|
fileBean.setFileSize(fileSize);
|
fileBean.setUpdateTime(DateUtils.getDate());
|
fileBean.setUpdateBy(modifyUserId);
|
fileBean.setFileId(userFile.getFileId());
|
baseMapper.updateById(fileBean);
|
userFile.setCreateTime(DateUtils.getDate());
|
docFilePathMapper.updateById(userFile);
|
}
|
|
|
/**
|
* 解压文件
|
* @param userFileId
|
* @param unzipMode
|
* @param filePath
|
*/
|
@Override
|
public void unzipFile(String userFileId, int unzipMode, String filePath) {
|
DocFilePath userFile = docFilePathMapper.selectById(userFileId);
|
DocFile fileBean = baseMapper.selectById(userFile.getFileId());
|
File destFile = new File(UFOPUtils.getStaticPath() + "temp" + File.separator + fileBean.getFileUrl());
|
|
|
Downloader downloader = new LocalStorageDownloader();
|
DownloadFile downloadFile = new DownloadFile();
|
downloadFile.setFileUrl(fileBean.getFileUrl());
|
InputStream inputStream = downloader.getInputStream(downloadFile);
|
|
try {
|
FileUtils.copyInputStreamToFile(inputStream, destFile);
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
|
|
String extendName = userFile.getExtendName();
|
|
String unzipUrl = UFOPUtils.getTempFile(fileBean.getFileUrl()).getAbsolutePath().replace("." + extendName, "");
|
|
List<String> fileEntryNameList = new ArrayList<>();
|
|
try {
|
fileEntryNameList = FileOperation.unzip(destFile, unzipUrl);
|
} catch (Exception e) {
|
e.printStackTrace();
|
log.error("解压失败" + e);
|
throw new QiwenException(500001, "解压异常：" + e.getMessage());
|
}
|
|
if (destFile.exists()) {
|
destFile.delete();
|
}
|
|
if (!fileEntryNameList.isEmpty() && unzipMode == 1) {
|
DocFilePath parent = fileDealComp.getParent(userFile.getFilePath());
|
DocFilePath qiwenDir = PathFileUtil.getDir( userFile.getFilePath(), userFile.getFileName(), parent);
|
docFilePathMapper.insert(qiwenDir);
|
}
|
|
fileEntryNameList.sort((a,b)->{
|
return a.length() - b.length();
|
});
|
|
for (int i = 0; i < fileEntryNameList.size(); i++){
|
String entryName = fileEntryNameList.get(i);
|
//asyncTaskComp.saveUnzipFile(userFile, fileBean, unzipMode, entryName, filePath);
|
this.saveUnzipFile(userFile, fileBean, unzipMode, entryName, filePath);
|
|
}
|
}
|
|
void saveUnzipFile(DocFilePath userFile, DocFile fileBean, int unzipMode, String entryName, String filePath) {
|
String unzipUrl = UFOPUtils.getTempFile(fileBean.getFileUrl()).getAbsolutePath().replace("." + userFile.getExtendName(), "");
|
String totalFileUrl = unzipUrl + entryName;
|
File currentFile = new File(totalFileUrl);
|
|
String fileId = null;
|
if (!currentFile.isDirectory()) {
|
|
FileInputStream fis = null;
|
String md5Str = UUID.randomUUID().toString();
|
try {
|
fis = new FileInputStream(currentFile);
|
md5Str = DigestUtils.md5Hex(fis);
|
} catch (FileNotFoundException e) {
|
e.printStackTrace();
|
} catch (IOException e) {
|
e.printStackTrace();
|
} finally {
|
IOUtils.closeQuietly(fis);
|
}
|
|
FileInputStream fileInputStream = null;
|
try {
|
Map<String, Object> param = new HashMap<>();
|
param.put("identifier", md5Str);
|
List<DocFile> list = baseMapper.selectByMap(param);
|
|
if (list != null && !list.isEmpty()) { //文件已存在
|
fileId = list.get(0).getFileId();
|
} else { //文件不存在
|
fileInputStream = new FileInputStream(currentFile);
|
CopyFile createFile = new CopyFile();
|
System.out.println("totalFileUrl::" + totalFileUrl);
|
createFile.setExtendName(FilenameUtils.getExtension(totalFileUrl));
|
String saveFileUrl =new LocalStorageCopier().copy(fileInputStream, createFile);
|
|
DocFile tempFileBean = new DocFile(saveFileUrl, currentFile.length(), md5Str, userFile.getUpdateBy());
|
;
|
baseMapper.insert(tempFileBean);
|
fileId = tempFileBean.getFileId();
|
}
|
} catch (IOException e) {
|
e.printStackTrace();
|
} finally {
|
IOUtils.closeQuietly(fileInputStream);
|
System.gc();
|
try {
|
Thread.sleep(100);
|
} catch (InterruptedException e) {
|
e.printStackTrace();
|
}
|
currentFile.delete();
|
}
|
|
|
}
|
|
|
QiwenFile qiwenFile = null;
|
if (unzipMode == 0) {
|
qiwenFile = new QiwenFile(userFile.getFilePath(), entryName, currentFile.isDirectory());
|
|
} else if (unzipMode == 1) {
|
qiwenFile = new QiwenFile(userFile.getFilePath() + "/" + userFile.getFileName(), entryName, currentFile.isDirectory());
|
} else if (unzipMode == 2) {
|
qiwenFile = new QiwenFile(filePath, entryName, currentFile.isDirectory());
|
}
|
|
|
System.out.println("qiwenFile:::" + qiwenFile.getPath());
|
DocFilePath saveUserFile = new DocFilePath(qiwenFile, userFile.getUpdateBy(), fileId);
|
System.out.println("saveUserFile:::" + saveUserFile.toString());
|
DocFilePath parent = fileDealComp.getParent(saveUserFile.getFilePath());
|
if (parent != null) {
|
System.out.println("parent::"+parent.toString());
|
saveUserFile.setParentId(parent.getPathId());
|
}
|
|
String fileName = fileDealComp.getRepeatFileName(saveUserFile, saveUserFile.getFilePath());
|
saveUserFile.setFileName(fileName);
|
if (saveUserFile.getIsDir() == Constant.IS_DIR && !fileName.equals(saveUserFile.getFileName())) {
|
//如果是目录，而且重复，什么也不做
|
} else {
|
|
System.out.println(saveUserFile.toString());
|
docFilePathMapper.insert(saveUserFile);
|
}
|
|
}
|
}
|
