文件上传更改
This commit is contained in:
XaoLi717
parent
d4f1e0e52e
commit
cd2e22c7bc
@ -0,0 +1,81 @@
|
||||
package cn.iocoder.yudao.module.bpm.service.onlyoffice.document;
|
||||
|
||||
import cn.iocoder.yudao.module.bpm.controller.admin.onlyoffice.vo.Document;
|
||||
import cn.iocoder.yudao.module.bpm.controller.admin.onlyoffice.vo.DocumentEditParam;
|
||||
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
import java.io.IOException;
|
||||
|
||||
/**
|
||||
* 文档业务 接口
|
||||
* @author: xx
|
||||
* @date: 2024/8/7 16:30
|
||||
*/
|
||||
public interface DocumentService {
|
||||
/**
|
||||
* 构建文档对象
|
||||
* @param filePath
|
||||
* @param fileName
|
||||
* @return documentKey 文档key
|
||||
*/
|
||||
String buildDocument(String filePath, String fileName);
|
||||
|
||||
/**
|
||||
* 从缓从中获取文档信息
|
||||
* @param documentKey
|
||||
* @return
|
||||
*/
|
||||
Document getDocument(String documentKey); /**
|
||||
* 从缓从中获取文档信息
|
||||
* @param documentKey
|
||||
* @return
|
||||
*/
|
||||
Document getDocumentToken(String documentKey,String token);
|
||||
|
||||
|
||||
/**
|
||||
* 下载文档实体文件
|
||||
* @param documentKey
|
||||
* @param request
|
||||
* @param response
|
||||
* @throws IOException
|
||||
*/
|
||||
void downloadDocumentFile(String documentKey, HttpServletRequest request, HttpServletResponse response) throws IOException;
|
||||
|
||||
/**
|
||||
* 构建文档编辑参数 对象
|
||||
* @param userId
|
||||
* @param userName
|
||||
* @return
|
||||
*/
|
||||
DocumentEditParam buildDocumentEditParam(String userId, String userName, String fileName);
|
||||
/**
|
||||
* 构建文档编辑参数 对象
|
||||
* @param userId
|
||||
* @param userName
|
||||
* @return
|
||||
*/
|
||||
DocumentEditParam buildDocumentEditParamToken(String userId, String userName, String fileName,String token);
|
||||
|
||||
/**
|
||||
* 编辑后保存文档实体文件
|
||||
* @param documentKey
|
||||
* @param downloadUrl
|
||||
* @throws IOException
|
||||
*/
|
||||
boolean saveDocumentFile(String documentKey, String downloadUrl) throws IOException;
|
||||
|
||||
/**
|
||||
* 获取服务暴露的host(包含端口)
|
||||
* @return
|
||||
*/
|
||||
Object getServerHost();
|
||||
|
||||
/**
|
||||
* 文档是否支持编辑
|
||||
* @param document
|
||||
* @return
|
||||
*/
|
||||
boolean canEdit(Document document);
|
||||
}
|
||||
@ -0,0 +1,334 @@
|
||||
package cn.iocoder.yudao.module.bpm.service.onlyoffice.document;
|
||||
|
||||
import cn.iocoder.yudao.module.bpm.controller.admin.onlyoffice.utils.DocumentConstants;
|
||||
import cn.iocoder.yudao.module.bpm.controller.admin.onlyoffice.utils.Md5Utils;
|
||||
import cn.iocoder.yudao.module.bpm.controller.admin.onlyoffice.vo.Document;
|
||||
import cn.iocoder.yudao.module.bpm.controller.admin.onlyoffice.vo.DocumentEditParam;
|
||||
import com.alibaba.fastjson.JSON;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.apache.commons.io.FileUtils;
|
||||
import org.apache.commons.io.FilenameUtils;
|
||||
import org.apache.commons.lang3.ArrayUtils;
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.hashids.Hashids;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.core.env.Environment;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
import java.io.*;
|
||||
import java.net.URL;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Paths;
|
||||
@Slf4j
|
||||
@Service
|
||||
public class DocumentServiceImpl implements DocumentService{
|
||||
@Autowired
|
||||
Environment environment;
|
||||
@Value("${document.server.host}")
|
||||
private String serverHost ;
|
||||
/**
|
||||
* 大小限制，默认10M
|
||||
*/
|
||||
@Value("${document.file-size.limit:10485760}")
|
||||
private Long docFileSizeLimit;
|
||||
@Value("${files.docservice.url.site}")
|
||||
private String documentServerHost;
|
||||
@Value("${files.docservice.url.api}")
|
||||
private String documentServerApiJs;
|
||||
// @Autowired
|
||||
//private DocumentCacheService cacheService;
|
||||
|
||||
@Override
|
||||
public String buildDocument(String filePath, String fileName) {
|
||||
if (StringUtils.isBlank(filePath)) {
|
||||
// throw new DocumentException(ErrorCodeEnum.DOC_FILE_NOT_EXISTS);
|
||||
}
|
||||
filePath = FilenameUtils.normalize(filePath);
|
||||
String fileType = StringUtils.lowerCase(FilenameUtils.getExtension(filePath));
|
||||
if (StringUtils.isBlank(fileType)) {
|
||||
// throw new DocumentException(ErrorCodeEnum.DOC_FILE_NO_EXTENSION);
|
||||
}
|
||||
// 如果指定了文件名，则需要校验和实体文件格式是否一致
|
||||
if (StringUtils.isNotBlank(fileName) && !fileType.equalsIgnoreCase(FilenameUtils.getExtension(fileName))) {
|
||||
// throw new DocumentException(ErrorCodeEnum.DOC_FILE_EXTENSION_NOT_MATCH);
|
||||
}
|
||||
File docFile = new File(filePath);
|
||||
// 校验文件实体
|
||||
preFileCheck(docFile);
|
||||
fileName = StringUtils.isNotBlank(fileName) ? fileName : docFile.getName();
|
||||
String fileKey = this.fileKey(docFile, fileName);
|
||||
Document document = Document.builder()
|
||||
.fileType(fileType)
|
||||
.title(fileName)
|
||||
.storage(filePath)
|
||||
.build();
|
||||
boolean cached = false;
|
||||
// try {
|
||||
// cached = cacheService.put(fileKey, document);
|
||||
// } catch (Exception e) {
|
||||
// log.error("$$$ 缓存失败~~", e);
|
||||
// }
|
||||
// if (!cached) {
|
||||
// throw new DocumentException(ErrorCodeEnum.DOC_CACHE_ERROR);
|
||||
// }
|
||||
document.setKey(fileKey);
|
||||
return JSON.toJSONString(document);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Document getDocument(String documentKey ) {
|
||||
Document doc = null;
|
||||
|
||||
try {
|
||||
doc = JSON.parseObject(documentKey,Document.class);
|
||||
} catch (Exception e) {
|
||||
log.error("$$$ 获取缓存失败~~", e);
|
||||
}
|
||||
if (doc == null) {
|
||||
// throw new DocumentException(ErrorCodeEnum.DOC_CACHE_NOT_EXISTS);
|
||||
}
|
||||
// 从缓存中取出后，再绑定非必需缓存字段（节省缓存大小）
|
||||
// doc.setKey(documentKey);
|
||||
doc.setUrl(fileUrl(doc.getTitle()));
|
||||
if (log.isDebugEnabled()) {
|
||||
log.info(doc.toString());
|
||||
}
|
||||
return doc;
|
||||
}
|
||||
@Override
|
||||
public Document getDocumentToken(String documentKey,String token) {
|
||||
Document doc = null;
|
||||
|
||||
try {
|
||||
doc = JSON.parseObject(documentKey,Document.class);
|
||||
} catch (Exception e) {
|
||||
log.error("$$$ 获取缓存失败~~", e);
|
||||
}
|
||||
if (doc == null) {
|
||||
// throw new DocumentException(ErrorCodeEnum.DOC_CACHE_NOT_EXISTS);
|
||||
}
|
||||
// 从缓存中取出后，再绑定非必需缓存字段（节省缓存大小）
|
||||
// doc.setKey(documentKey);
|
||||
doc.setUrl(fileUrlToken(doc.getTitle(),token));
|
||||
if (log.isDebugEnabled()) {
|
||||
log.info(doc.toString());
|
||||
}
|
||||
return doc;
|
||||
}
|
||||
|
||||
/**
|
||||
* 计算文件key值: 文件md5值+路径的短md5值+名称的短md5值
|
||||
* @param docFile
|
||||
* @param name 生成协作时文档的docuemnt.key的值
|
||||
* @return
|
||||
*/
|
||||
public String fileKey(File docFile, String name) {
|
||||
String docFileMd5 = Md5Utils.getFileMd5(docFile);
|
||||
if (StringUtils.isBlank(docFileMd5)) {
|
||||
log.error("$$$ 构建文件信息失败！计算文件 md5 失败！");
|
||||
// throw new DocumentException(ErrorCodeEnum.DOC_FILE_MD5_ERROR);
|
||||
}
|
||||
String pathShortMd5 = Md5Utils.md5(docFile.getAbsolutePath());
|
||||
String nameShortMd5 = Md5Utils.md5(name);
|
||||
Hashids hashids = new Hashids(DocumentConstants.HASH_KEY);
|
||||
// (将路径字符串短md5值 + 名称字符串短md5值) ==> 再转成短id形式 ==> 作为文档的key（暂且认为是不会重复的）
|
||||
String key = hashids.encodeHex(String.format("%s%s%s", docFileMd5,pathShortMd5, nameShortMd5));
|
||||
if (StringUtils.isBlank(key)) {
|
||||
// throw new DocumentException(ErrorCodeEnum.DOC_FILE_KEY_ERROR);
|
||||
}
|
||||
return key;
|
||||
}
|
||||
|
||||
/**
|
||||
* 文件key值
|
||||
* @param fileType
|
||||
* @param docCrc32
|
||||
* @return
|
||||
*/
|
||||
private String fileKey(String fileType, String docCrc32) {
|
||||
return String.format("%s_%s", fileType, docCrc32);
|
||||
}
|
||||
|
||||
/**
|
||||
* 文件url地址
|
||||
* @param
|
||||
* @return
|
||||
*/
|
||||
private String fileUrl(String filename) {
|
||||
return String.format(DocumentConstants.OFFICE_API_DOC_FILE, getServerHost(), "?name="+filename);
|
||||
// return "http://192.168.0.58:20053/download?name="+filename;
|
||||
}
|
||||
/**
|
||||
* 文件url地址
|
||||
* @param
|
||||
* @return
|
||||
*/
|
||||
private String fileUrlToken(String filename,String token) {
|
||||
return String.format(DocumentConstants.OFFICE_API_DOC_FILE, getServerHost(), "?name="+filename+"&token="+token);
|
||||
// return "http://192.168.0.58:20053/download?name="+filename;
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据文档信息下载文档文件
|
||||
* @param documentKey
|
||||
* @param request
|
||||
* @param response
|
||||
* @throws IOException
|
||||
*/
|
||||
@Override
|
||||
public void downloadDocumentFile(String documentKey, HttpServletRequest request, HttpServletResponse response) throws IOException {
|
||||
Document doc = this.getDocument(documentKey);
|
||||
File file = new File(doc.getStorage());
|
||||
try (InputStream reader = new FileInputStream(file);
|
||||
OutputStream out = response.getOutputStream()) {
|
||||
byte[] buf = new byte[(int) FileUtils.ONE_KB];
|
||||
int len = 0;
|
||||
//response.setContentType(mimeType(file));
|
||||
while ((len = reader.read(buf)) != -1) {
|
||||
out.write(buf, 0, len);
|
||||
}
|
||||
out.flush();
|
||||
} catch (Exception e) {
|
||||
log.error("下载失败！读取文件[" + doc.getStorage() + "]报错~~", e);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public DocumentEditParam buildDocumentEditParam(String userId, String userName, String fileName) {
|
||||
return DocumentEditParam.builder()
|
||||
// .callbackUrl(callbackUrl(fileName)+"&token="+token)
|
||||
.callbackUrl(callbackUrl(fileName))
|
||||
.user(DocumentEditParam.UserBean.builder()
|
||||
.id(userId)
|
||||
.name(userName)
|
||||
.build())
|
||||
.build();
|
||||
}
|
||||
@Override
|
||||
public DocumentEditParam buildDocumentEditParamToken(String userId, String userName, String fileName,String token) {
|
||||
return DocumentEditParam.builder()
|
||||
.callbackUrl(callbackUrl(fileName)+"&token="+token)
|
||||
// .callbackUrl(callbackUrl(fileName))
|
||||
.user(DocumentEditParam.UserBean.builder()
|
||||
.id(userId)
|
||||
.name(userName)
|
||||
.build())
|
||||
.build();
|
||||
}
|
||||
|
||||
private String callbackUrl(String fileName) {
|
||||
String format = String.format(DocumentConstants.OFFICE_API_CALLBACK, getServerHost());
|
||||
format=format+"?fileName="+fileName;
|
||||
return format;
|
||||
}
|
||||
|
||||
/**
|
||||
* 上传文档实体文件
|
||||
* @param documentKey
|
||||
* @param downloadUrl
|
||||
* @throws IOException
|
||||
*/
|
||||
@Override
|
||||
public boolean saveDocumentFile(String documentKey, String downloadUrl) {
|
||||
if (log.isInfoEnabled()) {
|
||||
log.info(downloadUrl);
|
||||
}
|
||||
// TODO 默认覆盖源文件，如果调用者指定，则存到临时目录？
|
||||
boolean isCover = true;
|
||||
Document doc = this.getDocument(documentKey);
|
||||
String saveFilePath = doc.getStorage();
|
||||
// if (!isCover) {
|
||||
// String baseDir = environment.getProperty("java.io.tmpdir");
|
||||
// saveFilePath = String.format("%s/office-api/%s/%s.%s", baseDir, documentKey, System.currentTimeMillis(), doc.getFileType());
|
||||
// }
|
||||
File saveFile = new File(saveFilePath);
|
||||
boolean success = false;
|
||||
try {
|
||||
FileUtils.copyURLToFile(new URL(downloadUrl), saveFile);
|
||||
if (saveFile.exists() && saveFile.length() > 0) {
|
||||
success = true;
|
||||
}
|
||||
} catch (IOException e) {
|
||||
log.error("$$$ 保存文档失败！", e);
|
||||
}
|
||||
return success;
|
||||
//TODO 编辑成功后，应该删除之前的编辑状态缓存
|
||||
}
|
||||
|
||||
@Override
|
||||
public Object getServerHost() {
|
||||
if (StringUtils.startsWith(serverHost, DocumentConstants.HTTP_SCHEME)) {
|
||||
return serverHost;
|
||||
}
|
||||
return String.format("http://%s", serverHost);
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean canEdit(Document document) {
|
||||
if (ArrayUtils.contains(DocumentConstants.FILE_TYPE_UNSUPPORT_EDIT, document.getFileType())) {
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取文档信息api地址
|
||||
* @param docId
|
||||
* @return
|
||||
*/
|
||||
private String docInfoUrl(String docId) {
|
||||
return String.format(DocumentConstants.OFFICE_API_DOC, getServerHost(), docId);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取文件的 mimetype
|
||||
* @param file
|
||||
* @deprecated
|
||||
* @return
|
||||
*/
|
||||
@Deprecated
|
||||
private String mimeType(File file) {
|
||||
try {
|
||||
return Files.probeContentType(Paths.get(file.toURI()));
|
||||
} catch (IOException e) {
|
||||
log.error("$$$ 获取文件mimeType错误！", e);
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 先校验文档文件
|
||||
* @param docFile
|
||||
* @return
|
||||
*/
|
||||
private void preFileCheck(File docFile) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("### 开始校验文档：[{}]", docFile.getAbsolutePath());
|
||||
}
|
||||
if (docFile == null || !docFile.exists()) {
|
||||
log.error("$$$ 目标文档不存在，无法打开！");
|
||||
// throw new DocumentException(ErrorCodeEnum.DOC_FILE_NOT_EXISTS);
|
||||
}
|
||||
if (docFile.isDirectory() || docFile.length() <= 0) {
|
||||
log.error("$$$ 目标文档[{}]是目录或空文件，无法打开！", docFile.getAbsolutePath());
|
||||
// throw new DocumentException(ErrorCodeEnum.DOC_FILE_EMPTY);
|
||||
}
|
||||
if (!docFile.canRead()) {
|
||||
log.error("$$$ 目标文档[{}]不可读，无法打开！", docFile.getAbsolutePath());
|
||||
// throw new DocumentException(ErrorCodeEnum.DOC_FILE_UNREADABLE);
|
||||
}
|
||||
if (docFile.length() > docFileSizeLimit) {
|
||||
log.error("$$$ 目标文档大小超过限制({}B > {}B)，无法打开！", docFile.length(), docFileSizeLimit);
|
||||
// throw new DocumentException(ErrorCodeEnum.DOC_FILE_OVERSIZE);
|
||||
}
|
||||
String ext = StringUtils.lowerCase(FilenameUtils.getExtension(docFile.getName()));
|
||||
if (!ArrayUtils.contains(DocumentConstants.FILE_TYPE_SUPPORT_VIEW, ext)) {
|
||||
log.error("$$$ 目标文档格式[{}]不正确，无法打开！（只支持：{}）",
|
||||
ext, StringUtils.join(DocumentConstants.FILE_TYPE_SUPPORT_VIEW, ","));
|
||||
// throw new DocumentException(ErrorCodeEnum.DOC_FILE_TYPE_UNSUPPORTED);
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,10 @@
|
||||
package cn.iocoder.yudao.module.bpm.service.onlyoffice.fileupload;
|
||||
|
||||
import cn.iocoder.yudao.module.bpm.dal.dataobject.onlyoffice.onlyofficeDO;
|
||||
import com.baomidou.mybatisplus.extension.service.IService;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public interface FileUploadService extends IService<onlyofficeDO> {
|
||||
List<onlyofficeDO> list();
|
||||
}
|
||||
@ -0,0 +1,21 @@
|
||||
package cn.iocoder.yudao.module.bpm.service.onlyoffice.fileupload;
|
||||
|
||||
import cn.iocoder.yudao.module.bpm.dal.dataobject.onlyoffice.onlyofficeDO;
|
||||
import cn.iocoder.yudao.module.bpm.dal.mysql.onlyoffice.FileUploadMapper;
|
||||
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import java.util.List;
|
||||
@Service
|
||||
public class FileUploadServiceImpl extends ServiceImpl<FileUploadMapper, onlyofficeDO> implements FileUploadService {
|
||||
@Autowired
|
||||
FileUploadMapper uploadMapper;
|
||||
|
||||
@Override
|
||||
public List<onlyofficeDO> list() {
|
||||
return uploadMapper.selectList(null);
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user