ak/file
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改文件后更新文件大小

Browse Source
This commit is contained in:
马超 2021-07-10 11:34:36 +08:00
parent 3578ec5f7f
commit a41f7b87c6
1 changed files with 35 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
src/main/java/com/qiwenshare/file/controller/OfficeController.java
View File

@ -124,7 +124,7 @@ public class OfficeController {
} }
} }
FileBean fileBean = new FileBean(); FileBean fileBean = new FileBean();
fileBean.setFileSize(Long.valueOf(0)); fileBean.setFileSize(Long.valueOf(1024 * 3));
fileBean.setFileUrl(fileUrl); fileBean.setFileUrl(fileUrl);
fileBean.setStorageType(0); fileBean.setStorageType(0);
fileBean.setPointCount(1); fileBean.setPointCount(1);
@ -261,41 +261,49 @@ public class OfficeController {
} }
PrintWriter writer = null; PrintWriter writer = null;
JSONObject jsonObj=null; JSONObject jsonObj=null;
try {
writer = response.getWriter();
Scanner scanner = new Scanner(request.getInputStream()).useDelimiter("\\A");
String body = scanner.hasNext() ? scanner.next() : "";
jsonObj = JSON.parseObject(body); writer = response.getWriter();
log.debug("===saveeditedfile:" + jsonObj.get("status")) ; Scanner scanner = new Scanner(request.getInputStream()).useDelimiter("\\A");
String status = jsonObj!=null?jsonObj.get("status").toString():""; String body = scanner.hasNext() ? scanner.next() : "";
if ("2".equals(status)) {//新建报告不强制手动操作时状态为2
String type = request.getParameter("type");
String downloadUri = (String) jsonObj.get("url");
if("edit".equals(type)){//修改报告 jsonObj = JSON.parseObject(body);
log.debug("====文档编辑完成，现在开始保存编辑后的文档，其下载地址为:" + downloadUri); log.info("===saveeditedfile:" + jsonObj.get("status")); ;
String fileId = request.getParameter("fileId"); String status = jsonObj!=null?jsonObj.get("status").toString():"";
String userFileId = request.getParameter("userFileId"); if ("2".equals(status)) {//新建报告不强制手动操作时状态为2
FileBean fileBean = fileService.getById(fileId); String type = request.getParameter("type");
if (fileBean.getPointCount() > 1) { String downloadUri = (String) jsonObj.get("url");
//该场景，暂不支持编辑修改
writer.write("{\"error\":1}");
return ;
}
URL url = new URL(downloadUri); if("edit".equals(type)){//修改报告
HttpURLConnection connection = (HttpURLConnection) url.openConnection(); log.debug("====文档编辑完成，现在开始保存编辑后的文档，其下载地址为:" + downloadUri);
String fileId = request.getParameter("fileId");
String userFileId = request.getParameter("userFileId");
FileBean fileBean = fileService.getById(fileId);
if (fileBean.getPointCount() > 1) {
//该场景，暂不支持编辑修改
writer.write("{\"error\":1}");
return ;
}
URL url = new URL(downloadUri);
HttpURLConnection connection = (HttpURLConnection) url.openConnection();
String md5Str = "";
int fileLength = 0;
try {
InputStream stream = connection.getInputStream(); InputStream stream = connection.getInputStream();
fileLength = connection.getContentLength();
md5Str = DigestUtils.md5Hex(stream);
Writer writer1 = ufoFactory.getWriter(fileBean.getStorageType()); Writer writer1 = ufoFactory.getWriter(fileBean.getStorageType());
WriteFile writeFile = new WriteFile(); WriteFile writeFile = new WriteFile();
writeFile.setFileUrl(fileBean.getFileUrl()); writeFile.setFileUrl(fileBean.getFileUrl());
int fileLength = connection.getContentLength();
log.info("当前修改文件大小为：" + fileLength); log.info("当前修改文件大小为：" + fileLength);
log.info("当前修改文件md5为：" + md5Str);
writeFile.setFileSize(connection.getContentLength()); writeFile.setFileSize(connection.getContentLength());
writer1.write(stream, writeFile); writer1.write(stream, writeFile);
} catch (Exception e) {
log.error(e.getMessage());
} finally {
//更新文件修改信息 //更新文件修改信息
LambdaUpdateWrapper<UserFile> userFileUpdateWrapper = new LambdaUpdateWrapper<>(); LambdaUpdateWrapper<UserFile> userFileUpdateWrapper = new LambdaUpdateWrapper<>();
userFileUpdateWrapper userFileUpdateWrapper
@ -303,7 +311,7 @@ public class OfficeController {
.eq(UserFile::getUserFileId, userFileId); .eq(UserFile::getUserFileId, userFileId);
userFileService.update(userFileUpdateWrapper); userFileService.update(userFileUpdateWrapper);
LambdaUpdateWrapper<FileBean> lambdaUpdateWrapper = new LambdaUpdateWrapper<>(); LambdaUpdateWrapper<FileBean> lambdaUpdateWrapper = new LambdaUpdateWrapper<>();
String md5Str = DigestUtils.md5Hex(stream);
log.info("当前修改文件大小为2222：" + Long.valueOf(fileLength)); log.info("当前修改文件大小为2222：" + Long.valueOf(fileLength));
lambdaUpdateWrapper lambdaUpdateWrapper
.set(FileBean::getIdentifier, md5Str) .set(FileBean::getIdentifier, md5Str)
@ -313,12 +321,9 @@ public class OfficeController {
connection.disconnect(); connection.disconnect();
} }
} }
}catch (Exception e) {
log.error(e.getMessage());
} }
String status = jsonObj!=null?jsonObj.get("status").toString():"";
if("3".equals(status)||"7".equals(status)) {//不强制手动保存时为6,"6".equals(status) if("3".equals(status)||"7".equals(status)) {//不强制手动保存时为6,"6".equals(status)
log.debug("====保存失败:"); log.debug("====保存失败:");
writer.write("{\"error\":1}"); writer.write("{\"error\":1}");